Skip to content

Commit 32f0399

Browse files
Enable multistatement test parameter
1 parent a574cd3 commit 32f0399

2 files changed

Lines changed: 14 additions & 5 deletions

File tree

Snowflake.Data.Tests/IntegrationTests/SFMultiStatementsIT.cs

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,13 @@ public void TestSelectWithBinding()
130130
{
131131
conn.ConnectionString = ConnectionString;
132132
conn.Open();
133-
SessionParameterAlterer.SetResultFormat(conn, ResultFormat.JSON);
133+
SessionParameterAlterer.SetResultFormat(conn, ResultFormat.ARROW);
134+
135+
using (var command = conn.CreateCommand())
136+
{
137+
command.CommandText = "alter session set ENABLE_FIX_1758055_ADD_ARROW_SUPPORT_FOR_MULTI_STMTS = true";
138+
command.ExecuteNonQuery();
139+
}
134140

135141
IDbCommand cmd = conn.CreateCommand();
136142
// Set statement count
@@ -183,7 +189,13 @@ public void TestMixedQueryTypeWithBinding()
183189
{
184190
conn.ConnectionString = ConnectionString;
185191
conn.Open();
186-
SessionParameterAlterer.SetResultFormat(conn, ResultFormat.JSON);
192+
SessionParameterAlterer.SetResultFormat(conn, ResultFormat.ARROW);
193+
194+
using (var cmd = conn.CreateCommand())
195+
{
196+
cmd.CommandText = "alter session set ENABLE_FIX_1758055_ADD_ARROW_SUPPORT_FOR_MULTI_STMTS = true";
197+
cmd.ExecuteNonQuery();
198+
}
187199

188200
using (DbCommand cmd = conn.CreateCommand())
189201
{

Snowflake.Data/Core/ArrowResultSet.cs

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,6 @@ internal override short GetInt16(int ordinal)
347347
case long ret: return (short)ret;
348348
case int ret: return (short)ret;
349349
case sbyte ret: return ret;
350-
case double ret: return Convert.ToInt16(ret);
351350
default: return (short)value;
352351
}
353352
}
@@ -364,7 +363,6 @@ internal override int GetInt32(int ordinal)
364363
case long ret: return (int)ret;
365364
case short ret: return ret;
366365
case sbyte ret: return ret;
367-
case double ret: return Convert.ToInt32(ret);
368366
default: return (int)value;
369367
}
370368
}
@@ -381,7 +379,6 @@ internal override long GetInt64(int ordinal)
381379
case int ret: return ret;
382380
case short ret: return ret;
383381
case sbyte ret: return ret;
384-
case double ret: return Convert.ToInt64(ret);
385382
default: return (long)value;
386383
}
387384
}

0 commit comments

Comments
 (0)