com.google.inject.AbstractModule源码实例Demo-ICode联盟

类com.google.inject.AbstractModule源码实例Demo

下面列出了怎么用com.google.inject.AbstractModule的API类实例代码及写法，或者点击链接到github查看源代码。

源代码1 项目： hadoop-ozone 文件： TestStorageContainerServiceProviderImpl.java

@Before
public void setup() {
  injector = Guice.createInjector(new AbstractModule() {
    @Override
    protected void configure() {
      try {
        StorageContainerLocationProtocol mockScmClient = mock(
            StorageContainerLocationProtocol.class);
        pipelineID = PipelineID.randomId().getProtobuf();
        when(mockScmClient.getPipeline(pipelineID))
            .thenReturn(mock(Pipeline.class));
        bind(StorageContainerLocationProtocol.class)
            .toInstance(mockScmClient);
        bind(StorageContainerServiceProvider.class)
            .to(StorageContainerServiceProviderImpl.class);
      } catch (Exception e) {
        Assert.fail();
      }
    }
  });
}

源代码2 项目： hivemq-community-edition 文件： GuiceBootstrap.java

@Nullable
public static Injector bootstrapInjector(
        final @NotNull SystemInformation systemInformation,
        final @NotNull MetricRegistry metricRegistry,
        final @NotNull HivemqId hiveMQId,
        final @NotNull FullConfigurationService fullConfigurationService,
        final @NotNull Injector persistenceInjector) {

    if (!Boolean.parseBoolean(System.getProperty("diagnosticMode"))) {
        log.trace("Turning Guice stack traces off");
        System.setProperty("guice_include_stack_traces", "OFF");
    }

    final ImmutableList.Builder<AbstractModule> modules = ImmutableList.builder();

    return getInjector(systemInformation,
            metricRegistry,
            hiveMQId,
            fullConfigurationService,
            modules,
            persistenceInjector);
}

源代码3 项目： hivemq-community-edition 文件： GuiceBootstrap.java

public static @NotNull Injector persistenceInjector(
        final @NotNull SystemInformation systemInformation,
        final @NotNull MetricRegistry metricRegistry,
        final @NotNull HivemqId hiveMQId,
        final @NotNull FullConfigurationService configService) {

    final ImmutableList.Builder<AbstractModule> modules = ImmutableList.builder();

    modules.add(new SystemInformationModule(systemInformation),
            new ConfigurationModule(configService, hiveMQId),
            new LazySingletonModule(),
            LifecycleModule.get(),
            new PersistenceMigrationModule(metricRegistry, configService.persistenceConfigurationService()));

    return Guice.createInjector(Stage.PRODUCTION, modules.build());
}

源代码4 项目： hivemq-community-edition 文件： SecurityModuleTest.java

@Test
public void test_ssl_factory_same() {

    final Injector injector = Guice.createInjector(Stage.PRODUCTION, new SecurityModule(), new AbstractModule() {
        @Override
        protected void configure() {
            bind(EventExecutorGroup.class).toInstance(mock(EventExecutorGroup.class));
            bind(ShutdownHooks.class).toInstance(mock(ShutdownHooks.class));
            bindScope(LazySingleton.class, LazySingletonScope.get());
        }
    });

    final SslFactory instance1 = injector.getInstance(SslFactory.class);
    final SslFactory instance2 = injector.getInstance(SslFactory.class);

    assertSame(instance1, instance2);

}

源代码5 项目： hivemq-community-edition 文件： MetricsModuleTest.java

@Before
public void before() {
    MockitoAnnotations.initMocks(this);

    injector = Guice.createInjector(new AbstractModule() {
        @Override
        protected void configure() {
            final Injector persistenceInjector = mock(Injector.class);
            when(persistenceInjector.getInstance(MetricsHolder.class)).thenReturn(mock(MetricsHolder.class));
            final NettyConfiguration nettyConfiguration = mock(NettyConfiguration.class);
            when(nettyConfiguration.getChildEventLoopGroup()).thenReturn(mock(EventLoopGroup.class));

            bind(NettyConfiguration.class).toInstance(nettyConfiguration);
            bind(ChannelGroup.class).toInstance(mock(ChannelGroup.class));
            bind(ClientSessionLocalPersistence.class).toInstance(mock(ClientSessionLocalPersistence.class));
            bind(LocalTopicTree.class).toInstance(mock(TopicTreeImpl.class));
            bind(RetainedMessagePersistence.class).toInstance(mock(RetainedMessagePersistence.class));
            bind(SystemInformation.class).toInstance(mock(SystemInformation.class));
            bindScope(LazySingleton.class, LazySingletonScope.get());

            install(new MetricsModule(new MetricRegistry(), persistenceInjector));
        }
    });
}

源代码6 项目： hivemq-community-edition 文件： PersistenceMigrationModuleTest.java

@Test
public void test_startup_singleton() {
    final Injector injector = Guice.createInjector(
            new PersistenceMigrationModule(new MetricRegistry(), persistenceConfigurationService),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(SystemInformation.class).toInstance(systemInformation);
                    bindScope(LazySingleton.class, LazySingletonScope.get());
                    bind(MqttConfigurationService.class).toInstance(mqttConfigurationService);
                }
            });

    final PersistenceStartup instance1 = injector.getInstance(PersistenceStartup.class);
    final PersistenceStartup instance2 = injector.getInstance(PersistenceStartup.class);

    assertSame(instance1, instance2);
}

源代码7 项目： hivemq-community-edition 文件： PersistenceMigrationModuleTest.java

@Test
public void test_memory_persistence() {
    when(persistenceConfigurationService.getMode()).thenReturn(PersistenceConfigurationService.PersistenceMode.IN_MEMORY);

    final Injector injector = Guice.createInjector(
            new PersistenceMigrationModule(new MetricRegistry(), persistenceConfigurationService),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(SystemInformation.class).toInstance(systemInformation);
                    bindScope(LazySingleton.class, LazySingletonScope.get());
                    bind(MqttConfigurationService.class).toInstance(mqttConfigurationService);
                }
            });

    assertTrue(injector.getInstance(RetainedMessageLocalPersistence.class) instanceof RetainedMessageMemoryLocalPersistence);
    assertTrue(injector.getInstance(PublishPayloadLocalPersistence.class) instanceof PublishPayloadMemoryLocalPersistence);
}

源代码8 项目： hivemq-community-edition 文件： ConfigurationModuleTest.java

@Before
public void setUp() throws Exception {

    MockitoAnnotations.initMocks(this);

    testConfigurationBootstrap = new TestConfigurationBootstrap();
    final FullConfigurationService fullConfigurationService = testConfigurationBootstrap.getFullConfigurationService();

    injector = Guice.createInjector(new SystemInformationModule(new SystemInformationImpl()), new ConfigurationModule(fullConfigurationService, new HivemqId()),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(SharedSubscriptionService.class).toInstance(sharedSubscriptionService);
                }
            });
}

源代码9 项目： hivemq-community-edition 文件： HiveMQMainModuleTest.java

@Test
public void test_topic_matcher_not_same() {

    final Injector injector = Guice.createInjector(Stage.PRODUCTION,
            new HiveMQMainModule(),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(EventExecutorGroup.class).toInstance(Mockito.mock(EventExecutorGroup.class));
                }
            });

    final TopicMatcher instance1 = injector.getInstance(TopicMatcher.class);
    final TopicMatcher instance2 = injector.getInstance(TopicMatcher.class);

    assertNotSame(instance1, instance2);

}

源代码10 项目： arcusplatform 文件： TestIpcdBridgeServiceImpl.java

@SuppressWarnings("unchecked")
@Before
public void setUp() throws Exception {
	Bootstrap bootstrap = Bootstrap.builder()
	         .withModuleClasses(IpcdServerModule.class, GsonModule.class, MessagesModule.class)
	         .withModules(new AbstractModule() {
               @Override
               protected void configure() {
                  bind(ProtocolMessageBus.class).to(InMemoryProtocolMessageBus.class);
                  bind(PlatformMessageBus.class).to(InMemoryPlatformMessageBus.class);
               }

	         })
           .withConfigPaths("src/dist/conf/ipcd-bridge.properties")
           .build();
	   ServiceLocator.init(GuiceServiceLocator.create(bootstrap.bootstrap()));

		this.protocolBus = ServiceLocator.getInstance(InMemoryProtocolMessageBus.class);
		this.ipcdDevice = new Device();
		this.ipcdDevice.setIpcdver("0.3");
		this.ipcdDevice.setVendor("Blackbox");
		this.ipcdDevice.setModel("Switch1");
		this.ipcdDevice.setSn("123456789");
		this.clientId = ProtocolDeviceId.hashDeviceId(ipcdDevice.getVendor() + "-" + ipcdDevice.getModel() + "-" + ipcdDevice.getSn()).getRepresentation();
}

源代码11 项目： arcusplatform 文件： IrisTestCase.java

protected Injector bootstrap() throws ClassNotFoundException, BootstrapException {
   final Module m = ProviderMethodsModule.forObject(this);
   return
      Bootstrap
         .builder()
         .withModuleClassnames(moduleClassNames())
         .withModuleClasses(moduleClasses())
         .withModules(modules())
         .withModules(new AbstractModule() {
            @Override
            protected void configure() {
               // generating sources for ProviderMethods will result in an NPE
               m.configure(binder().skipSources(IrisTestCase.this.getClass()));
               // get injections
               bind(IrisTestCase.class).toInstance(IrisTestCase.this);
               IrisTestCase.this.configure(binder());
            }
         })
         .withConfigPaths(configs())
         .build()
         .bootstrap();
}

源代码12 项目： cryptotrader 文件： ResteasyContextListenerTest.java

@BeforeMethod
public void setUp() {

    target = new ResteasyContextListener();

    provider = mock(ConfigurationProvider.class);

    trader = mock(Trader.class);

    endpoint = new EndpointImpl(Guice.createInjector(new AbstractModule() {
        @Override
        protected void configure() {
            bind(ConfigurationProvider.class).toInstance(provider);
            bind(Trader.class).toInstance(trader);
        }
    }));

}

源代码13 项目： titus-control-plane 文件： ActivationProvisionListenerTest.java

@Test
public void testActivationLifecycle() throws Exception {
    LifecycleInjector injector = InjectorBuilder.fromModules(
            new ContainerEventBusModule(),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(TitusRuntime.class).toInstance(TitusRuntimes.internal());
                    bind(ActivationProvisionListenerTest.class).toInstance(ActivationProvisionListenerTest.this);
                    bind(ServiceA.class).asEagerSingleton();
                }
            }).createInjector();

    ActivationLifecycle activationLifecycle = injector.getInstance(ActivationLifecycle.class);
    ServiceA serviceA = injector.getInstance(ServiceA.class);

    assertThat(serviceA.state).isEqualTo("NOT_READY");

    activationLifecycle.activate();
    assertThat(activationLifecycle.isActive(serviceA)).isTrue();
    assertThat(serviceA.state).isEqualTo("ACTIVATED");

    activationLifecycle.deactivate();
    assertThat(activationLifecycle.isActive(serviceA)).isFalse();
    assertThat(serviceA.state).isEqualTo("DEACTIVATED");
}

源代码14 项目： titus-control-plane 文件： ActivationProvisionListenerTest.java

@Test
public void testServiceReordering() throws Exception {
    LifecycleInjector injector = InjectorBuilder.fromModules(
            new ContainerEventBusModule(),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(TitusRuntime.class).toInstance(TitusRuntimes.internal());
                    bind(ActivationProvisionListenerTest.class).toInstance(ActivationProvisionListenerTest.this);
                    bind(ServiceB.class).asEagerSingleton();
                    bind(ServiceA.class).asEagerSingleton();
                }
            }).createInjector();

    ActivationLifecycle activationLifecycle = injector.getInstance(ActivationLifecycle.class);
    activationLifecycle.activate();

    assertThat(activationTrace).containsExactlyInAnyOrder(
            Pair.of("serviceA", "ACTIVATED"),
            Pair.of("serviceB", "ACTIVATED")
    );
}

源代码15 项目： titus-control-plane 文件： ActivationProvisionListenerTest.java

@Test
public void testDeactivationOrderIsReverseOfActivation() throws Exception {
    LifecycleInjector injector = InjectorBuilder.fromModules(
            new ContainerEventBusModule(),
            new AbstractModule() {
                @Override
                protected void configure() {
                    bind(TitusRuntime.class).toInstance(TitusRuntimes.internal());
                    bind(ActivationProvisionListenerTest.class).toInstance(ActivationProvisionListenerTest.this);
                    bind(ServiceB.class).asEagerSingleton();
                    bind(ServiceA.class).asEagerSingleton();
                }
            }).createInjector();

    ActivationLifecycle activationLifecycle = injector.getInstance(ActivationLifecycle.class);
    activationLifecycle.activate();
    activationLifecycle.deactivate();

    assertThat(activationTrace).hasSize(4);
    String firstActivated = activationTrace.get(0).getLeft();
    String secondActivated = activationTrace.get(1).getLeft();
    assertThat(activationTrace.subList(2, 4)).containsExactly(
            Pair.of(secondActivated, "DEACTIVATED"),
            Pair.of(firstActivated, "DEACTIVATED")
    );
}

源代码16 项目： nifi-config 文件： MainTest.java

@Test
public void mainUndeployTest() throws Exception {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(TemplateService.class).toInstance(templateServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);

    Main.main(new String[]{"-nifi", "http://localhost:8080/nifi-api", "-branch", "\"root>N2\"", "-conf", "adr", "-m", "undeploy"});
    verify(templateServiceMock).undeploy(Arrays.asList("root", "N2"));
}

源代码17 项目： nifi-config 文件： MainTest.java

@Test
public void mainHttpsUndeployTest() throws Exception {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(TemplateService.class).toInstance(templateServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);

    Main.main(new String[]{"-nifi", "https://localhost:8080/nifi-api", "-branch", "\"root>N2\"", "-m", "undeploy", "-noVerifySsl"});
    verify(templateServiceMock).undeploy(Arrays.asList("root", "N2"));
}

源代码18 项目： nifi-config 文件： MainTest.java

@Test
public void mainDeployTest() throws Exception {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(TemplateService.class).toInstance(templateServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);

    Main.main(new String[]{"-nifi", "http://localhost:8080/nifi-api", "-branch", "\"root>N2\"", "-conf", "adr", "-m", "deployTemplate"});
    verify(templateServiceMock).installOnBranch(Arrays.asList("root", "N2"), "adr", false);
}

源代码19 项目： nifi-config 文件： MainTest.java

@Test
public void mainUpdateWithPasswordTest() throws Exception {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(UpdateProcessorService.class).toInstance(updateProcessorServiceMock);
            // bind(ConnectionPortService.class).toInstance(createRouteServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);

    Main.main(new String[]{"-nifi", "http://localhost:8080/nifi-api", "-branch", "\"root>N2\"", "-conf", "adr", "-m", "updateConfig", "-user", "user", "-password", "password"});
    verify(updateProcessorServiceMock).updateByBranch(Arrays.asList("root", "N2"), "adr", false);
}

源代码20 项目： nifi-config 文件： MainTest.java

@Test
public void mainUpdateWithPasswordFromEnvTest() throws Exception {
    PowerMockito.mockStatic(System.class);
    Mockito.when(System.getenv(Main.ENV_NIFI_PASSWORD)).thenReturn("nifi_pass");

    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(UpdateProcessorService.class).toInstance(updateProcessorServiceMock);
            // bind(ConnectionPortService.class).toInstance(createRouteServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);

    Main.main(new String[]{"-nifi", "http://localhost:8080/nifi-api", "-branch", "\"root>N2\"", "-conf", "adr", "-m", "updateConfig", "-user", "user", "-password", "password"});
    verify(updateProcessorServiceMock).updateByBranch(Arrays.asList("root", "N2"), "adr", false);
    verify(accessServiceMock).addTokenOnConfiguration(false, "user", "nifi_pass");
}

源代码21 项目： nifi-config 文件： MainTest.java

@Test
public void mainExtractWithoutBranchTest() throws Exception {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(ExtractProcessorService.class).toInstance(extractProcessorServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);

    Main.main(new String[]{"-nifi", "http://localhost:8080/nifi-api", "-conf", "adr", "-m", "extractConfig", "-accessFromTicket"});
    verify(extractProcessorServiceMock).extractByBranch(Arrays.asList("root"), "adr", false);
}

源代码22 项目： nifi-config 文件： MainTest.java

@Test(expected = ConfigException.class)
public void mainExceptionTest() throws Exception {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(AccessService.class).toInstance(accessServiceMock);
            bind(InformationService.class).toInstance(informationServiceMock);
            bind(TemplateService.class).toInstance(templateServiceMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(10);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(10);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
            bind(Double.class).annotatedWith(Names.named("placeWidth")).toInstance(1200d);
            bind(PositionDTO.class).annotatedWith(Names.named("startPosition")).toInstance(new PositionDTO());
        }
    });
    //given
    PowerMockito.mockStatic(Guice.class);
    Mockito.when(Guice.createInjector((AbstractModule) anyObject())).thenReturn(injector);
    doThrow(new ApiException()).when(accessServiceMock).addTokenOnConfiguration(false, null, null);
    Main.main(new String[]{"-nifi", "http://localhost:8080/nifi-api", "-branch", "\"root>N2\"", "-conf", "adr", "-m", "undeploy"});
}

源代码23 项目： nifi-config 文件： ProcessorServiceTest.java

@Test
public void setStateAlreadyTest() {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(ProcessorsApi.class).toInstance(processorsApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    ProcessorService processorService = injector.getInstance(ProcessorService.class);
    ProcessorEntity processor = TestUtils.createProcessorEntity("id", "name");
    processor.getComponent().setState(ProcessorDTO.StateEnum.RUNNING);
    processorService.setState(processor, ProcessorDTO.StateEnum.RUNNING);
    verify(processorsApiMock, never()).updateProcessor(anyString(), anyObject());
}

源代码24 项目： nifi-config 文件： ProcessorServiceTest.java

@Test
public void setStateTest() {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(ProcessorsApi.class).toInstance(processorsApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    ProcessorService processorService = injector.getInstance(ProcessorService.class);
    ProcessorEntity processor = TestUtils.createProcessorEntity("id", "name");
    processor.getComponent().setState(ProcessorDTO.StateEnum.STOPPED);

    ProcessorEntity processorResponse = TestUtils.createProcessorEntity("id", "name");
    processorResponse.getComponent().setState(ProcessorDTO.StateEnum.RUNNING);
    when(processorsApiMock.updateProcessor(eq("id"), any() )).thenReturn(processorResponse);
    when(processorsApiMock.getProcessor(eq("id"))).thenReturn(processorResponse);

    processorService.setState(processor, ProcessorDTO.StateEnum.RUNNING);
    ArgumentCaptor<ProcessorEntity> processorEntity = ArgumentCaptor.forClass(ProcessorEntity.class);
    verify(processorsApiMock).updateProcessor(eq("id"), processorEntity.capture());
    assertEquals("id", processorEntity.getValue().getComponent().getId());
    assertEquals( ProcessorDTO.StateEnum.RUNNING, processorEntity.getValue().getComponent().getState());
}

源代码25 项目： nifi-config 文件： ProcessorServiceTest.java

@Test(expected = ConfigException.class)
public void setStateExceptionTest() {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(ProcessorsApi.class).toInstance(processorsApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    ProcessorService processorService = injector.getInstance(ProcessorService.class);
    ProcessorEntity processor = TestUtils.createProcessorEntity("id", "name");
    processor.getComponent().setState(ProcessorDTO.StateEnum.STOPPED);

    ProcessorEntity processorResponse = TestUtils.createProcessorEntity("id", "name");
    processorResponse.getComponent().setState(ProcessorDTO.StateEnum.RUNNING);
    when(processorsApiMock.updateProcessor(eq("id"), any() )).thenThrow(new ApiException());
    when(processorsApiMock.getProcessor(eq("id") )).thenReturn(processorResponse);

    processorService.setState(processor, ProcessorDTO.StateEnum.RUNNING);
}

源代码26 项目： nifi-config 文件： ConnectionServiceTest.java

@Test
public void waitEmptyQueueTest() throws ApiException, IOException, URISyntaxException {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(ConnectionsApi.class).toInstance(connectionsApiMock);
            bind(FlowfileQueuesApi.class).toInstance(flowfileQueuesApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    ConnectionService connectionService = injector.getInstance(ConnectionService.class);
    ConnectionEntity connection = TestUtils.createConnectionEntity("id","sourceId","destinationId");
    connection.setStatus(new ConnectionStatusDTO());
    connection.getStatus().setAggregateSnapshot(new ConnectionStatusSnapshotDTO());
    connection.getStatus().getAggregateSnapshot().setQueuedCount("0");
    connection.getStatus().getAggregateSnapshot().setQueuedSize("0");
    when(connectionsApiMock.getConnection("id")).thenReturn(connection);
    connectionService.waitEmptyQueue(connection);
}

源代码27 项目： nifi-config 文件： ConnectionServiceTest.java

@Test(expected = TimeoutException.class)
public void waitEmptyQueueTimeOutTest() throws ApiException, IOException, URISyntaxException {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(ConnectionsApi.class).toInstance(connectionsApiMock);
            bind(FlowfileQueuesApi.class).toInstance(flowfileQueuesApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    ConnectionService connectionService = injector.getInstance(ConnectionService.class);
    ConnectionEntity connection = TestUtils.createConnectionEntity("id","sourceId","destinationId");
    connection.setStatus(new ConnectionStatusDTO());
    connection.getStatus().setAggregateSnapshot(new ConnectionStatusSnapshotDTO());
    connection.getStatus().getAggregateSnapshot().setQueuedCount("1");
    connection.getStatus().getAggregateSnapshot().setQueuedSize("0");
    when(connectionsApiMock.getConnection("id")).thenReturn(connection);
    connectionService.waitEmptyQueue(connection);
}

源代码28 项目： nifi-config 文件： PortServiceTest.java

/**
 * Creates a token for accessing the REST API via username/password
 * <p>
 * The token returned is formatted as a JSON Web Token (JWT). The token is base64 encoded and comprised of three parts. The header, the body, and the signature. The expiration of the token is a contained within the body. The token can be used in the Authorization header in the format &#39;Authorization: Bearer &lt;token&gt;&#39;.
 *
 * @throws ApiException if the Api call fails
 */
@Test
public void getByIdInputTest() throws ApiException, IOException, URISyntaxException {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(InputPortsApi.class).toInstance(inputPortsApiMock);
            bind(OutputPortsApi.class).toInstance(outputPortsApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    PortService portService = injector.getInstance(PortService.class);
    PortEntity port = new PortEntity();
    port.setComponent(new PortDTO());
    port.getComponent().setId("id");
    when(inputPortsApiMock.getInputPort("id")).thenReturn(port);
    PortEntity portResult = portService.getById("id", PortDTO.TypeEnum.INPUT_PORT);
    assertEquals("id", portResult.getComponent().getId());
}

源代码29 项目： nifi-config 文件： PortServiceTest.java

@Test
public void getByIdOutputTest() throws ApiException, IOException, URISyntaxException {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(InputPortsApi.class).toInstance(inputPortsApiMock);
            bind(OutputPortsApi.class).toInstance(outputPortsApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    PortService portService = injector.getInstance(PortService.class);
    PortEntity port = new PortEntity();
    port.setComponent(new PortDTO());
    port.getComponent().setId("id");
    when(outputPortsApiMock.getOutputPort("id")).thenReturn(port);
    PortEntity portResult = portService.getById("id", PortDTO.TypeEnum.OUTPUT_PORT);
    assertEquals("id", portResult.getComponent().getId());
}

源代码30 项目： nifi-config 文件： PortServiceTest.java

@Test
public void setStateTest() throws ApiException, IOException, URISyntaxException {
    Injector injector = Guice.createInjector(new AbstractModule() {
        protected void configure() {
            bind(InputPortsApi.class).toInstance(inputPortsApiMock);
            bind(OutputPortsApi.class).toInstance(outputPortsApiMock);
            bind(Integer.class).annotatedWith(Names.named("timeout")).toInstance(1);
            bind(Integer.class).annotatedWith(Names.named("interval")).toInstance(1);
            bind(Boolean.class).annotatedWith(Names.named("forceMode")).toInstance(false);
        }
    });
    PortService portService = injector.getInstance(PortService.class);
    PortEntity port = new PortEntity();
    port.setComponent(new PortDTO());
    port.getComponent().setId("id");
    port.getComponent().setState(PortDTO.StateEnum.STOPPED);
    port.setStatus(new PortStatusDTO());
    port.getStatus().setRunStatus("Stopped");
    portService.setState(port, PortDTO.StateEnum.STOPPED);
    verify(inputPortsApiMock,never()).updateInputPort(eq("id"), any());
    verify(outputPortsApiMock,never()).updateOutputPort(eq("id"), any());
}

类所在包

com.google.inject

类方法

同包方法

com.google.inject.Inject

com.google.inject.Injector

com.google.inject.AbstractModule

com.google.inject.Guice

com.google.inject.Singleton

com.google.inject.Module

com.google.inject.Provides

com.google.inject.Provider