[native] Use the correct session while creating a table in NativeQuer…

…yRunnerUtils
prestodb · Nov 13, 2024 · 0c5444c · 0c5444c
1 parent 93526d3
commit 0c5444c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/...tive-execution/src/test/java/com/facebook/presto/nativeworker/NativeQueryRunnerUtils.java b/...tive-execution/src/test/java/com/facebook/presto/nativeworker/NativeQueryRunnerUtils.java
@@ -209,19 +209,19 @@ public static void createNationWithFormat(QueryRunner queryRunner, String storag
     public static void createNationWithFormat(Session session, QueryRunner queryRunner, String storageFormat)
     {
         queryRunner.execute(session, "DROP TABLE IF EXISTS nation");
-        if (storageFormat.equals("PARQUET") && !queryRunner.tableExists(queryRunner.getDefaultSession(), "nation")) {
+        if (storageFormat.equals("PARQUET") && !queryRunner.tableExists(session, "nation")) {
             queryRunner.execute(session, "CREATE TABLE nation AS SELECT * FROM tpch.tiny.nation");
         }
 
-        if (storageFormat.equals("ORC") && !queryRunner.tableExists(queryRunner.getDefaultSession(), "nation")) {
+        if (storageFormat.equals("ORC") && !queryRunner.tableExists(session, "nation")) {
             queryRunner.execute(session, "CREATE TABLE nation AS SELECT * FROM tpch.tiny.nation");
         }
 
-        if (storageFormat.equals("JSON") && !queryRunner.tableExists(queryRunner.getDefaultSession(), "nation_json")) {
+        if (storageFormat.equals("JSON") && !queryRunner.tableExists(session, "nation_json")) {
             queryRunner.execute(session, "CREATE TABLE nation_json WITH (FORMAT = 'JSON') AS SELECT * FROM tpch.tiny.nation");
         }
 
-        if (storageFormat.equals("TEXTFILE") && !queryRunner.tableExists(queryRunner.getDefaultSession(), "nation_text")) {
+        if (storageFormat.equals("TEXTFILE") && !queryRunner.tableExists(session, "nation_text")) {
             queryRunner.execute(session, "CREATE TABLE nation_text WITH (FORMAT = 'TEXTFILE') AS SELECT * FROM tpch.tiny.nation");
         }
     }
@@ -272,7 +272,7 @@ public static void createCustomer(QueryRunner queryRunner)
 
     public static void createCustomer(Session session, QueryRunner queryRunner)
     {
-        if (!queryRunner.tableExists(queryRunner.getDefaultSession(), "customer")) {
+        if (!queryRunner.tableExists(session, "customer")) {
             queryRunner.execute(session, "CREATE TABLE customer AS " +
                     "SELECT custkey, name, address, nationkey, phone, acctbal, comment, mktsegment " +
                     "FROM tpch.tiny.customer");
@@ -394,7 +394,7 @@ public static void createPart(QueryRunner queryRunner)
 
     public static void createPart(Session session, QueryRunner queryRunner)
     {
-        if (!queryRunner.tableExists(queryRunner.getDefaultSession(), "part")) {
+        if (!queryRunner.tableExists(session, "part")) {
             queryRunner.execute(session, "CREATE TABLE part AS SELECT * FROM tpch.tiny.part");
         }
     }
@@ -406,7 +406,7 @@ public static void createPartSupp(QueryRunner queryRunner)
 
     public static void createPartSupp(Session session, QueryRunner queryRunner)
     {
-        if (!queryRunner.tableExists(queryRunner.getDefaultSession(), "partsupp")) {
+        if (!queryRunner.tableExists(session, "partsupp")) {
             queryRunner.execute(session, "CREATE TABLE partsupp AS SELECT * FROM tpch.tiny.partsupp");
         }
     }