@Test public void testStringTemplatesInConfig() throws ConfigurationException, ClassNotFoundException { envVars.set("CUSTOM_SOURCE_NAME", "TestSourceName"); BenderConfig config = BenderConfig.load("/config/config_with_env.json"); List<SourceConfig> sources = config.getSources(); assertEquals(sources.size(), 1); assertEquals(sources.get(0).getName(), "TestSourceName"); }
@Test public void testYaml() throws ConfigurationException, ClassNotFoundException { envVars.set("CUSTOM_SOURCE_NAME", "TestSourceName"); BenderConfig config = BenderConfig.load("/config/config_with_env.yaml"); List<SourceConfig> sources = config.getSources(); assertEquals(sources.size(), 1); assertEquals(sources.get(0).getName(), "TestSourceName"); } }
@Test public void testLoadConfg() { BenderConfig config = BenderConfig.load("/com/nextdoor/bender/partition/partition_config.json"); PartitionOperationConfig op = (PartitionOperationConfig) config.getSources().get(0).getOperationConfigs().get(0); PartitionSpec spec = op.getPartitionSpecs().get(0); assertEquals("type", spec.getName()); assertEquals(Interpreter.STRING, spec.getInterpreter()); List<String> expected = Arrays.asList("one", "two", "three"); assertEquals(expected, spec.getSources()); }
public HandlerResources(BenderConfig config) throws ClassNotFoundException { setSources(config.getSources()); setSerializerProcessor(config.getSerializerConfig()); setTransportFactory(config.getTransportConfig()); setWrapperFactory(config.getWrapperConfig()); setReporters(config.getReporters()); }
public HandlerResources(BenderConfig config) throws ClassNotFoundException { setSources(config.getSources()); setSerializerProcessor(config.getSerializerConfig()); setTransportFactory(config.getTransportConfig()); setWrapperFactory(config.getWrapperConfig()); setReporters(config.getReporters()); }
@Test public void testLoadConfgTime() { BenderConfig config = BenderConfig.load("/com/nextdoor/bender/partition/partition_config_time.json"); PartitionOperationConfig op = (PartitionOperationConfig) config.getSources().get(0).getOperationConfigs().get(0); PartitionSpec spec = op.getPartitionSpecs().get(0); assertEquals("dt", spec.getName()); assertEquals(Interpreter.SECONDS, spec.getInterpreter()); List<String> expected = Arrays.asList("epoch"); assertEquals(expected, spec.getSources()); assertEquals("2017-01-19 05:05:59", spec.interpret("1484802359")); }
@Test public void testS3Config() throws HandlerException { BaseHandler handler = new DummyHandler(); handler.CONFIG_FILE = null; handler.s3ClientFactory = this.clientFactory; List<DummyEvent> events = new ArrayList<DummyEvent>(1); TestContext context = new TestContext(); context.setInvokedFunctionArn("arn:aws:lambda:us-east-1:123:function:test"); handler.handler(events, context); assertEquals("s3://" + S3_BUCKET + "/bender/config.json", handler.config.getConfigFile()); assertEquals("Test Events", handler.config.getSources().get(0).getName()); } }
@Test public void testStringFormatUpper() { BenderConfig config = BenderConfig.load("/com/nextdoor/bender/partition/partition_config_format.json"); PartitionOperationConfig op = (PartitionOperationConfig) config.getSources().get(0).getOperationConfigs().get(0); PartitionSpec spec = op.getPartitionSpecs().get(0); assertEquals("type", spec.getName()); assertEquals(Interpreter.STRING, spec.getInterpreter()); assertEquals(StringFormat.TOUPPER, spec.getStringFormat()); assertEquals("THISISATEST", spec.interpret("ThisIsATest")); }
@Test public void testStringFormatLower() { BenderConfig config = BenderConfig.load("/com/nextdoor/bender/partition/partition_config_format.json"); PartitionOperationConfig op = (PartitionOperationConfig) config.getSources().get(0).getOperationConfigs().get(0); PartitionSpec spec = op.getPartitionSpecs().get(0); spec.setStringFormat(StringFormat.TOLOWER); assertEquals("type", spec.getName()); assertEquals(Interpreter.STRING, spec.getInterpreter()); assertEquals("thisisatest", spec.interpret("ThisIsATest")); } }