/** * Called by our {@link DependencyTrackerImpl} when all dependencies are satisfied. */ public void componentStarted(BundleContext context) throws Exception { final File bundleDataArea = context.getDataFile(""); m_agentContext = new AgentContextImpl(bundleDataArea); m_agentContext.setHandler(LoggingHandler.class, new LoggingHandlerImpl(context)); m_agentContext.setHandler(ConfigurationHandler.class, new ConfigurationHandlerImpl(context)); m_agentContext.setHandler(EventsHandler.class, new EventsHandlerImpl(context)); m_agentContext.setHandler(ScheduledExecutorService.class, m_executorService); m_agentContext.setHandler(DownloadHandler.class, new DownloadHandlerImpl(bundleDataArea)); m_agentContext.setHandler(DeploymentHandler.class, new DeploymentHandlerImpl(context, m_packageAdmin)); m_agentContext.setHandler(AgentUpdateHandler.class, new AgentUpdateHandlerImpl(context)); m_agentContext.setHandler(FeedbackHandler.class, new FeedbackHandlerImpl()); IdentificationHandler identificationHandler = (m_identificationHandler != null) ? m_identificationHandler : new IdentificationHandlerImpl(); m_agentContext.setHandler(IdentificationHandler.class, identificationHandler); DiscoveryHandler discoveryHandler = (m_discoveryHandler != null) ? m_discoveryHandler : new DiscoveryHandlerImpl(); m_agentContext.setHandler(DiscoveryHandler.class, discoveryHandler); ConnectionHandler connectionHandler = (m_connectionHandler != null) ? m_connectionHandler : new ConnectionHandlerImpl(); m_agentContext.setHandler(ConnectionHandler.class, connectionHandler); m_agentContext.addComponent(new EventLoggerImpl(context)); // Lastly, inject the (custom) controller for this agent... m_agentContext.setController(m_controller); m_agentContext.start(); m_agentControlRegistration = context.registerService(AgentControl.class.getName(), new AgentControlImpl(m_agentContext), null); }
protected <T> void setMockedHandler(AgentContextImpl context, Class<T> clazz) { context.setHandler(clazz, addTestMock(clazz)); }
@BeforeClass(alwaysRun = true) public void setUpOnceAgain() throws Exception { m_agentContext = mockAgentContext(); BundleContext bc = mockBundleContext(); m_eventsHandler = new EventsHandlerImpl(bc); m_agentContext.setHandler(EventsHandler.class, m_eventsHandler); m_agentContext.setHandler(ConfigurationHandler.class, new ConfigurationHandlerImpl(bc)); m_agentContext.setHandler(FeedbackHandler.class, new TestFeedbackHandler()); replayTestMocks(); m_agentContext.start(); }
@BeforeClass public void setUpAgain() throws Exception { m_agentContextImpl = mockAgentContext(); replayTestMocks(); BundleContext bc = mockBundleContext(); m_agentContextImpl.setHandler(FeedbackHandler.class, new FeedbackHandlerImpl()); m_agentContextImpl.setHandler(EventsHandler.class, new EventsHandlerImpl(bc)); m_agentContextImpl.setHandler(ConfigurationHandler.class, new ConfigurationHandlerImpl(bc)); m_agentContextImpl.start(); }
@BeforeClass public void setUpOnceAgain() throws Exception { m_webServer = new TestWebServer(PORT, "/", "generated"); m_webServer.start(); m_availableURL1 = new URL("http://localhost:" + PORT); m_availableURL2 = new URL("http://127.0.0.1:" + PORT); m_unavailableURL1 = new URL("http://localhost:19999"); m_unavailableURL2 = new URL("http://127.0.0.1:19999"); BundleContext bc = mockBundleContext(); m_agentContextImpl = mockAgentContext(); m_agentContext = m_agentContextImpl; // Make sure the default server URL is not reachable, as used for this test... m_agentContextImpl.setHandler(DiscoveryHandler.class, new DiscoveryHandlerImpl(m_unavailableURL1.toExternalForm(), true)); m_agentContextImpl.setHandler(EventsHandler.class, new EventsHandlerImpl(bc)); m_agentContextImpl.setHandler(ConfigurationHandler.class, new ConfigurationHandlerImpl(bc)); m_agentContextImpl.setHandler(ConnectionHandler.class, new ConnectionHandlerImpl()); replayTestMocks(); m_agentContextImpl.start(); }
@BeforeClass public void setUp() throws Exception { m_agentContext = mockAgentContext(); m_agentContext.setHandler(DownloadHandler.class, new DownloadHandlerImpl(m_agentContext.getWorkDir())); m_agentContext.setHandler(ConnectionHandler.class, new ConnectionHandlerImpl()); replayTestMocks(); m_agentContext.start(); }
private void resetConfigurationHandler(ConfigurationHandler configurationHandler) throws Exception { ConfigurationHandler oldConfigurationHandler = m_agentContextImpl.getHandler(ConfigurationHandler.class); if (oldConfigurationHandler instanceof ComponentBase) { ((ComponentBase) oldConfigurationHandler).stop(); } m_agentContextImpl.setHandler(ConfigurationHandler.class, configurationHandler); m_agentContextImpl.start(); } }
@BeforeClass public void setUpAgain() throws Exception { m_agentContextImpl = mockAgentContext(); m_agentContext = m_agentContextImpl; m_agentContextImpl.setHandler(IdentificationHandler.class, new IdentificationHandlerImpl()); m_agentContextImpl.start(); replayTestMocks(); }
@BeforeClass public void setUpOnceAgain() throws Exception { File dataLocation = new File("generated"); m_200file = File.createTempFile("download", ".bin", dataLocation); m_200file.deleteOnExit(); int port = 8883; m_200url = new URL("http://localhost:" + port + "/" + m_200file.getName()); m_404url = new URL("http://localhost:" + port + "/error?status=404"); m_503url = new URL("http://localhost:" + port + "/error?status=503&retry=500"); DigestOutputStream dos = new DigestOutputStream(new FileOutputStream(m_200file), MessageDigest.getInstance("MD5")); for (int i = 0; i < 10000; i++) { dos.write(String.valueOf(System.currentTimeMillis()).getBytes()); dos.write(" Lorum Ipsum Lorum Ipsum Lorum Ipsum Lorum Ipsum Lorum Ipsum\n".getBytes()); } dos.close(); m_200digest = new String(dos.getMessageDigest().digest()); m_webServer = new TestWebServer(port, "/", dataLocation.getName()); m_webServer.addServlet(new TestErrorServlet(), "/error"); m_webServer.start(); m_agentContextImpl = mockAgentContext(); m_agentContext = m_agentContextImpl; BundleContext bc = mockBundleContext(); m_agentContextImpl.setHandler(EventsHandler.class, new EventsHandlerImpl(bc)); m_agentContextImpl.setHandler(ConnectionHandler.class, new ConnectionHandlerImpl()); m_agentContextImpl.setHandler(LoggingHandler.class, new LoggingHandlerImpl(bc, Levels.DEBUG)); m_agentContextImpl.setHandler(DownloadHandler.class, new DownloadHandlerImpl(dataLocation)); m_agentContextImpl.start(); replayTestMocks(); }
@BeforeClass public void setUpOnceAgain() throws Exception { m_basicAuthURL = new URL("http://localhost:" + PORT + "/basicauth"); m_webServer = new TestWebServer(PORT, "/", "generated"); m_webServer.addServlet(new BasicAuthServlet(USERNAME, PASSWORD), "/basicauth/*"); m_webServer.start(); BundleContext bc = mockBundleContext(); m_agentContext = mockAgentContext(); m_agentContext.setHandler(EventsHandler.class, new EventsHandlerImpl(bc)); m_agentContext.setHandler(ConfigurationHandler.class, new ConfigurationHandlerImpl(bc)); m_agentContext.setHandler(ConnectionHandler.class, new ConnectionHandlerImpl()); replayTestMocks(); m_agentContext.start(); }
protected AgentContextImpl mockAgentContext(String subDir) throws Exception { if (m_contextDir != null) { cleanDir(m_contextDir); m_contextDir.delete(); } m_contextDir = new File(getWorkDir(), subDir); m_contextDir.mkdirs(); cleanDir(m_contextDir); AgentContextImpl context = new AgentContextImpl(m_contextDir); for (Class<?> handlerClass : AgentContextImpl.KNOWN_HANDLERS) { if (ScheduledExecutorService.class.equals(handlerClass)) { // always inject a proper executor service that simply invokes synchronously... context.setHandler(ScheduledExecutorService.class, new SynchronousExecutorService()); } else { setMockedHandler(context, handlerClass); } } return context; }
@BeforeMethod public void setUpAgain(Method method) throws Exception { // this setup is needed because a real Feedbackstore is initialized in the constructor. m_agentContext = mockAgentContext(); URL serverURL = new URL("http://localhost:" + PORT + "/"); m_webServer = new TestWebServer(PORT, "/", "generated"); m_webServer.start(); DiscoveryHandler discoveryHandler = addTestMock(DiscoveryHandler.class); expect(discoveryHandler.getServerUrl()).andReturn(serverURL).anyTimes(); IdentificationHandler identificationHandler = addTestMock(IdentificationHandler.class); expect(identificationHandler.getAgentId()).andReturn("identification").anyTimes(); replayTestMocks(); m_agentContext.setHandler(DiscoveryHandler.class, discoveryHandler); m_agentContext.setHandler(IdentificationHandler.class, identificationHandler); m_agentContext.setHandler(ConnectionHandler.class, new ConnectionHandlerImpl()); m_feedbackChannelImpl = new FeedbackChannelImpl(m_agentContext, "test"); m_agentContext.start(); }
@BeforeMethod public void setUpAgain() throws Exception { m_dummyInputStream = new FileInputStream(m_dummyFile); DownloadResult downloadResult = addTestMock(DownloadResult.class); expect(downloadResult.isComplete()).andReturn(true).anyTimes(); expect(downloadResult.getInputStream()).andReturn(m_dummyInputStream).anyTimes(); Future<DownloadResult> future = addTestMock(Future.class); expect(future.get()).andReturn(downloadResult).anyTimes(); DownloadHandle downloadHandle = addTestMock(DownloadHandle.class); expect(downloadHandle.start(isNull(DownloadProgressListener.class))).andReturn(future).anyTimes(); DeploymentHandler deploymentHandler = addTestMock(DeploymentHandler.class); expect(deploymentHandler.getInstalledVersion()).andReturn(m_version2).anyTimes(); expect(deploymentHandler.getAvailableVersions()).andReturn(m_availableVersions).anyTimes(); expect(deploymentHandler.getDownloadHandle(eq(m_version3), eq(true))).andReturn(downloadHandle).once(); deploymentHandler.install(notNull(InputStream.class)); expectLastCall().once(); m_agentContext = mockAgentContext(); m_agentContext.setHandler(DeploymentHandler.class, deploymentHandler); replayTestMocks(); m_agentContext.start(); m_agentControl = new AgentControlImpl(m_agentContext); }
m_agentContext.setHandler(IdentificationHandler.class, identificationHandler); m_agentContext.setHandler(DiscoveryHandler.class, discoveryHandler); m_agentContext.setHandler(ConfigurationHandler.class, configurationHandler); m_agentContext.setHandler(ConnectionHandler.class, new ConnectionHandlerImpl()); m_agentContext.setHandler(DeploymentHandler.class, new DeploymentHandlerImpl(deploymentAdmin)); replayTestMocks(); m_agentContext.start();