Add test with parameters

Author: ekuvardin

query/src/test/java/tech/ydb/query/ScriptExampleTest.java

@@ -68,6 +68,7 @@ public static void clean() {
      */
     @Test
     public void createScript() {
+
         Result<OperationProtos.Operation> operationResult = retryCtx.supplyResult(session -> session.executeScript(""
                         + "CREATE TABLE series ("
                         + "  series_id UInt64,"
@@ -229,34 +230,74 @@ public void createInsertScript() {
                         + "  first_aired Date,"
                         + "  last_aired Date,"
                         + "  PRIMARY KEY(series_id, season_id)"
-                        + ");"
-                , Params.of("$values", seriesData), executeScriptSettings)
+                        + ")"
+                )
         ).join();
 
-        retryCtx.supplyResult(session -> session.executeScript(""
+        try {
+            Thread.sleep(1000);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
+/*
+
+        retryCtx.supplyResult(session -> session.createQuery(""
                         + "DECLARE $values AS List<Struct<"
                         + "  series_id: Uint64,"
+                        + "  season_id: Uint64,"
                         + "  title: Text,"
-                        + "  series_info: Text,"
-                        + "  release_date: Date"
+                        + "  first_aired: Date,"
+                        + "  last_aired: Date"
                         + ">>;"
-                        + "UPSERT INTO series SELECT * FROM AS_TABLE($values);"
-                        + ""
+                        + "UPSERT INTO seasons SELECT * FROM AS_TABLE($values)",
+                TxMode.SERIALIZABLE_RW,
+                Params.of("$values", seasonsData)
+        ).execute()).join().getStatus().expectSuccess("upsert problem");*/
+
+        Result<OperationProtos.Operation> t1 = retryCtx.supplyResult(session -> session.executeScript(""
                         + "DECLARE $values AS List<Struct<"
                         + "  series_id: Uint64,"
                         + "  season_id: Uint64,"
                         + "  title: Text,"
                         + "  first_aired: Date,"
                         + "  last_aired: Date"
                         + ">>;"
-                        + "UPSERT INTO seasons SELECT * FROM AS_TABLE($series)",
-                Params.of("$values", seriesData, "$series", seasonsData), executeScriptSettings)
+                        + "DECLARE $values1 AS List<Struct<" +
+                        "                        series_id: Uint64," +
+                        "                        title: Text," +
+                        "                        series_info: Text," +
+                        "                        release_date: Date" +
+                        "                        >>;"
+                        + "UPSERT INTO seasons SELECT * FROM AS_TABLE($values);"
+                        + "UPSERT INTO series SELECT * FROM AS_TABLE($values1);",
+                Params.of("$values", seasonsData, "$values1", seriesData), executeScriptSettings)
+        ).join();
+        t1.getValue();
+
+    /*    retryCtx.supplyResult(session -> session.executeScript(""+
+                        "DECLARE $values1 AS List<Struct<" +
+                        "                        series_id: Uint64," +
+                        "                        season_id: Uint64," +
+                        "                        title: Text," +
+                        "                        first_aired: Date," +
+                        "                        last_aired: Date" +
+                        "                        >>;"
+                        + "UPSERT INTO seasons SELECT * FROM AS_TABLE($values1);"
+                        /*+ "UPSERT INTO series SELECT * FROM AS_TABLE($values1);",
+                Params.of("$values1", seasonsData), executeScriptSettings)
         ).join().getStatus().expectSuccess("upsert problem");
+*/
 
 
+        try {
+            Thread.sleep(1000);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
+
         String query
-                = "SELECT series_id, title, release_date "
-                + "FROM series WHERE series_id = 1";
+                = "SELECT series_id "
+                + "FROM seasons WHERE series_id = 1";
 
         // Executes data query with specified transaction control settings.
         Result<QueryReader> result = retryCtx.supplyResult(
@@ -267,10 +308,10 @@ public void createInsertScript() {
 
         Assert.assertTrue(rs.next());
         Assert.assertEquals(1, rs.getColumn("series_id").getUint64());
-        Assert.assertEquals("IT Crowd", rs.getColumn("title").getText());
-        Assert.assertEquals(LocalDate.of(2006, Month.FEBRUARY, 3), rs.getColumn("release_date").getDate());
+     //   Assert.assertEquals("IT Crowd", rs.getColumn("title").getText());
+     //   Assert.assertEquals(LocalDate.of(2006, Month.FEBRUARY, 3), rs.getColumn("release_date").getDate());
+
 
-        Assert.assertFalse(rs.next());
 
 
         operationResult.getValue();