add test

Author: laurit

instrumentation-api-incubator/src/main/java/io/opentelemetry/instrumentation/api/incubator/semconv/db/SqlStatementSanitizer.java

@@ -56,6 +56,11 @@ private static SqlStatementInfo sanitizeImpl(@Nullable String statement, SqlDial
     return AutoSqlSanitizer.sanitize(statement, dialect);
   }
 
+  // visible for tests
+  static boolean isCached(String statement) {
+    return sqlToStatementInfoCache.get(CacheKey.create(statement, SqlDialect.DEFAULT)) != null;
+  }
+
   @AutoValue
   abstract static class CacheKey {
 

instrumentation-api-incubator/src/test/java/io/opentelemetry/instrumentation/api/incubator/semconv/db/SqlStatementSanitizerTest.java

@@ -137,6 +137,27 @@ public void longInStatementDoesntCauseStackOverflow() {
     assertThat(sanitized).isEqualTo("select col from table where col in (?)");
   }
 
+  @Test
+  public void largeStatementCached() {
+    // test that short statement is cached
+    String shortStatement = "SELECT * FROM TABLE WHERE FIELD = 1234";
+    String sanitizedShort =
+        SqlStatementSanitizer.create(true).sanitize(shortStatement).getFullStatement();
+    assertThat(sanitizedShort).doesNotContain("1234");
+    assertThat(SqlStatementSanitizer.isCached(shortStatement)).isTrue();
+
+    // test that large statement is not cached
+    StringBuffer s = new StringBuffer();
+    for (int i = 0; i < 10000; i++) {
+      s.append("SELECT * FROM TABLE WHERE FIELD = 1234 AND ");
+    }
+    String largeStatement = s.toString();
+    String sanitizedLarge =
+        SqlStatementSanitizer.create(true).sanitize(largeStatement).getFullStatement();
+    assertThat(sanitizedLarge).doesNotContain("1234");
+    assertThat(SqlStatementSanitizer.isCached(largeStatement)).isFalse();
+  }
+
   static class SqlArgs implements ArgumentsProvider {
 
     @Override