from db_test import DBDakTestCase
-from daklib.dbconn import DBConn, BinContents
+from daklib.dbconn import DBConn, BinContents, OverrideType
from sqlalchemy.exc import FlushError, IntegrityError
import unittest
self.session.add(contents2)
self.session.flush()
+ def test_overridetype(self):
+ '''
+ Test the OverrideType class.
+ '''
+ debtype = OverrideType(overridetype = 'deb')
+ self.session.add(debtype)
+ self.session.flush()
+ self.assertEqual('deb', debtype.overridetype)
+ self.assertEqual(0, debtype.overrides.count())
+
if __name__ == '__main__':
unittest.main()