@Before public void setup() throws Exception { zkComponent = new ZKServerComponent(); zkComponent.start(); client = ConfigurationsUtils.getClient(zkComponent.getConnectionString()); client.start(); cache = new ZKConfigurationsCache(client);
@After public void teardown() throws Exception { if(cache != null) { cache.close(); } if(client != null) { client.close(); } if(zkComponent != null) { zkComponent.stop(); } }
@BeforeClass public static void setupBeforeClass() throws Exception { UnitTestHelper.setJavaLoggingLevel(Level.SEVERE); LOG.info("Starting up YARN cluster"); zkServerComponent = new ZKServerComponent(); yarnComponent = new YarnComponent().withApplicationMasterClass(ApplicationMaster.class).withTestName(MaasIntegrationTest.class.getSimpleName()); runner = new ComponentRunner.Builder() .withComponent("yarn", yarnComponent) .withComponent("zk", zkServerComponent) .withMillisecondsBetweenAttempts(15000) .withNumRetries(10) .build(); runner.start(); String zookeeperUrl = zkServerComponent.getConnectionString(); RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3); client = CuratorFrameworkFactory.newClient(zookeeperUrl, retryPolicy); client.start(); }
@Before public void setup() throws Exception { // a component that uploads the global configuration Map<String, Object> globals = new HashMap<>(); ConfigUploadComponent configUploader = new ConfigUploadComponent() .withGlobals(globals); // create zookeeper component properties = new Properties(); zkServer = getZKServerComponent(properties); // can only get the zookeeperUrl AFTER it has started zkServer.withPostStartCallback((zk) -> { zookeeperURL = zk.getConnectionString(); configUploader.withZookeeperURL(zookeeperURL); }); // start the integration test components runner = new ComponentRunner.Builder() .withComponent("zk", zkServer) .withComponent("config", configUploader) .build(); runner.start(); context = mock(InterpreterContext.class); }
"--zk_quorum", zkServerComponent.getConnectionString(), "--zk_root", configRoot, "--master_memory", "512", "--name", "dummy", "--version", "1.0", "--zk_quorum", zkServerComponent.getConnectionString(), "--zk_root", configRoot, "--local_model_path", "src/test/resources/maas", "--name", "dummy", "--version", "1.0", "--zk_quorum", zkServerComponent.getConnectionString(), "--zk_root", configRoot, "--num_instances", "1",