Fixes for issue ThatRendle#227

ThatRendle · ThatRendle · commit 44a3589125c1 · 2012-10-15T20:48:41.000+01:00
diff --git a/Simple.Data.BehaviourTest/GetCountTest.cs b/Simple.Data.BehaviourTest/GetCountTest.cs
@@ -0,0 +1,45 @@
+﻿namespace Simple.Data.IntegrationTest
+{
+    using System;
+    using Mocking.Ado;
+    using NUnit.Framework;
+
+    [TestFixture]
+    public class GetCountTest : DatabaseIntegrationContext
+    {
+        protected override void SetSchema(MockSchemaProvider schemaProvider)
+        {
+            schemaProvider.SetTables(new[] {"dbo", "Users", "BASE TABLE"});
+
+            schemaProvider.SetColumns(new object[] {"dbo", "Users", "Id", true},
+                                      new[] {"dbo", "Users", "Name"},
+                                      new[] {"dbo", "Users", "Password"},
+                                      new[] {"dbo", "Users", "Age"});
+        }
+         
+        [Test]
+        public void GetCountBasic()
+        {
+            EatException<SimpleDataException>(() => _db.Users.GetCount());
+            GeneratedSqlIs("select count(*) from [dbo].[users]");
+        }
+
+        [Test]
+        public void GetCountWithColumnThrowsException()
+        {
+            Assert.Throws<BadExpressionException>(() => _db.Users.GetCount(_db.Users.Id));
+        }
+
+        [Test]
+        public void AssigningToColumnThrowsException()
+        {
+            Assert.Throws<BadExpressionException>(() => _db.Users.GetCount(_db.Users.Id = 1));
+        }
+        
+        [Test]
+        public void MultipleArgumentsThrowsException()
+        {
+            Assert.Throws<ArgumentException>(() => _db.Users.GetCount(_db.Users.Id == 1, _db.Users.Name == "Bob"));
+        }
+    }
+}
