use of io.fabric8.insight.metrics.model.Server in project fabric8 by jboss-fuse.
the class DataStoreBootstrapTemplate method doWith.
@Override
public void doWith(ProfileRegistry profileRegistry, DataStore dataStore) throws Exception {
String versionId = options.getVersion();
int minimumPort = options.getMinimumPort();
int maximumPort = options.getMaximumPort();
String zooKeeperServerHost = options.getBindAddress();
int zooKeeperServerPort = options.getZooKeeperServerPort();
int zooKeeperServerConnectionPort = options.getZooKeeperServerConnectionPort();
int mappedPort = Ports.mapPortToRange(zooKeeperServerPort, minimumPort, maximumPort);
CuratorFramework curator = null;
try {
curator = createCuratorFramework(connectionUrl, options);
curator.start();
curator.getZookeeperClient().blockUntilConnectedOrTimedOut();
LockHandle writeLock = profileRegistry.aquireWriteLock();
try {
// Make the import path absolute
File importPath = new File(options.getImportPath());
if (!importPath.isAbsolute()) {
importPath = new File(homeDir, options.getImportPath());
}
// Import data into the DataStore
if (options.isAutoImportEnabled()) {
if (importPath.isDirectory()) {
profileRegistry.importFromFileSystem(importPath.getAbsolutePath());
} else {
LOGGER.warn("Profile import dir does not exist: {}", importPath);
}
}
// set the fabric configuration
ZooKeeperUtils.setData(curator, ZkPath.CONFIG_DEFAULT_VERSION.getPath(), versionId);
// Default JAAS config
Map<String, String> jaasConfig = Collections.singletonMap("encryption.enabled", "${zk:/fabric/authentication/encryption.enabled}");
// Default Zookeeper config
Properties zkProps = new Properties();
zkProps.setProperty("zookeeper.url", "${zk:" + ZkPath.CONFIG_ENSEMBLE_URL.getPath() + "}");
zkProps.setProperty("zookeeper.password", "${zk:" + ZkPath.CONFIG_ENSEMBLE_PASSWORD.getPath() + "}");
// Create or update default profile
Profile defaultProfile = profileRegistry.getProfile(versionId, "default");
if (defaultProfile == null) {
ProfileBuilder prfBuilder = ProfileBuilder.Factory.create(versionId, "default");
prfBuilder.addConfiguration("io.fabric8.jaas", jaasConfig);
prfBuilder.addFileConfiguration("io.fabric8.zookeeper.properties", DataStoreUtils.toBytes(zkProps));
Profile profile = prfBuilder.getProfile();
if (profileRegistry.hasVersion(versionId)) {
profileRegistry.createProfile(profile);
} else {
VersionBuilder verBuilder = VersionBuilder.Factory.create(versionId);
Version version = verBuilder.addProfile(profile).getVersion();
profileRegistry.createVersion(version);
}
} else {
ProfileBuilder builder = ProfileBuilder.Factory.createFrom(defaultProfile);
// be careful not to mess with existing *.properties files - for better "diffability"
// during git-based patching
Map<String, String> existingJaasConfig = builder.getConfiguration("io.fabric8.jaas");
Map<String, String> existingZkConfig = builder.getConfiguration("io.fabric8.zookeeper");
existingJaasConfig.putAll(jaasConfig);
existingZkConfig.put("zookeeper.url", zkProps.getProperty("zookeeper.url"));
existingZkConfig.put("zookeeper.password", zkProps.getProperty("zookeeper.password"));
builder.addConfiguration("io.fabric8.jaas", existingJaasConfig);
builder.addConfiguration("io.fabric8.zookeeper", existingZkConfig);
profileRegistry.updateProfile(builder.getProfile());
}
ZooKeeperUtils.setData(curator, ZkPath.CONFIG_ENSEMBLE_URL.getPath(), "${zk:" + name + "/ip}:" + zooKeeperServerConnectionPort);
ZooKeeperUtils.setData(curator, ZkPath.CONFIG_ENSEMBLE_PASSWORD.getPath(), PasswordEncoder.encode(options.getZookeeperPassword()));
// Ensemble properties
Properties ensembleProps = new Properties();
ensembleProps.put("tickTime", String.valueOf(options.getZooKeeperServerTickTime()));
ensembleProps.put("initLimit", String.valueOf(options.getZooKeeperServerInitLimit()));
ensembleProps.put("syncLimit", String.valueOf(options.getZooKeeperServerSyncLimit()));
ensembleProps.put("snapRetainCount", String.valueOf(options.getZookeeperSnapRetainCount()));
ensembleProps.put("purgeInterval", String.valueOf(options.getZookeeperPurgeInterval()));
ensembleProps.put("dataDir", options.getZooKeeperServerDataDir() + File.separator + "0000");
loadPropertiesFrom(ensembleProps, importPath + "/fabric/profiles/default.profile/io.fabric8.zookeeper.server.properties");
// Create ensemble profile
String profileId = "fabric-ensemble-0000";
IllegalStateAssertion.assertFalse(profileRegistry.hasProfile(versionId, profileId), "Profile already exists: " + versionId + "/" + profileId);
ProfileBuilder ensembleProfileBuilder = ProfileBuilder.Factory.create(versionId, profileId);
ensembleProfileBuilder.addAttribute(Profile.ABSTRACT, "true").addAttribute(Profile.HIDDEN, "true");
ensembleProfileBuilder.addFileConfiguration("io.fabric8.zookeeper.server-0000.properties", DataStoreUtils.toBytes(ensembleProps));
String ensembleProfileId = profileRegistry.createProfile(ensembleProfileBuilder.getProfile());
// Ensemble server properties
Properties serverProps = new Properties();
serverProps.put("clientPort", String.valueOf(mappedPort));
serverProps.put("clientPortAddress", zooKeeperServerHost);
// Create ensemble server profile
profileId = "fabric-ensemble-0000-1";
IllegalStateAssertion.assertFalse(profileRegistry.hasProfile(versionId, profileId), "Profile already exists: " + versionId + "/" + profileId);
ProfileBuilder serverProfileBuilder = ProfileBuilder.Factory.create(versionId, profileId);
serverProfileBuilder.addAttribute(Profile.HIDDEN, "true").addAttribute(Profile.PARENTS, ensembleProfileId);
serverProfileBuilder.addFileConfiguration("io.fabric8.zookeeper.server-0000.properties", DataStoreUtils.toBytes(serverProps));
profileRegistry.createProfile(serverProfileBuilder.getProfile());
ZooKeeperUtils.setData(curator, ZkPath.CONFIG_ENSEMBLES.getPath(), "0000");
ZooKeeperUtils.setData(curator, ZkPath.CONFIG_ENSEMBLE.getPath("0000"), name);
// configure fabric profile
Profile fabricProfile = profileRegistry.getProfile(versionId, "fabric");
if (fabricProfile == null) {
ProfileBuilder builder = ProfileBuilder.Factory.create(versionId, "fabric");
Properties agentProps = new Properties();
agentProps.put("feature.fabric-commands", "fabric-commands");
builder.addFileConfiguration("io.fabric8.agent.properties", DataStoreUtils.toBytes(agentProps));
String createdId = profileRegistry.createProfile(builder.getProfile());
fabricProfile = profileRegistry.getRequiredProfile(versionId, createdId);
} else {
ProfileBuilder builder = ProfileBuilder.Factory.createFrom(fabricProfile);
Map<String, String> agentProps = builder.getConfiguration(Constants.AGENT_PID);
agentProps.put("feature.fabric-commands", "fabric-commands");
builder.addConfiguration(Constants.AGENT_PID, agentProps);
String updatedId = profileRegistry.updateProfile(builder.getProfile());
fabricProfile = profileRegistry.getRequiredProfile(versionId, updatedId);
}
} finally {
writeLock.unlock();
}
ZooKeeperUtils.createDefault(curator, ZkPath.CONFIG_CONTAINER.getPath(name), versionId);
StringBuilder profilesBuilder = new StringBuilder();
Set<String> profiles = options.getProfiles();
profilesBuilder.append("fabric").append(" ").append("fabric-ensemble-0000-1");
for (String p : profiles) {
if (!(p.equals("fabric") || p.equals("fabric-ensemble-0000-1"))) {
profilesBuilder.append(" ").append(p);
}
}
if (!options.isAgentEnabled()) {
profilesBuilder.append(" ").append("unmanaged");
}
ZooKeeperUtils.createDefault(curator, ZkPath.CONFIG_VERSIONS_CONTAINER.getPath(versionId, name), profilesBuilder.toString());
// outside of the profile storage area, so we'll keep these in zk
EncryptionSupport encryption = addUsersToZookeeper(curator, options.getUsers());
ZooKeeperUtils.createDefault(curator, "/fabric/authentication/encryption.enabled", Boolean.valueOf(encryption != null).toString());
ZooKeeperUtils.createDefault(curator, "/fabric/authentication/domain", "karaf");
ZooKeeperUtils.createDefault(curator, ZkPath.AUTHENTICATION_CRYPT_ALGORITHM.getPath(), "PBEWithMD5AndDES");
ZooKeeperUtils.createDefault(curator, ZkPath.AUTHENTICATION_CRYPT_PASSWORD.getPath(), PasswordEncoder.encode(options.getZookeeperPassword()));
// Ensure ACLs are from the beggining of the fabric tree.
aclManager.fixAcl(curator, "/fabric", true);
} finally {
if (curator != null) {
curator.close();
}
}
}
use of io.fabric8.insight.metrics.model.Server in project fabric8 by jboss-fuse.
the class MavenProxyServletSupportTest method testUpload.
private Map<String, String> testUpload(String path, final byte[] contents, String location, String profile, String version, boolean hasLocationHeader) throws Exception {
final String old = System.getProperty("karaf.data");
System.setProperty("karaf.data", new File("target").getCanonicalPath());
FileUtils.deleteDirectory(new File("target/tmp"));
Server server = new Server(0);
server.setHandler(new AbstractHandler() {
@Override
public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
response.setStatus(HttpServletResponse.SC_NO_CONTENT);
}
});
server.start();
try {
int localPort = ((ServerConnector) server.getConnectors()[0]).getLocalPort();
List<String> remoteRepos = Arrays.asList("http://relevant.not/maven2@id=central");
RuntimeProperties props = new MockRuntimeProperties();
MavenResolver resolver = createResolver("target/tmp", remoteRepos, "http", "localhost", localPort, "fuse", "fuse", null);
MavenUploadProxyServlet servlet = new MavenUploadProxyServlet(resolver, props, projectDeployer, new File("target/upload"), 0);
HttpServletRequest request = EasyMock.createMock(HttpServletRequest.class);
EasyMock.expect(request.getPathInfo()).andReturn(path);
EasyMock.expect(request.getContentType()).andReturn("text/plain").anyTimes();
EasyMock.expect(request.getInputStream()).andReturn(new ServletInputStream() {
private int i;
@Override
public int read() throws IOException {
if (i >= contents.length) {
return -1;
}
return (contents[i++] & 0xFF);
}
@Override
public boolean isReady() {
return false;
}
@Override
public boolean isFinished() {
return false;
}
@Override
public void setReadListener(ReadListener readListener) {
}
});
EasyMock.expect(request.getHeader("X-Location")).andReturn(location);
EasyMock.expect(request.getParameter("profile")).andReturn(profile);
EasyMock.expect(request.getParameter("version")).andReturn(version);
final Map<String, String> headers = new HashMap<>();
HttpServletResponse rm = EasyMock.createMock(HttpServletResponse.class);
HttpServletResponse response = new HttpServletResponseWrapper(rm) {
@Override
public void addHeader(String name, String value) {
headers.put(name, value);
}
};
response.setStatus(EasyMock.anyInt());
EasyMock.expectLastCall().anyTimes();
response.setContentLength(EasyMock.anyInt());
EasyMock.expectLastCall().anyTimes();
response.setContentType((String) EasyMock.anyObject());
EasyMock.expectLastCall().anyTimes();
response.setDateHeader((String) EasyMock.anyObject(), EasyMock.anyLong());
EasyMock.expectLastCall().anyTimes();
response.setHeader((String) EasyMock.anyObject(), (String) EasyMock.anyObject());
EasyMock.expectLastCall().anyTimes();
EasyMock.replay(request, rm);
servlet.start();
servlet.doPut(request, response);
EasyMock.verify(request, rm);
Assert.assertEquals(hasLocationHeader, headers.containsKey("X-Location"));
return headers;
} finally {
server.stop();
if (old != null) {
System.setProperty("karaf.data", old);
}
}
}
use of io.fabric8.insight.metrics.model.Server in project fabric8 by jboss-fuse.
the class MavenProxyServletSupportTest method testDownloadMetadata.
@Test(timeout = 30000)
public void testDownloadMetadata() throws Exception {
final String old = System.getProperty("karaf.data");
System.setProperty("karaf.data", new File("target").getCanonicalPath());
FileUtils.deleteDirectory(new File("target/tmp"));
Server server = new Server(0);
server.setHandler(new AbstractHandler() {
@Override
public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
String result = null;
if ("/repo1/org/apache/camel/camel-core/maven-metadata.xml".equals(target)) {
result = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + "<metadata>\n" + " <groupId>org.apache.camel</groupId>\n" + " <artifactId>camel-core</artifactId>\n" + " <versioning>\n" + " <latest>2.14.0</latest>\n" + " <release>2.14.0</release>\n" + " <versions>\n" + " <version>1.6.1</version>\n" + " <version>1.6.2</version>\n" + " <version>1.6.3</version>\n" + " <version>1.6.4</version>\n" + " <version>2.0-M2</version>\n" + " <version>2.0-M3</version>\n" + " <version>2.0.0</version>\n" + " <version>2.1.0</version>\n" + " <version>2.2.0</version>\n" + " <version>2.3.0</version>\n" + " <version>2.4.0</version>\n" + " <version>2.5.0</version>\n" + " <version>2.6.0</version>\n" + " <version>2.7.0</version>\n" + " <version>2.7.1</version>\n" + " <version>2.7.2</version>\n" + " <version>2.7.3</version>\n" + " <version>2.7.4</version>\n" + " <version>2.7.5</version>\n" + " <version>2.8.0</version>\n" + " <version>2.8.1</version>\n" + " <version>2.8.2</version>\n" + " <version>2.8.3</version>\n" + " <version>2.8.4</version>\n" + " <version>2.8.5</version>\n" + " <version>2.8.6</version>\n" + " <version>2.9.0-RC1</version>\n" + " <version>2.9.0</version>\n" + " <version>2.9.1</version>\n" + " <version>2.9.2</version>\n" + " <version>2.9.3</version>\n" + " <version>2.9.4</version>\n" + " <version>2.9.5</version>\n" + " <version>2.9.6</version>\n" + " <version>2.9.7</version>\n" + " <version>2.9.8</version>\n" + " <version>2.10.0</version>\n" + " <version>2.10.1</version>\n" + " <version>2.10.2</version>\n" + " <version>2.10.3</version>\n" + " <version>2.10.4</version>\n" + " <version>2.10.5</version>\n" + " <version>2.10.6</version>\n" + " <version>2.10.7</version>\n" + " <version>2.11.0</version>\n" + " <version>2.11.1</version>\n" + " <version>2.11.2</version>\n" + " <version>2.11.3</version>\n" + " <version>2.11.4</version>\n" + " <version>2.12.0</version>\n" + " <version>2.12.1</version>\n" + " <version>2.12.2</version>\n" + " <version>2.12.3</version>\n" + " <version>2.12.4</version>\n" + " <version>2.13.0</version>\n" + " <version>2.13.1</version>\n" + " <version>2.13.2</version>\n" + " <version>2.14.0</version>\n" + " </versions>\n" + " <lastUpdated>20140918132816</lastUpdated>\n" + " </versioning>\n" + "</metadata>\n" + "\n";
} else if ("/repo2/org/apache/camel/camel-core/maven-metadata.xml".equals(target)) {
result = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + "<metadata modelVersion=\"1.1.0\">\n" + " <groupId>org.apache.camel</groupId>\n" + " <artifactId>camel-core</artifactId>\n" + " <versioning>\n" + " <latest>2.14.0.redhat-620034</latest>\n" + " <release>2.14.0.redhat-620034</release>\n" + " <versions>\n" + " <version>2.10.0.redhat-60074</version>\n" + " <version>2.12.0.redhat-610312</version>\n" + " <version>2.12.0.redhat-610328</version>\n" + " <version>2.12.0.redhat-610355</version>\n" + " <version>2.12.0.redhat-610378</version>\n" + " <version>2.12.0.redhat-610396</version>\n" + " <version>2.12.0.redhat-610399</version>\n" + " <version>2.12.0.redhat-610401</version>\n" + " <version>2.12.0.redhat-610402</version>\n" + " <version>2.12.0.redhat-611403</version>\n" + " <version>2.12.0.redhat-611405</version>\n" + " <version>2.12.0.redhat-611406</version>\n" + " <version>2.12.0.redhat-611408</version>\n" + " <version>2.12.0.redhat-611409</version>\n" + " <version>2.12.0.redhat-611410</version>\n" + " <version>2.12.0.redhat-611411</version>\n" + " <version>2.12.0.redhat-611412</version>\n" + " <version>2.14.0.redhat-620031</version>\n" + " <version>2.14.0.redhat-620033</version>\n" + " <version>2.14.0.redhat-620034</version>\n" + " </versions>\n" + " <lastUpdated>20141019130841</lastUpdated>\n" + " </versioning>\n" + "</metadata>\n" + "\n";
}
if (result == null) {
response.setStatus(HttpServletResponse.SC_NOT_FOUND);
baseRequest.setHandled(true);
response.getOutputStream().close();
} else {
response.setStatus(HttpServletResponse.SC_OK);
baseRequest.setHandled(true);
response.getOutputStream().write(result.getBytes());
response.getOutputStream().close();
}
}
});
server.start();
try {
int localPort = ((ServerConnector) server.getConnectors()[0]).getLocalPort();
List<String> remoteRepos = Arrays.asList("http://relevant.not/repo1@id=repo1,http://relevant.not/repo2@id=repo2");
RuntimeProperties props = new MockRuntimeProperties();
// TODO: local repo should point to target/tmp
MavenResolver resolver = createResolver("target/tmp", remoteRepos, "http", "localhost", localPort, "fuse", "fuse", null);
MavenDownloadProxyServlet servlet = new MavenDownloadProxyServlet(resolver, props, projectDeployer, 5, 0);
AsyncContext context = EasyMock.createMock(AsyncContext.class);
EasyMock.makeThreadSafe(context, true);
final Map<String, Object> attributes = new HashMap<>();
HttpServletRequest request = EasyMock.createMock(HttpServletRequest.class);
HttpServletRequest requestWrapper = new HttpServletRequestWrapper(request) {
@Override
public Object getAttribute(String name) {
return attributes.get(name);
}
@Override
public void setAttribute(String name, Object o) {
attributes.put(name, o);
}
};
EasyMock.makeThreadSafe(request, true);
EasyMock.expect(request.getMethod()).andReturn("GET");
EasyMock.expect(request.getPathInfo()).andReturn("org/apache/camel/camel-core/maven-metadata.xml");
EasyMock.expect(request.startAsync()).andReturn(context);
context.setTimeout(EasyMock.anyInt());
EasyMock.expectLastCall();
context.addListener((AsyncListener) EasyMock.anyObject());
EasyMock.expectLastCall();
HttpServletResponse response = EasyMock.createMock(HttpServletResponse.class);
EasyMock.makeThreadSafe(response, true);
response.setStatus(EasyMock.anyInt());
EasyMock.expectLastCall().anyTimes();
response.setContentLength(EasyMock.anyInt());
EasyMock.expectLastCall().anyTimes();
response.setContentType((String) EasyMock.anyObject());
EasyMock.expectLastCall().anyTimes();
response.setDateHeader((String) EasyMock.anyObject(), EasyMock.anyLong());
EasyMock.expectLastCall().anyTimes();
response.setHeader((String) EasyMock.anyObject(), (String) EasyMock.anyObject());
EasyMock.expectLastCall().anyTimes();
response.flushBuffer();
EasyMock.expectLastCall().anyTimes();
final CountDownLatch latchDispatch = new CountDownLatch(1);
context.dispatch();
EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {
@Override
public Object answer() throws Throwable {
latchDispatch.countDown();
return null;
}
});
EasyMock.replay(request, response, context);
servlet.start();
servlet.doGet(requestWrapper, response);
latchDispatch.await();
EasyMock.verify(request, response, context);
EasyMock.reset(request, response, context);
EasyMock.expect(request.getPathInfo()).andReturn("org/apache/camel/camel-core/maven-metadata.xml");
EasyMock.expect(request.startAsync()).andReturn(context);
context.setTimeout(EasyMock.anyInt());
EasyMock.expectLastCall();
context.addListener((AsyncListener) EasyMock.anyObject());
EasyMock.expectLastCall();
final ByteArrayOutputStream baos = new ByteArrayOutputStream();
EasyMock.expect(response.getOutputStream()).andReturn(new ServletOutputStream() {
@Override
public void write(int b) throws IOException {
baos.write(b);
}
@Override
public void write(byte[] b, int off, int len) throws IOException {
baos.write(b, off, len);
}
@Override
public boolean isReady() {
return true;
}
@Override
public void setWriteListener(WriteListener writeListener) {
}
}).anyTimes();
response.flushBuffer();
EasyMock.expectLastCall().anyTimes();
final CountDownLatch latchComplete = new CountDownLatch(1);
context.complete();
EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {
@Override
public Object answer() throws Throwable {
latchComplete.countDown();
return null;
}
});
EasyMock.replay(request, response, context);
servlet.doGet(requestWrapper, response);
EasyMock.verify(request, response, context);
org.apache.maven.artifact.repository.metadata.Metadata m = new MetadataXpp3Reader().read(new ByteArrayInputStream(baos.toByteArray()), false);
assertEquals("2.14.0.redhat-620034", m.getVersioning().getLatest());
assertTrue(m.getVersioning().getVersions().contains("2.10.4"));
assertTrue(m.getVersioning().getVersions().contains("2.12.0.redhat-610399"));
EasyMock.verify(request, response, context);
} finally {
server.stop();
if (old != null) {
System.setProperty("karaf.data", old);
}
}
}
use of io.fabric8.insight.metrics.model.Server in project fabric8 by jboss-fuse.
the class MavenProxyServletSupportTest method testDownload.
private void testDownload(Handler serverHandler) throws Exception {
final String old = System.getProperty("karaf.data");
System.setProperty("karaf.data", new File("target").getCanonicalPath());
FileUtils.deleteDirectory(new File("target/tmp"));
Server server = new Server(0);
server.setHandler(serverHandler);
server.start();
try {
int localPort = ((ServerConnector) server.getConnectors()[0]).getLocalPort();
List<String> remoteRepos = Arrays.asList("http://relevant.not/maven2@id=central");
RuntimeProperties props = new MockRuntimeProperties();
// TODO: local repo should point to target/tmp
MavenResolver resolver = createResolver("target/tmp", remoteRepos, "http", "localhost", localPort, "fuse", "fuse", null);
MavenDownloadProxyServlet servlet = new MavenDownloadProxyServlet(resolver, props, projectDeployer, 5, 0);
AsyncContext context = EasyMock.createMock(AsyncContext.class);
EasyMock.makeThreadSafe(context, true);
final Map<String, Object> attributes = new HashMap<>();
HttpServletRequest request = EasyMock.createMock(HttpServletRequest.class);
HttpServletRequest requestWrapper = new HttpServletRequestWrapper(request) {
@Override
public Object getAttribute(String name) {
return attributes.get(name);
}
@Override
public void setAttribute(String name, Object o) {
attributes.put(name, o);
}
};
EasyMock.makeThreadSafe(request, true);
EasyMock.expect(request.getMethod()).andReturn("GET");
EasyMock.expect(request.getPathInfo()).andReturn("org.apache.camel/camel-core/2.13.0/camel-core-2.13.0-sources.jar");
EasyMock.expect(request.startAsync()).andReturn(context);
context.setTimeout(EasyMock.anyInt());
EasyMock.expectLastCall();
context.addListener((AsyncListener) EasyMock.anyObject());
EasyMock.expectLastCall();
HttpServletResponse response = EasyMock.createMock(HttpServletResponse.class);
EasyMock.makeThreadSafe(response, true);
response.setStatus(EasyMock.anyInt());
EasyMock.expectLastCall();
response.setContentLength(EasyMock.anyInt());
EasyMock.expectLastCall();
response.setContentType((String) EasyMock.anyObject());
EasyMock.expectLastCall();
response.setDateHeader((String) EasyMock.anyObject(), EasyMock.anyLong());
EasyMock.expectLastCall();
response.setHeader((String) EasyMock.anyObject(), (String) EasyMock.anyObject());
EasyMock.expectLastCall().anyTimes();
final CountDownLatch latchDispatch = new CountDownLatch(1);
context.dispatch();
EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {
@Override
public Object answer() throws Throwable {
latchDispatch.countDown();
return null;
}
});
EasyMock.replay(request, response, context);
servlet.start();
servlet.doGet(requestWrapper, response);
latchDispatch.await();
EasyMock.verify(request, response, context);
//
// Subsequent call
//
final ByteArrayOutputStream baos = new ByteArrayOutputStream();
ServletOutputStream outputStream = new ServletOutputStream() {
@Override
public void write(int b) throws IOException {
baos.write(b);
}
@Override
public void write(byte[] b, int off, int len) throws IOException {
baos.write(b, off, len);
}
@Override
public boolean isReady() {
return true;
}
@Override
public void setWriteListener(WriteListener writeListener) {
}
};
while (true) {
long size = (Long) attributes.get(AsynchronousFileChannel.class.getName() + ".size");
long pos = (Long) attributes.get(AsynchronousFileChannel.class.getName() + ".position");
ByteBuffer buffer = (ByteBuffer) attributes.get(ByteBuffer.class.getName());
if (pos + buffer.position() >= size) {
break;
}
EasyMock.reset(request, response, context);
EasyMock.expect(request.getPathInfo()).andReturn("org.apache.camel/camel-core/2.13.0/camel-core-2.13.0-sources.jar");
EasyMock.expect(request.startAsync()).andReturn(context);
context.setTimeout(EasyMock.anyInt());
EasyMock.expectLastCall();
context.addListener((AsyncListener) EasyMock.anyObject());
EasyMock.expectLastCall();
final CountDownLatch latch = new CountDownLatch(1);
context.dispatch();
EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {
@Override
public Object answer() throws Throwable {
latch.countDown();
return null;
}
});
EasyMock.expect(response.getOutputStream()).andReturn(outputStream);
EasyMock.replay(request, response, context);
servlet.doGet(requestWrapper, response);
latch.await();
EasyMock.verify(request, response, context);
}
//
// Last calls
//
EasyMock.reset(request, response, context);
EasyMock.expect(request.getPathInfo()).andReturn("org.apache.camel/camel-core/2.13.0/camel-core-2.13.0-sources.jar");
EasyMock.expect(request.startAsync()).andReturn(context);
context.setTimeout(EasyMock.anyInt());
EasyMock.expectLastCall();
context.addListener((AsyncListener) EasyMock.anyObject());
EasyMock.expectLastCall();
final CountDownLatch latchComplete = new CountDownLatch(1);
context.complete();
EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {
@Override
public Object answer() throws Throwable {
latchComplete.countDown();
return null;
}
});
EasyMock.expect(response.getOutputStream()).andReturn(outputStream);
response.flushBuffer();
EasyMock.expectLastCall();
EasyMock.replay(request, response, context);
servlet.doGet(requestWrapper, response);
latchComplete.await();
EasyMock.verify(request, response, context);
Assert.assertArrayEquals(new byte[] { 0x42 }, baos.toByteArray());
} finally {
server.stop();
if (old != null) {
System.setProperty("karaf.data", old);
}
}
}
use of io.fabric8.insight.metrics.model.Server in project fabric8 by jboss-fuse.
the class AetherTimeoutTest method readTimeout.
@Test
public void readTimeout() throws Exception {
// case 1: resolution fails because of timeout
// - aether uses 500ms read timeout
// - server responds with 1000ms delay
// - we interrupt resolution after 3000ms in case read timeout in socket.setSoTimeout() is set to 0
MavenConfigurationImpl mavenConfiguration = basicMavenConfiguration(500);
mavenConfiguration.setSettings(settingsWithJettyRepository());
Future<Boolean> task = pool.submit(new ResolveArtifactTask(mavenConfiguration, 1000));
try {
boolean resolved = task.get(3000, TimeUnit.MILLISECONDS);
assertFalse("Should not be resolved due to read timeout", resolved);
} catch (TimeoutException e) {
task.cancel(true);
fail("Should fail due to socket read timeout earlier, not due to future.get() timeout.");
}
// case 2: resolution doesn't fail, we're cancelling the task earlier
// - aether uses 6s read timeout
// - server responds with 3s delay
// - we interrupt resolution after 2s ensuring that Aether resolution didn't end on timeout yet
mavenConfiguration = basicMavenConfiguration(6000);
mavenConfiguration.setSettings(settingsWithJettyRepository());
task = pool.submit(new ResolveArtifactTask(mavenConfiguration, 3000));
boolean timedOut;
try {
task.get(2000, TimeUnit.MILLISECONDS);
timedOut = false;
fail("Task should not be completed yet");
} catch (TimeoutException e) {
timedOut = true;
task.cancel(true);
}
assertTrue("Resolution task should be interrupted", timedOut);
}
Aggregations