from collections import defaultdict
from daklib import srcformats
+from daklib.formats import parse_format
from daklib.dak_exceptions import UnknownFormatError
class SourceFormatTestCase(unittest.TestCase):
class ParseFormatTestCase(unittest.TestCase):
def assertParse(self, format, expected):
- self.assertEqual(srcformats.parse_format(format), expected)
+ self.assertEqual(parse_format(format), expected)
def assertParseFail(self, format):
self.assertRaises(
UnknownFormatError,
- lambda: srcformats.parse_format(format)
+ lambda: parse_format(format)
)
def testParse(self):
self.assertInvalid((0, 0))
self.assertInvalid((3, 0, 'native'))
+class FormatFromStringTestCase(unittest.TestCase):
+ def assertFormat(self, txt, klass):
+ self.assertEqual(srcformats.get_format_from_string(txt), klass)
+
+ def assertInvalid(self, txt):
+ self.assertRaises(
+ UnknownFormatError,
+ lambda: srcformats.get_format_from_string(txt),
+ )
+
+ def testFormats(self):
+ self.assertFormat('1.0', srcformats.FormatOne)
+ self.assertFormat('3.0 (native)', srcformats.FormatThree)
+ self.assertFormat('3.0 (quilt)', srcformats.FormatThreeQuilt)
+
+ def testInvalidFormats(self):
+ self.assertInvalid('')
+ self.assertInvalid('.')
+ self.assertInvalid('3.0 (cvs)')
+ self.assertInvalid(' 1.0 ')
+ self.assertInvalid('8.4 (hardy)')
+
if __name__ == '__main__':
unittest.main()