test that new S class is returned by Workspace

universalcore · Apr 30, 2015 · 3f3eabc · 3f3eabc
1 parent 3c4222c
commit 3f3eabc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/elasticgit/tests/test_workspace.py b/elasticgit/tests/test_workspace.py
@@ -181,6 +181,10 @@ def test_to_object(self):
         model = result.to_object()
         self.assertNotEqual(model, person)
 
+    def test_search_class(self):
+        workspace = self.workspace
+        self.assertIsInstance(workspace.S(TestPerson), S)
+
     def test_search_results_to_python(self):
         person_es_data = {
             'age': 1,