X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdb_test.py;h=b73ae7d310cccfdcadc869903d98276945aa7838;hb=0881b94be7b98c6b8f7e3590f3648793e50e78fa;hp=dfd9c5386401ec5e7cd475a29d760d1d47dae39a;hpb=73c90cff1566ae5aa9f8b21f3888945ba0da0866;p=dak.git

diff --git a/tests/db_test.py b/tests/db_test.py
index dfd9c538..b73ae7d3 100644
--- a/tests/db_test.py
+++ b/tests/db_test.py
@@ -235,6 +235,39 @@ class DBDakTestCase(DakTestCase):
         self.binary['python-hello_2.2-1_i386'].suites.append(self.suite['squeeze'])
         self.session.add_all(self.binary.values())
 
+    def setup_overridetypes(self):
+        '''
+        Setup self.otype of class OverrideType.
+        '''
+        if 'otype' in self.__dict__:
+            return
+        self.otype = {}
+        self.otype['deb'] = OverrideType(overridetype = 'deb')
+        self.session.add_all(self.otype.values())
+        self.session.flush()
+
+    def setup_sections(self):
+        '''
+        Setup self.section of class Section.
+        '''
+        if 'section' in self.__dict__:
+            return
+        self.section = {}
+        self.section['python'] = Section(section = 'python')
+        self.session.add_all(self.section.values())
+        self.session.flush()
+
+    def setup_priorities(self):
+        '''
+        Setup self.prio of class Priority.
+        '''
+        if 'prio' in self.__dict__:
+            return
+        self.prio = {}
+        self.prio['standard'] = Priority(priority = 'standard', level = 7)
+        self.session.add_all(self.prio.values())
+        self.session.flush()
+
     def setUp(self):
         if self.metadata is None:
             self.initialize()