Skip to content

Commit 76ffae5

Browse files
committed
Add duplicate checker
1 parent d28d3b8 commit 76ffae5

File tree

1 file changed

+23
-4
lines changed

1 file changed

+23
-4
lines changed

stroom-analytics/stroom-analytics-impl/src/test/java/stroom/analytics/impl/TestDuplicateCheckFactory.java

+23-4
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ void after() {
7070
@Test
7171
void test() {
7272
final DuplicateCheckFactoryImpl duplicateCheckFactory = createDuplicateCheckFactory();
73-
final DuplicateCheck duplicateCheck = createDuplicateCheck(duplicateCheckFactory);
73+
final DuplicateCheck duplicateCheck = createDuplicateCheck(duplicateCheckFactory, "test");
7474
final Row row = new Row("test", List.of("test"), 0, "", "");
7575
assertThat(duplicateCheck.check(row)).isTrue();
7676
for (int i = 0; i < 10; i++) {
@@ -81,7 +81,7 @@ void test() {
8181
@Test
8282
void testReload() {
8383
final DuplicateCheckFactoryImpl duplicateCheckFactory = createDuplicateCheckFactory();
84-
final DuplicateCheck duplicateCheck = createDuplicateCheck(duplicateCheckFactory);
84+
final DuplicateCheck duplicateCheck = createDuplicateCheck(duplicateCheckFactory, "test");
8585
final Row row = new Row("test", List.of("test"), 0, "", "");
8686
assertThat(duplicateCheck.check(row)).isTrue();
8787
for (int i = 0; i < 10; i++) {
@@ -90,7 +90,24 @@ void testReload() {
9090
duplicateCheckFactory.close();
9191

9292
final DuplicateCheckFactoryImpl duplicateCheckFactory2 = createDuplicateCheckFactory();
93-
final DuplicateCheck duplicateCheck2 = createDuplicateCheck(duplicateCheckFactory2);
93+
final DuplicateCheck duplicateCheck2 = createDuplicateCheck(duplicateCheckFactory2, "test");
94+
for (int i = 0; i < 10; i++) {
95+
assertThat(duplicateCheck2.check(row)).isFalse();
96+
}
97+
}
98+
99+
@Test
100+
void testDifferentAnalytic() {
101+
final DuplicateCheckFactoryImpl duplicateCheckFactory = createDuplicateCheckFactory();
102+
final DuplicateCheck duplicateCheck1 = createDuplicateCheck(duplicateCheckFactory, "test1");
103+
final Row row = new Row("test", List.of("test"), 0, "", "");
104+
assertThat(duplicateCheck1.check(row)).isTrue();
105+
for (int i = 0; i < 10; i++) {
106+
assertThat(duplicateCheck1.check(row)).isFalse();
107+
}
108+
109+
final DuplicateCheck duplicateCheck2 = createDuplicateCheck(duplicateCheckFactory, "test2");
110+
assertThat(duplicateCheck2.check(row)).isTrue();
94111
for (int i = 0; i < 10; i++) {
95112
assertThat(duplicateCheck2.check(row)).isFalse();
96113
}
@@ -111,8 +128,10 @@ private DuplicateCheckFactoryImpl createDuplicateCheckFactory() {
111128
new AnalyticResultStoreConfig());
112129
}
113130

114-
private DuplicateCheck createDuplicateCheck(final DuplicateCheckFactoryImpl duplicateCheckFactory) {
131+
private DuplicateCheck createDuplicateCheck(final DuplicateCheckFactoryImpl duplicateCheckFactory,
132+
final String ruleUUID) {
115133
final AnalyticRuleDoc analyticRuleDoc = AnalyticRuleDoc.builder()
134+
.uuid(ruleUUID)
116135
.languageVersion(QueryLanguageVersion.STROOM_QL_VERSION_0_1)
117136
.query("test")
118137
.analyticProcessType(AnalyticProcessType.SCHEDULED_QUERY)

0 commit comments

Comments
 (0)