[Date Prev][Date Next] [Thread Prev][Thread Next] [Date Index] [Thread Index]

[dak/master] Start new PackageTestCase.



Signed-off-by: Torsten Werner <twerner@debian.org>
---
 daklib/dbconn.py         |    5 +++--
 tests/dbtest_packages.py |   37 +++++++++++++++++++++++++++++++++++++
 2 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100755 tests/dbtest_packages.py

diff --git a/daklib/dbconn.py b/daklib/dbconn.py
index 0db4caf..4abfcc5 100755
--- a/daklib/dbconn.py
+++ b/daklib/dbconn.py
@@ -147,8 +147,9 @@ __all__.append('session_wrapper')
 ################################################################################
 
 class Architecture(object):
-    def __init__(self, *args, **kwargs):
-        pass
+    def __init__(self, arch_string = None, description = None):
+        self.arch_string = arch_string
+        self.description = description
 
     def __eq__(self, val):
         if isinstance(val, str):
diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py
new file mode 100755
index 0000000..0df6eb2
--- /dev/null
+++ b/tests/dbtest_packages.py
@@ -0,0 +1,37 @@
+#!/usr/bin/env python
+
+from db_test import DBDakTestCase
+
+from daklib.dbconn import Architecture, Suite
+
+import unittest
+
+class PackageTestCase(DBDakTestCase):
+    """
+    xxx
+    """
+
+    def setup_architectures(self):
+        "setup a hash of Architecture objects in self.arch"
+
+        self.arch = {}
+        for arch_string in ('source', 'all', 'i386', 'amd64'):
+            self.arch[arch_string] = Architecture(arch_string)
+        # hard code ids for source and all
+        self.arch['source'].arch_id = 1
+        self.arch['all'].arch_id = 2
+        for _, architecture in self.arch.items():
+            self.session.add(architecture)
+            self.session.flush()
+            self.session.refresh(architecture)
+
+    def setUp(self):
+        super(PackageTestCase, self).setUp()
+        self.setup_architectures()
+
+    def test_packages(self):
+        self.assertEqual(1, self.arch['source'].arch_id)
+        self.assertEqual(2, self.arch['all'].arch_id)
+
+if __name__ == '__main__':
+    unittest.main()
-- 
1.5.6.5



Reply to: