X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_ormobject.py;fp=tests%2Fdbtest_ormobject.py;h=a5b775535a28683ca2633175ce398d29b51f9ba0;hb=a6b1715c117b4f306a7d301ecc4492c67c6ef98b;hp=0000000000000000000000000000000000000000;hpb=1480ca40a492d78a68b8239c330bd40d024aaefc;p=dak.git

diff --git a/tests/dbtest_ormobject.py b/tests/dbtest_ormobject.py
new file mode 100755
index 00000000..a5b77553
--- /dev/null
+++ b/tests/dbtest_ormobject.py
@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+
+from db_test import DBDakTestCase
+
+from daklib.dbconn import Architecture, Suite
+
+import json
+import re
+import unittest
+
+class ORMObjectTestCase(DBDakTestCase):
+    """
+    The ORMObjectTestCase tests the behaviour of the ORMObject.
+    """
+
+    def test_strings(self):
+        'tests json(), __repr__(), and __str__()'
+        architecture = Architecture(arch_string = 'i386')
+        # test json()
+        data = json.loads(architecture.json())
+        self.assertEqual('i386', data['arch_string'])
+        # test repr()
+        self.assertEqual('<Architecture i386>', repr(architecture))
+        # test str()
+        self.assertTrue(re.match('<Architecture {.*}>', str(architecture)))
+        self.assertTrue(re.search('"arch_string": "i386"', str(architecture)))
+        sid = Suite(suite_name = 'sid')
+        squeeze = Suite(suite_name = 'squeeze')
+        architecture.suites = [sid, squeeze]
+        self.assertTrue(re.search('"suites_count": 2', str(architecture)))
+
+if __name__ == '__main__':
+    unittest.main()