Improve the test

snowflakedb · May 25, 2018 · de5d945 · de5d945
1 parent ea48b6f
commit de5d945
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/Snowflake.Data.Tests/SFDbDataReaderIT.cs b/Snowflake.Data.Tests/SFDbDataReaderIT.cs
@@ -598,7 +598,7 @@ public void testGetGuid()
         }
 
         [Test]
-        public void TestCopyWithZeroFiles()
+        public void TestCopyCmdUpdateCount()
         {
             using (IDbConnection conn = new SnowflakeDbConnection())
             {
@@ -609,19 +609,26 @@ public void TestCopyWithZeroFiles()
                 cmd.CommandText = "create or replace stage emptyStage";
                 cmd.ExecuteNonQuery();
 
-                cmd.CommandText = "create or replace table testZeroCopy (cola string)";
+                cmd.CommandText = "create or replace table testCopy (cola string)";
                 cmd.ExecuteNonQuery();
 
-                cmd.CommandText = "copy into testZeroCopy from @emptyStage";
-
+                cmd.CommandText = "copy into testCopy from @emptyStage";
                 int updateCount = cmd.ExecuteNonQuery();
                 Assert.AreEqual(0, updateCount);
 
+                // test rows_loaded exists
+                cmd.CommandText = "copy into @%testcopy from (select 'test_string')";
+                cmd.ExecuteNonQuery();
+
+                cmd.CommandText = "copy into testcopy";
+                updateCount = cmd.ExecuteNonQuery();
+                Assert.AreEqual(1, updateCount);
+
                 // clean up
                 cmd.CommandText = "drop stage emptyStage";
                 cmd.ExecuteNonQuery();
 
-                cmd.CommandText = "drop table testZeroCopy";
+                cmd.CommandText = "drop table testCopy";
                 cmd.ExecuteNonQuery();
 
                 conn.Close();