use of ddf.catalog.source.impl.SourceDescriptorImpl in project ddf by codice.
the class TestKmlEndpoint method setUp.
@BeforeClass
public static void setUp() throws IOException, URISyntaxException, SourceUnavailableException {
when(mockUriInfo.getBaseUri()).thenReturn(new URI("http://example.com"));
URL bomberLocation = TestKmlEndpoint.class.getResource(ICONS_DIR + BOMBER_ICON);
bomberPath = bomberLocation.getPath().replaceAll(BOMBER_ICON, "");
bomberBytes = IOUtils.toByteArray(bomberLocation.openStream());
URL jetLocation = TestKmlEndpoint.class.getResource(TEST_ICONS_DIR + JET_ICON);
jetPath = jetLocation.getPath().replaceAll(JET_ICON, "");
jetBtyes = IOUtils.toByteArray(jetLocation.openStream());
System.setProperty(SystemBaseUrl.HOST, TEST_HOST);
System.setProperty(SystemBaseUrl.HTTPS_PORT, TEST_PORT);
System.setProperty(SystemBaseUrl.HTTP_PORT, TEST_PORT);
System.setProperty(SystemBaseUrl.ROOT_CONTEXT, "/services");
System.setProperty(SystemInfo.SITE_CONTACT, "example@example.com");
when(mockFramework.getSourceInfo(any(SourceInfoRequest.class))).thenReturn(mockSourceInfoResponse);
SourceDescriptorImpl localDescriptor = new SourceDescriptorImpl(LOCAL_SITE_NAME, null);
SourceDescriptorImpl remoteDescriptor = new SourceDescriptorImpl(REMOTE_SITE_NAME, null);
descriptors.add(localDescriptor);
descriptors.add(remoteDescriptor);
when(mockSourceInfoResponse.getSourceInfo()).thenReturn(descriptors);
when(mockBranding.getProductName()).thenReturn("PRODUCT 0.0.1");
}
use of ddf.catalog.source.impl.SourceDescriptorImpl in project ddf by codice.
the class SourceDescriptorComparatorTest method setup.
@Before
public void setup() throws Exception {
firstSource = new SourceDescriptorImpl("aSource", null, Collections.emptyList());
nextSource = new SourceDescriptorImpl("bSource", null, Collections.emptyList());
lastSource = new SourceDescriptorImpl("cSource", null, Collections.emptyList());
nullSource = new SourceDescriptorImpl(null, null, Collections.emptyList());
}
use of ddf.catalog.source.impl.SourceDescriptorImpl in project ddf by codice.
the class CatalogServiceImplTest method testGetDocumentSourcesSuccess.
/**
* Tests getting source information
*/
@Test
@SuppressWarnings({ "unchecked" })
public void testGetDocumentSourcesSuccess() throws Exception {
final String localSourceId = "local";
final String fed1SourceId = "fed1";
final String fed2SourceId = "fed2";
final String version = "4.0";
final String jsonMimeTypeString = "application/json";
Set<ContentType> contentTypes = new HashSet<>();
contentTypes.add(new ContentTypeImpl("ct1", "v1"));
contentTypes.add(new ContentTypeImpl("ct2", "v2"));
contentTypes.add(new ContentTypeImpl("ct3", null));
JSONArray contentTypesInJSON = new JSONArray();
for (ContentType ct : contentTypes) {
JSONObject ob = new JSONObject();
ob.put("name", ct.getName());
ob.put("version", ct.getVersion() != null ? ct.getVersion() : "");
contentTypesInJSON.add(ob);
}
Set<SourceDescriptor> sourceDescriptors = new HashSet<>();
SourceDescriptorImpl localDescriptor = new SourceDescriptorImpl(localSourceId, contentTypes, Collections.emptyList());
localDescriptor.setVersion(version);
localDescriptor.setAvailable(true);
SourceDescriptorImpl fed1Descriptor = new SourceDescriptorImpl(fed1SourceId, contentTypes, Collections.emptyList());
fed1Descriptor.setVersion(version);
fed1Descriptor.setAvailable(true);
SourceDescriptorImpl fed2Descriptor = new SourceDescriptorImpl(fed2SourceId, null, Collections.emptyList());
fed2Descriptor.setAvailable(true);
sourceDescriptors.add(localDescriptor);
sourceDescriptors.add(fed1Descriptor);
sourceDescriptors.add(fed2Descriptor);
SourceInfoResponse sourceInfoResponse = new SourceInfoResponseImpl(null, null, sourceDescriptors);
CatalogFramework framework = mock(CatalogFramework.class);
when(framework.getSourceInfo(isA(SourceInfoRequestEnterprise.class))).thenReturn(sourceInfoResponse);
CatalogServiceImpl catalogService = new CatalogServiceImpl(framework, attachmentParser, attributeRegistry);
BinaryContent content = catalogService.getSourcesInfo();
assertEquals(jsonMimeTypeString, content.getMimeTypeValue());
String responseMessage = IOUtils.toString(content.getInputStream());
JSONArray srcList = (JSONArray) new JSONParser().parse(responseMessage);
assertEquals(3, srcList.size());
for (Object o : srcList) {
JSONObject src = (JSONObject) o;
assertEquals(true, src.get("available"));
String id = (String) src.get("id");
if (id.equals(localSourceId)) {
assertThat((Iterable<Object>) src.get("contentTypes"), hasItems(contentTypesInJSON.toArray()));
assertEquals(contentTypes.size(), ((JSONArray) src.get("contentTypes")).size());
assertEquals(version, src.get("version"));
} else if (id.equals(fed1SourceId)) {
assertThat((Iterable<Object>) src.get("contentTypes"), hasItems(contentTypesInJSON.toArray()));
assertEquals(contentTypes.size(), ((JSONArray) src.get("contentTypes")).size());
assertEquals(version, src.get("version"));
} else if (id.equals(fed2SourceId)) {
assertEquals(0, ((JSONArray) src.get("contentTypes")).size());
assertEquals("", src.get("version"));
} else {
fail("Invalid ID returned");
}
}
}
use of ddf.catalog.source.impl.SourceDescriptorImpl in project ddf by codice.
the class KmlEndpointTest method setUp.
@BeforeClass
public static void setUp() throws IOException, URISyntaxException, SourceUnavailableException {
when(mockUriInfo.getBaseUri()).thenReturn(new URI("http://example.com"));
URL bomberLocation = KmlEndpointTest.class.getResource(ICONS_DIR + BOMBER_ICON);
bomberPath = bomberLocation.getPath().replaceAll(BOMBER_ICON, "");
bomberBytes = IOUtils.toByteArray(bomberLocation.openStream());
URL jetLocation = KmlEndpointTest.class.getResource(TEST_ICONS_DIR + JET_ICON);
jetPath = jetLocation.getPath().replaceAll(JET_ICON, "");
jetBtyes = IOUtils.toByteArray(jetLocation.openStream());
System.setProperty(SystemBaseUrl.EXTERNAL_HOST, TEST_HOST);
System.setProperty(SystemBaseUrl.EXTERNAL_HTTPS_PORT, TEST_PORT);
System.setProperty(SystemBaseUrl.EXTERNAL_HTTP_PORT, TEST_PORT);
System.setProperty(SystemBaseUrl.INTERNAL_ROOT_CONTEXT, "/services");
System.setProperty(SystemInfo.SITE_CONTACT, "example@example.com");
when(mockFramework.getSourceInfo(any(SourceInfoRequest.class))).thenReturn(mockSourceInfoResponse);
SourceDescriptorImpl localDescriptor = new SourceDescriptorImpl(LOCAL_SITE_NAME, null, Collections.emptyList());
SourceDescriptorImpl remoteDescriptor = new SourceDescriptorImpl(REMOTE_SITE_NAME, null, Collections.emptyList());
descriptors.add(localDescriptor);
descriptors.add(remoteDescriptor);
when(mockSourceInfoResponse.getSourceInfo()).thenReturn(descriptors);
when(mockBranding.getProductName()).thenReturn("PRODUCT 0.0.1");
}
use of ddf.catalog.source.impl.SourceDescriptorImpl in project ddf by codice.
the class TestRestEndpoint method testGetDocumentSourcesSuccess.
/**
* Tests getting source information
*
* @throws Exception
*/
@Test
public void testGetDocumentSourcesSuccess() throws Exception {
final String localSourceId = "local";
final String fed1SourceId = "fed1";
final String fed2SourceId = "fed2";
final String version = "4.0";
final String jsonMimeTypeString = "application/json";
Set<ContentType> contentTypes = new HashSet<ContentType>();
contentTypes.add(new ContentTypeImpl("ct1", "v1"));
contentTypes.add(new ContentTypeImpl("ct2", "v2"));
contentTypes.add(new ContentTypeImpl("ct3", null));
JSONArray contentTypesInJSON = new JSONArray();
for (ContentType ct : contentTypes) {
JSONObject ob = new JSONObject();
ob.put("name", ct.getName());
ob.put("version", ct.getVersion() != null ? ct.getVersion() : "");
contentTypesInJSON.add(ob);
}
Set<SourceDescriptor> sourceDescriptors = new HashSet<SourceDescriptor>();
SourceDescriptorImpl localDescriptor = new SourceDescriptorImpl(localSourceId, contentTypes);
localDescriptor.setVersion(version);
localDescriptor.setAvailable(true);
SourceDescriptorImpl fed1Descriptor = new SourceDescriptorImpl(fed1SourceId, contentTypes);
fed1Descriptor.setVersion(version);
fed1Descriptor.setAvailable(true);
SourceDescriptorImpl fed2Descriptor = new SourceDescriptorImpl(fed2SourceId, null);
fed2Descriptor.setAvailable(true);
sourceDescriptors.add(localDescriptor);
sourceDescriptors.add(fed1Descriptor);
sourceDescriptors.add(fed2Descriptor);
SourceInfoResponse sourceInfoResponse = new SourceInfoResponseImpl(null, null, sourceDescriptors);
CatalogFramework framework = mock(CatalogFramework.class);
when(framework.getSourceInfo(isA(SourceInfoRequestEnterprise.class))).thenReturn(sourceInfoResponse);
RESTEndpoint restEndpoint = new RESTEndpoint(framework);
Response response = restEndpoint.getDocument(null, null);
assertEquals(OK, response.getStatus());
assertEquals(jsonMimeTypeString, response.getMetadata().get("Content-Type").get(0));
String responseMessage = IOUtils.toString((ByteArrayInputStream) response.getEntity());
JSONArray srcList = (JSONArray) new JSONParser().parse(responseMessage);
assertEquals(3, srcList.size());
for (Object o : srcList) {
JSONObject src = (JSONObject) o;
assertEquals(true, src.get("available"));
String id = (String) src.get("id");
if (id.equals(localSourceId)) {
assertThat((Iterable<Object>) src.get("contentTypes"), hasItems(contentTypesInJSON.toArray()));
assertEquals(contentTypes.size(), ((JSONArray) src.get("contentTypes")).size());
assertEquals(version, src.get("version"));
} else if (id.equals(fed1SourceId)) {
assertThat((Iterable<Object>) src.get("contentTypes"), hasItems(contentTypesInJSON.toArray()));
assertEquals(contentTypes.size(), ((JSONArray) src.get("contentTypes")).size());
assertEquals(version, src.get("version"));
} else if (id.equals(fed2SourceId)) {
assertEquals(0, ((JSONArray) src.get("contentTypes")).size());
assertEquals("", src.get("version"));
} else {
fail("Invalid ID returned");
}
}
}
Aggregations