@Test
public void testIncrementalUpdate() {
assertInitialState();
// update active state
registry.set(CollectionSettingNames.CS_QUERY_EXRTERNAL_RESOURCE_NAME, Boolean.TRUE);
boolean active = generator.isGeneratingExternalResources();
assertTrue("Active state should be true after update", active);
// add application to ignore list
ApplicationName appName = ApplicationName.valueOf("test-app");
registry.set(
CollectionSettingNames.createApplicationCollectionSettingName(appName), Boolean.FALSE);
Collection<ApplicationName> apps = generator.getDisabledApplicationNames();
assertEquals("Total disabled apps", 1, apps.size());
assertTrue(appName + "is known when disabled", generator.isApplicationNameKnown(appName));
ApplicationName disabledAppName = ListUtil.getFirstMember(apps);
assertEquals("The only disabled app", appName, disabledAppName);
// removed application from ignore list
registry.set(
CollectionSettingNames.createApplicationCollectionSettingName(appName), Boolean.TRUE);
apps = generator.getDisabledApplicationNames();
assertTrue("Total disabled apps is empty", apps.isEmpty());
assertTrue(appName + "is known when enabled", generator.isApplicationNameKnown(appName));
}
private void assertInitialState() {
boolean active = generator.isGeneratingExternalResources();
Collection<ApplicationName> apps = generator.getDisabledApplicationNames();
assertFalse("Generating queries external resources should be disabled by default", active);
assertNotNull("Disabled apps collection", apps);
assertTrue("Disabled apps should be empty by default", apps.isEmpty());
}
@Test
public void testLocateExternalResourceNameWhenEnabledWithSqlAndDisabledApp() {
assertInitialState();
// enable query external resources creation
registry.set(CollectionSettingNames.CS_QUERY_EXRTERNAL_RESOURCE_NAME, Boolean.TRUE);
Operation op = createJdbcOperation("jdbc:foobar://huh:8080");
op.type(JdbcOperationExternalResourceAnalyzer.TYPE);
Frame frame = createJdbcFrame(op);
Trace trace = createJdbcTrace(frame);
registry.set(
CollectionSettingNames.createApplicationCollectionSettingName(trace.getAppName()),
Boolean.FALSE);
Collection<ExternalResourceDescriptor> resources = locateExternalResourceName(trace);
// make sure that no query external resources were created
assertNotNull("external resource descriptors list", resources);
assertEquals("total external resource descriptors", 1, resources.size());
ExternalResourceDescriptor firstDescriptor = ListUtil.getFirstMember(resources);
ExternalResourceType type = ExternalResourceType.valueOf(firstDescriptor.getType());
assertSame("first descriptor type", ExternalResourceType.DATABASE, type);
assertFalse("first descriptor is a parent", firstDescriptor.isParent());
assertTrue(
trace.getAppName() + "is known", generator.isApplicationNameKnown(trace.getAppName()));
}
private void assertQueryExternalResources(Trace trace, String sql) {
Collection<ExternalResourceDescriptor> resources = locateExternalResourceName(trace);
assertNotNull("external resource descriptors list", resources);
assertEquals("total external resource descriptors", 2, resources.size());
assertTrue(
trace.getAppName() + "is known", generator.isApplicationNameKnown(trace.getAppName()));
int totalDbs = 0;
int totalQuerys = 0;
for (ExternalResourceDescriptor desc : resources) {
ExternalResourceType type = ExternalResourceType.valueOf(desc.getType());
switch (type) {
case DATABASE:
totalDbs++;
break;
case QUERY:
assertQueryDescriptor(desc, sql);
totalQuerys++;
break;
default:
assertNotSame(
"Unexpected external resource type (should be QUERY or DATABASE)",
ExternalResourceType.OTHER,
type);
}
}
assertEquals("Total db external resource descriptors", 1, totalDbs);
assertEquals("Total query external resource descriptors", 1, totalQuerys);
}
@Test
public void testLocateExternalResourceNameWhenDisabled() {
assertInitialState();
Operation op = createJdbcOperation("jdbc:foobar://huh:8080");
op.type(JdbcOperationExternalResourceAnalyzer.TYPE);
Frame frame = createJdbcFrame(op);
Trace trace = createJdbcTrace(frame);
Collection<ExternalResourceDescriptor> resources = locateExternalResourceName(trace);
// make sure that no query external resources were created
assertNotNull("external resource descriptors list", resources);
assertEquals("total external resource descriptors", 1, resources.size());
ExternalResourceDescriptor firstDescriptor = ListUtil.getFirstMember(resources);
ExternalResourceType type = ExternalResourceType.valueOf(firstDescriptor.getType());
assertSame("first descriptor type", ExternalResourceType.DATABASE, type);
assertFalse("first descriptor is a parent", firstDescriptor.isParent());
assertTrue(
trace.getAppName() + " is known", generator.isApplicationNameKnown(trace.getAppName()));
}
private Collection<ExternalResourceDescriptor> locateExternalResourceName(Trace trace) {
Collection<ExternalResourceDescriptor> dbDescs = analyzer.locateExternalResourceName(trace);
return generator.createAndAddQueryExternalResourceDescriptors(dbDescs, trace);
}
