[DB-20271] Update EF tests to allow different return types

philip-stoev · Nov 7, 2017 · 9d0a7f1 · 9d0a7f1
1 parent 973a52a
commit 9d0a7f1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/NUnitEFCodeFirstTestProject/Utils.cs b/NUnitEFCodeFirstTestProject/Utils.cs
@@ -16,6 +16,12 @@ class Utils
         static string schema = "hockey";
         static internal string connectionString = "Server=  " + host + "; Database=\"" + database + "\"; User = " + user + " ;Password   = '" + password + "';Schema=\"" + schema + "\"";
 
+        internal static int ToInt(object value)
+        {
+            // go through a parse step to accomodate both 32 and 64 bit return values
+            return Int32.Parse(value.ToString());
+        }
+
         internal static void DropTable(NuoDbConnection cnn, string tableName)
         {
             try
@@ -250,7 +256,7 @@ internal static int GetTableRows()
                 DbCommand command = new NuoDbCommand("select count(*) from hockey", connection);
 
                 connection.Open();
-                return (int)command.ExecuteScalar();
+                return Utils.ToInt(command.ExecuteScalar());
             }
         }
     }