Examples with EmbeddedTitusCell com.netflix.titus.testkit.embedded.cell.EmbeddedTitusCell used on opensource projects

Example 1 with EmbeddedTitusCell

use of com.netflix.titus.testkit.embedded.cell.EmbeddedTitusCell in project titus-control-plane by Netflix.

the class EmbeddedTitusCellTest method testShutdownCleanup.

@Test
@Ignore
public void testShutdownCleanup() {
    Set<Thread> threadsBefore = Thread.getAllStackTraces().keySet();
    EmbeddedTitusCell titusStack = EmbeddedTitusCells.basicKubeCell(1);
    titusStack.boot();
    titusStack.shutdown();
    Set<Thread> threadsAfter = Thread.getAllStackTraces().keySet();
    // Filter-out non Titus threads
    Set<Thread> titusRemainingThreads = threadsAfter.stream().filter(t -> {
        ThreadGroup threadGroup = t.getThreadGroup();
        if (threadGroup == null) {
            // if thread died
            return false;
        }
        return !matches(OK_THREAD_GROUPS, threadGroup.getName());
    }).filter(t -> !matches(OK_THREADS, t.getName())).collect(Collectors.toSet());
    CollectionsExt.copyAndRemove(titusRemainingThreads, threadsBefore).forEach(t -> {
        logger.info("Unexpected thread {}:\n{}", t.getName(), String.join("\n", Arrays.stream(t.getStackTrace()).map(e -> "  " + e).collect(Collectors.toList())));
    });
    assertThat(titusRemainingThreads).isSubsetOf(threadsBefore);
}

Example 2 with EmbeddedTitusCell

use of com.netflix.titus.testkit.embedded.cell.EmbeddedTitusCell in project titus-control-plane by Netflix.

the class EmbeddedTitusStackRunner method main.

public static void main(String[] args) throws InterruptedException {
    CommandLineFacade cliFacade = buildCliFacade(args);
    EmbeddedTitusMaster titusMaster = EmbeddedTitusMaster.aTitusMaster().withEmbeddedKubeCluster(EmbeddedKubeClusters.basicCluster(2)).withApiPort(8080).withGrpcPort(7104).withEnableDisruptionBudget(true).build();
    boolean federationEnabled = cliFacade.isEnabled("f");
    EmbeddedTitusCell cell = EmbeddedTitusCell.aTitusCell().withMaster(titusMaster).withGateway(EmbeddedTitusGateway.aDefaultTitusGateway().withMasterEndpoint("localhost", 8090, 8080).withHttpPort(8081).withGrpcPort(8091).withProperty("titus.relocation.grpcClient.grpcPort", "7105").withProperty("titus.feature.mergingTaskMigrationPlanInGatewayEnabled", "true").build(), !federationEnabled).build();
    cell.boot();
    if (federationEnabled) {
        EmbeddedTitusFederation stack = EmbeddedTitusFederation.aDefaultTitusFederation().withCell(".*", cell).withHttpPort(8082).withGrpcPort(8092).build();
        stack.boot();
    }
    System.out.println("TitusStack started");
    Thread.sleep(24 * 60 * 60 * 1000L);
}