1 files changed, 10 insertions, 4 deletions

diff --git a/kronos/tests/itest_sqlite_storage.py b/kronos/tests/itest_sqlite_storage.py
index 4628a23..e2a44d2 100644
--- a/kronos/tests/itest_sqlite_storage.py
+++ b/kronos/tests/itest_sqlite_storage.py
@@ -37,13 +37,14 @@ class TestSQLitBackEnd(unittest.TestCase):
         self.model1.bar = '456'
         self.model1.baz = '789'
 
+        self.storage.save(self.model1)
+
     def test_no_connect_should_cause_error(self):
         self.storage.connection = None
         assert_raises(NotConnected, self.storage.save, self.model1)
 
     def test_save_and_select(self):
-        self.storage.save(self.model1)
-        results = self.storage.get(SampleModel, foo='123')
+        results = self.storage.get(SampleModel, foo='123')[0]
 
         assert isinstance(results, SampleModel)
         assert_equals(results.foo, '123')
@@ -51,8 +52,7 @@ class TestSQLitBackEnd(unittest.TestCase):
         assert_equals(results.baz, '789')
 
     def test_save_and_update(self):
-        self.storage.save(self.model1)
-        results = self.storage.get(SampleModel, foo='123')
+        results = self.storage.get(SampleModel, foo='123')[0]
         results.foo = 'test'
         self.storage.save(results)
 
@@ -61,6 +61,12 @@ class TestSQLitBackEnd(unittest.TestCase):
         assert_equals(results.bar, '456')
         assert_equals(results.baz, '789')
 
+    def test_get_without_args(self):
+        self.storage.save(self.model1)
+        results = self.storage.get(SampleModel)
+
+        assert_equals(len(results), 2)
+
 
 if __name__ == "__main__":
     unittest.main()