From 0c5444cf270aa08ba9a32cf3f1d7c00819bc655a Mon Sep 17 00:00:00 2001
From: Reetika Agrawal <reetika.agrawal@ibm.com>
Date: Tue, 12 Nov 2024 12:39:06 +0530
Subject: [PATCH] [native] Use the correct session while creating a table in
 NativeQueryRunnerUtils

---
 .../nativeworker/NativeQueryRunnerUtils.java       | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/presto-native-execution/src/test/java/com/facebook/presto/nativeworker/NativeQueryRunnerUtils.java b/presto-native-execution/src/test/java/com/facebook/presto/nativeworker/NativeQueryRunnerUtils.java
index cfe11534c15fd..e63b9436b77df 100644
--- a/presto-native-execution/src/test/java/com/facebook/presto/nativeworker/NativeQueryRunnerUtils.java
+++ b/presto-native-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");
         }
     }