Examples with CellInfoResolver com.netflix.titus.federation.service.CellInfoResolver used on opensource projects

Example 6 with CellInfoResolver

use of com.netflix.titus.federation.service.CellInfoResolver in project titus-control-plane by Netflix.

the class ApplicationCellRouterTest method jobToCellAffinity.

@Test
public void jobToCellAffinity() {
    JobDescriptor withCellAffinity = APP_2.toBuilder().putAttributes(JobAttributes.JOB_PARAMETER_ATTRIBUTES_CELL_REQUEST, "cell2").build();
    TitusFederationConfiguration titusFederationConfiguration = mock(TitusFederationConfiguration.class);
    when(titusFederationConfiguration.getCells()).thenReturn("cell1=hostName1:7001;cell2=hostName2:7002");
    when(titusFederationConfiguration.getRoutingRules()).thenReturn("cell1=(app1.*|app2.*);cell2=(app3.*)");
    CellInfoResolver cellInfoResolver = new DefaultCellInfoResolver(titusFederationConfiguration);
    ApplicationCellRouter cellRouter = new ApplicationCellRouter(cellInfoResolver, titusFederationConfiguration);
    assertThat(cellRouter.routeKey(withCellAffinity).get().getName()).isEqualTo("cell2");
}

Example 7 with CellInfoResolver

use of com.netflix.titus.federation.service.CellInfoResolver in project titus-control-plane by Netflix.

the class ApplicationCellRouterTest method cellsWithNoRulesCanExist.

@Test
public void cellsWithNoRulesCanExist() {
    TitusFederationConfiguration titusFederationConfiguration = mock(TitusFederationConfiguration.class);
    when(titusFederationConfiguration.getCells()).thenReturn("cell1=hostName1:7001;cell2=hostName2:7002");
    when(titusFederationConfiguration.getRoutingRules()).thenReturn("cell1=(app1.*|app2.*)");
    CellInfoResolver cellInfoResolver = new DefaultCellInfoResolver(titusFederationConfiguration);
    ApplicationCellRouter cellRouter = new ApplicationCellRouter(cellInfoResolver, titusFederationConfiguration);
    Cell cell = cellRouter.routeKey(APP_2).get();
    assertThat(cell.getName()).isEqualTo("cell1");
}