Merge pull request #21 from uber/fix/ca/add_get_from_db_test

Add a test to get from database

charlatan/tests/data/relationships.yaml

@@ -8,3 +8,7 @@ relationship_alone:
   fields:
     name: "red"
   model: charlatan.tests.fixtures.models:Color
+
+from_database:
+  id: 1
+  model: charlatan.tests.fixtures.models:Toaster

charlatan/tests/test_sqlalchemy.py

@@ -15,6 +15,7 @@ def setUp(self):
 
     def tearDown(self):
         Base.metadata.drop_all(engine)
+        self.session.close()
 
     def test_double_install(self):
         """Verify that there's no double install."""
@@ -23,3 +24,12 @@ def test_double_install(self):
 
         self.assertEqual(self.session.query(Toaster).count(), 1)
         self.assertEqual(self.session.query(Color).count(), 1)
+
+    def test_getting_from_database(self):
+        """Verify that we can get from the database."""
+        installed = Toaster(id=1)
+        self.session.add(installed)
+        self.session.commit()
+
+        toaster = self.manager.install_fixture("from_database")
+        self.assertEqual(toaster.id, 1)