Update tests.

extensions/ql-vscode/test/unit-tests/data-extensions-editor/auto-model.test.ts

@@ -39,7 +39,7 @@ describe("createAutoModelRequest", () => {
       typeName: "Connection",
       methodName: "createQuery",
       methodParameters: "(String)",
-      supported: true,
+      supported: false,
       usages: [
         {
           label: "createQuery(...)",
@@ -69,7 +69,7 @@ describe("createAutoModelRequest", () => {
       typeName: "Query",
       methodName: "executeScalar",
       methodParameters: "(Class)",
-      supported: true,
+      supported: false,
       usages: [
         {
           label: "executeScalar(...)",
@@ -99,7 +99,7 @@ describe("createAutoModelRequest", () => {
       typeName: "Sql2o",
       methodName: "open",
       methodParameters: "()",
-      supported: true,
+      supported: false,
       usages: [
         {
           label: "open(...)",
@@ -129,7 +129,7 @@ describe("createAutoModelRequest", () => {
       typeName: "PrintStream",
       methodName: "println",
       methodParameters: "(String)",
-      supported: true,
+      supported: false,
       usages: [
         {
           label: "println(...)",
@@ -149,7 +149,7 @@ describe("createAutoModelRequest", () => {
       typeName: "Sql2o",
       methodName: "Sql2o",
       methodParameters: "(String,String,String)",
-      supported: true,
+      supported: false,
       usages: [
         {
           label: "new Sql2o(...)",
@@ -169,7 +169,7 @@ describe("createAutoModelRequest", () => {
       typeName: "Sql2o",
       methodName: "Sql2o",
       methodParameters: "(String)",
-      supported: true,
+      supported: false,
       usages: [
         {
           label: "new Sql2o(...)",
@@ -235,6 +235,32 @@ describe("createAutoModelRequest", () => {
     ).toEqual({
       language: "java",
       samples: [
+        {
+          package: "org.sql2o",
+          type: "Sql2o",
+          name: "open",
+          signature: "()",
+          classification: {
+            type: "CLASSIFICATION_TYPE_NEUTRAL",
+            kind: "",
+            explanation: "",
+          },
+          usages: usages["org.sql2o.Sql2o#open()"],
+          input: "Argument[-1]",
+        },
+        {
+          package: "org.sql2o",
+          type: "Sql2o",
+          name: "Sql2o",
+          signature: "(String)",
+          classification: {
+            type: "CLASSIFICATION_TYPE_SINK",
+            kind: "jndi-injection",
+            explanation: "",
+          },
+          usages: usages["org.sql2o.Sql2o#Sql2o(String)"],
+          input: "Argument[-1]",
+        },
         {
           package: "org.sql2o",
           type: "Sql2o",
@@ -250,6 +276,15 @@ describe("createAutoModelRequest", () => {
         },
       ],
       candidates: [
+        {
+          package: "org.sql2o",
+          type: "Connection",
+          name: "createQuery",
+          signature: "(String)",
+          usages: usages["org.sql2o.Connection#createQuery(String)"],
+          input: "Argument[-1]",
+          classification: undefined,
+        },
         {
           package: "org.sql2o",
           type: "Connection",
@@ -259,6 +294,15 @@ describe("createAutoModelRequest", () => {
           input: "Argument[0]",
           classification: undefined,
         },
+        {
+          package: "org.sql2o",
+          type: "Query",
+          name: "executeScalar",
+          signature: "(Class)",
+          usages: usages["org.sql2o.Query#executeScalar(Class)"],
+          input: "Argument[-1]",
+          classification: undefined,
+        },
         {
           package: "org.sql2o",
           type: "Query",
@@ -268,6 +312,18 @@ describe("createAutoModelRequest", () => {
           input: "Argument[0]",
           classification: undefined,
         },
+        {
+          package: "org.springframework.boot",
+          type: "SpringApplication",
+          name: "run",
+          signature: "(Class,String[])",
+          usages:
+            usages[
+              "org.springframework.boot.SpringApplication#run(Class,String[])"
+            ],
+          input: "Argument[-1]",
+          classification: undefined,
+        },
         {
           package: "org.springframework.boot",
           type: "SpringApplication",
@@ -292,6 +348,15 @@ describe("createAutoModelRequest", () => {
           input: "Argument[1]",
           classification: undefined,
         },
+        {
+          package: "java.io",
+          type: "PrintStream",
+          name: "println",
+          signature: "(String)",
+          usages: usages["java.io.PrintStream#println(String)"],
+          input: "Argument[-1]",
+          classification: undefined,
+        },
         {
           package: "java.io",
           type: "PrintStream",
@@ -301,6 +366,15 @@ describe("createAutoModelRequest", () => {
           input: "Argument[0]",
           classification: undefined,
         },
+        {
+          package: "org.sql2o",
+          type: "Sql2o",
+          name: "Sql2o",
+          signature: "(String,String,String)",
+          usages: usages["org.sql2o.Sql2o#Sql2o(String,String,String)"],
+          input: "Argument[-1]",
+          classification: undefined,
+        },
         {
           package: "org.sql2o",
           type: "Sql2o",