add test for subclass

Chris Santero · Chris Santero · commit 015b7ffc0797 · 2015-07-17T13:36:02.000-04:00
diff --git a/JSONAPI.EntityFramework.Tests/DbContextExtensionsTests.cs b/JSONAPI.EntityFramework.Tests/DbContextExtensionsTests.cs
@@ -30,6 +30,11 @@ private class NotAnEntity
             public string Temporary { get; set; }
         }
 
+        private class SubPost : Post
+        {
+            public string Foo { get; set; }
+        }
+
         private DbConnection _conn;
         private TestDbContext _context;
 
@@ -77,6 +82,17 @@ public void GetKeyNamesNonStandardIdTest()
             keyNames.First().Should().Be("Url");
         }
 
+        [TestMethod]
+        public void GetKeyNamesForChildClass()
+        {
+            // Act
+            IEnumerable<string> keyNames = _context.GetKeyNames(typeof(SubPost)).ToArray();
+
+            // Assert
+            keyNames.Count().Should().Be(1);
+            keyNames.First().Should().Be("Id");
+        }
+
         [TestMethod]
         public void GetKeyNamesNotAnEntityTest()
         {
