@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); }