public ModuleList(Injector baseInjector) { this.baseInjector = baseInjector; this.modulesConfig = baseInjector.getInstance(ModulesConfig.class); this.jsonMapper = baseInjector.getInstance(Key.get(ObjectMapper.class, Json.class)); this.smileMapper = baseInjector.getInstance(Key.get(ObjectMapper.class, Smile.class)); this.modules = new ArrayList<>(); }
protected Optional<JobHistoryStore> createJobHistoryStore(Properties jobProps) { boolean jobHistoryStoreEnabled = Boolean .valueOf(jobProps.getProperty(ConfigurationKeys.JOB_HISTORY_STORE_ENABLED_KEY, Boolean.FALSE.toString())); if (jobHistoryStoreEnabled) { Injector injector = Guice.createInjector(new MetaStoreModule(jobProps)); return Optional.of(injector.getInstance(JobHistoryStore.class)); } else { return Optional.absent(); } }
@Test public void testMultiConditionalBind_cat() { props.setProperty("animal.type", "cat"); Injector injector = Guice.createInjector(new Module() { @Override public void configure(Binder binder) { ConditionalMultibind.create(props, binder, Animal.class) .addConditionBinding(ANIMAL_TYPE, Predicates.equalTo("cat"), Cat.class) .addConditionBinding(ANIMAL_TYPE, Predicates.equalTo("dog"), Dog.class); } }); Set<Animal> animalSet = injector.getInstance(Key.get(new TypeLiteral<Set<Animal>>() { })); Assert.assertEquals(1, animalSet.size()); Assert.assertEquals(animalSet, ImmutableSet.<Animal>of(new Cat())); }
@Test public void testExcludesCurrentNode() throws Exception .initialize(); ServiceSelector selector = injector.getInstance(Key.get(ServiceSelector.class, serviceType("presto"))); assertEquals(selector.selectAllServices().size(), 1); HeartbeatFailureDetector detector = injector.getInstance(HeartbeatFailureDetector.class); detector.updateMonitoredServices(); assertEquals(detector.getTotalCount(), 0); assertEquals(detector.getActiveCount(), 0); assertEquals(detector.getFailedCount(), 0); assertTrue(detector.getFailed().isEmpty());
public void testApplyInputStream() { InputStream is = getClass().getResourceAsStream("/versions.xml"); Injector injector = Guice.createInjector(new SaxParserModule()); Factory factory = injector.getInstance(ParseSax.Factory.class); SortedMap<String, URI> result = factory.create( injector.getInstance(SupportedVersionsHandler.class)).parse(is); assertEquals(result, ImmutableSortedMap.of("0.8", URI .create("https://services.vcloudexpress.terremark.com/api/v0.8/login"))); } }
public void testInstanceTypeHandler() { InputStream is = getClass().getResourceAsStream("/instanceType.xml"); InstanceTypeHandler handler = injector.getInstance(InstanceTypeHandler.class); String result = factory.create(handler).parse(is); assertEquals(result, InstanceType.M1_SMALL); }
@Test public void test() { InputStream is = getClass().getResourceAsStream("/records.xml"); FluentIterable<ResourceRecordDetail> expected = expected(); ResourceRecordListHandler handler = injector.getInstance(ResourceRecordListHandler.class); FluentIterable<ResourceRecordDetail> result = factory.create(handler).parse(is); assertEquals(result.toList().toString(), expected.toList().toString()); }
@Test public void test() { ThrottlingGuiceServletConfig guiceServletConfig = new ThrottlingGuiceServletConfig(); ThrottlingPolicyFactory factory = new ThrottlingPolicyFactory(); SharedLimiterKey res1key = new SharedLimiterKey("res1"); Map<String, String> configMap = avro.shaded.com.google.common.collect.ImmutableMap.<String, String>builder() .put(BrokerConfigurationKeyGenerator.generateKey(factory, res1key, null, ThrottlingPolicyFactory.POLICY_KEY), CountBasedPolicy.FACTORY_ALIAS) .put(BrokerConfigurationKeyGenerator.generateKey(factory, res1key, null, CountBasedPolicy.COUNT_KEY), "100") .build(); guiceServletConfig.initialize(ConfigFactory.parseMap(configMap)); Injector injector = guiceServletConfig.getInjector(); PoliciesResource policiesResource = injector.getInstance(PoliciesResource.class); Policy policy = policiesResource.get("res1"); Assert.assertEquals(policy.getPolicyName(), CountBasedPolicy.class.getSimpleName()); Assert.assertEquals(policy.getResource(), "res1"); Assert.assertEquals(policy.getParameters().get("maxPermits"), "100"); guiceServletConfig.close(); }
@Test(groups = "fast") public void testAOPForClass() throws Exception { // Make sure it works as expected without any AOP magic final IAopTester simpleTester = new AopTester(); try { simpleTester.createRefund(); } catch (Exception e) { Assert.fail(e.getLocalizedMessage()); } // Now, verify the interception works configureShiro(); final Injector injector = Guice.createInjector(Stage.PRODUCTION, new ShiroModuleNoDB(configSource), new KillBillShiroAopModule(), new TestSecurityModuleNoDB(configSource), new CacheModule(configSource), new AbstractModule() { @Override protected void configure() { bind(IDBI.class).toInstance(Mockito.mock(IDBI.class)); bind(IDBI.class).annotatedWith(Names.named(MAIN_RO_DATA_SOURCE_ID)).toInstance(Mockito.mock(IDBI.class)); bind(TenantInternalApi.class).toInstance(Mockito.mock(TenantInternalApi.class)); bind(NonEntityDao.class).toInstance(Mockito.mock(NonEntityDao.class)); } }); final AopTester aopedTester = injector.getInstance(AopTester.class); verifyAopedTester(aopedTester); }
@Test public void test() { ThrottlingGuiceServletConfig guiceServletConfig = new ThrottlingGuiceServletConfig(); guiceServletConfig.initialize(ConfigFactory.empty()); Injector injector = guiceServletConfig.getInjector(); LimiterServerResource limiterServer = injector.getInstance(LimiterServerResource.class); PermitRequest request = new PermitRequest(); request.setPermits(10); request.setResource("myResource"); PermitAllocation allocation = limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord())); Assert.assertTrue(allocation.getPermits() >= 10); }
@Test public void testWithoutConfigAllIdsMatch() { Set<String> regionIds = ImmutableSet.of("us-east-1", "eu-west-1"); RegionIdFilter filter = Guice.createInjector(new AbstractModule(){ @Override protected void configure() { bindConstant().annotatedWith(Provider.class).to("aws-ec2"); } }).getInstance(AnyOrConfiguredRegionId.class); assertEquals(Sets.filter(regionIds, filter), ImmutableSet.of("us-east-1", "eu-west-1")); }
@BeforeClass public static void setup() throws InterruptedException, IOException { new MockNodeProbe(); injector = Guice.createInjector(new BRTestModule()); filesystem = (FakeBackupFileSystem) injector.getInstance( Key.get(IBackupFileSystem.class, Names.named("backup"))); }
public void testApplyInputStream() { InputStream is = getClass().getResourceAsStream("/versions.xml"); Injector injector = Guice.createInjector(new SaxParserModule()); Factory factory = injector.getInstance(ParseSax.Factory.class); SortedMap<String, URI> result = factory.create( injector.getInstance(SupportedVersionsHandler.class)).parse(is); assertEquals(result, ImmutableSortedMap.of("0.8", URI .create("https://services.vcloudexpress.terremark.com/api/v0.8/login"))); } }
@BeforeMethod public void startUp() { Injector injector = Guice.createInjector(new JsonModule(), new HandleJsonModule()); objectMapper = injector.getInstance(ObjectMapper.class); }
@Override public Emitter apply(String s) { return injector.getInstance(Key.get(Emitter.class, Names.named(s))); } }
public void testUnencodeStringValueHandler() { InputStream is = getClass().getResourceAsStream("/userData.xml"); UnencodeStringValueHandler handler = injector.getInstance(UnencodeStringValueHandler.class); String result = factory.create(handler).parse(is); assertEquals(result, "#!/bin/bash\n"); } }
@Test public void test() { InputStream is = getClass().getResourceAsStream("/records_by_name_and_type.xml"); FluentIterable<ResourceRecordDetail> expected = expected(); ResourceRecordListHandler handler = injector.getInstance(ResourceRecordListHandler.class); FluentIterable<ResourceRecordDetail> result = factory.create(handler).parse(is); assertEquals(result.toList().toString(), expected.toList().toString()); }
@Test public void testMetrics() throws Exception { ThrottlingGuiceServletConfig guiceServletConfig = new ThrottlingGuiceServletConfig(); guiceServletConfig.initialize(ConfigFactory.empty()); Injector injector = guiceServletConfig.getInjector(); LimiterServerResource limiterServer = injector.getInstance(LimiterServerResource.class); PermitRequest request = new PermitRequest(); request.setPermits(10); request.setResource("myResource"); limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord())); limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord())); limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord())); MetricContext metricContext = limiterServer.metricContext; Timer timer = metricContext.timer(LimiterServerResource.REQUEST_TIMER_NAME); Assert.assertEquals(timer.getCount(), 3); }
@Test(groups = "fast") public void testAOPForInterface() throws Exception { // Make sure it works as expected without any AOP magic final IAopTester simpleTester = new AopTesterImpl(); try { simpleTester.createRefund(); } catch (Exception e) { Assert.fail(e.getLocalizedMessage()); } // Now, verify the interception works configureShiro(); final Injector injector = Guice.createInjector(Stage.PRODUCTION, new ShiroModuleNoDB(configSource), new KillBillShiroAopModule(), new TestSecurityModuleNoDB(configSource), new CacheModule(configSource), new AbstractModule() { @Override public void configure() { bind(IDBI.class).toInstance(Mockito.mock(IDBI.class)); bind(IDBI.class).annotatedWith(Names.named(MAIN_RO_DATA_SOURCE_ID)).toInstance(Mockito.mock(IDBI.class)); bind(IAopTester.class).to(AopTesterImpl.class).asEagerSingleton(); bind(TenantInternalApi.class).toInstance(Mockito.mock(TenantInternalApi.class)); bind(NonEntityDao.class).toInstance(Mockito.mock(NonEntityDao.class)); } }); final IAopTester aopedTester = injector.getInstance(IAopTester.class); verifyAopedTester(aopedTester); }
@Test void testHear() { Injector i = Guice.createInjector(new AbstractModule() { @Override protected void configure() { bindListener(any(), blawr); } }); assertEquals(i.getInstance(A.class).logger.getCategory(), getClass().getName() + "$A"); assertEquals(i.getInstance(B.class).logger.getCategory(), getClass().getName() + "$B"); assertEquals(i.getInstance(B.class).blogger.getCategory(), "blogger"); }