move JDBC test on injection with generated keys to SQL Server file

Author: na-ji

dd-java-agent/instrumentation/jdbc/src/test/groovy/DBMInjectionForkedTest.groovy

@@ -47,16 +47,4 @@ class DBMInjectionForkedTest extends AgentTestRunner {
     then:
     assert statement.sql == "/*${fullInjection}*/ ${query}"
   }
-
-  def "single query with generated keys"() {
-    setup:
-    def connection = new TestConnection(false)
-
-    when:
-    def statement = connection.createStatement() as TestStatement
-    statement.executeUpdate(query, 1)
-
-    then:
-    assert statement.sql == "${query} /*${fullInjection}*/"
-  }
 }

dd-java-agent/instrumentation/jdbc/src/test/groovy/SQLServerInjectionForkedTest.groovy

@@ -36,4 +36,19 @@ class SQLServerInjectionForkedTest extends AgentTestRunner {
     // Verify that the SQL does NOT contain traceparent
     assert !statement.sql.contains("traceparent")
   }
+
+  def "SQL Server apend comment when getting generated keys"() {
+    setup:
+    def connection = new TestConnection(false)
+    def metadata = new TestDatabaseMetaData()
+    metadata.setURL("jdbc:microsoft:sqlserver://localhost:1433;DatabaseName=testdb;")
+    connection.setMetaData(metadata)
+
+    when:
+    def statement = connection.createStatement() as TestStatement
+    statement.executeUpdate(query, 1)
+
+    then:
+    assert statement.sql == "${query} /*${serviceInjection}*/"
+  }
 }