3 from base_test import DakTestCase
5 from daklib import srcformats
6 from collections import defaultdict
7 from daklib.formats import parse_format
8 from daklib.dak_exceptions import UnknownFormatError
10 class SourceFormatTestCase(DakTestCase):
11 def get_rejects(self, has_vars):
12 has = defaultdict(lambda: 0)
14 return list(self.fmt.reject_msgs(has))
16 def assertAccepted(self, has):
17 self.assertEqual(self.get_rejects(has), [])
19 def assertRejected(self, has):
20 self.assertNotEqual(self.get_rejects(has), [])
22 class FormatOneTestCase(SourceFormatTestCase):
23 fmt = srcformats.FormatOne
26 self.assertRejected({})
29 self.assertAccepted({'native_tar': 1, 'native_tar_gz': 1})
31 def testStandard(self):
38 def testDisallowed(self):
66 class FormatTreeTestCase(SourceFormatTestCase):
67 fmt = srcformats.FormatThree
70 self.assertRejected({})
73 self.assertAccepted({'native_tar': 1})
75 def testDisallowed(self):
76 self.assertRejected({'native_tar': 1, 'orig_tar': 1})
77 self.assertRejected({'native_tar': 1, 'debian_diff': 1})
78 self.assertRejected({'native_tar': 1, 'debian_tar': 1})
79 self.assertRejected({'native_tar': 1, 'more_orig_tar': 1})
81 class FormatTreeQuiltTestCase(SourceFormatTestCase):
82 fmt = srcformats.FormatThreeQuilt
85 self.assertRejected({})
88 self.assertAccepted({'orig_tar': 1, 'debian_tar': 1})
90 def testMultipleTarballs(self):
97 def testDisallowed(self):
103 self.assertRejected({
109 class FormatFromStringTestCase(DakTestCase):
110 def assertFormat(self, txt, klass):
111 self.assertEqual(srcformats.get_format_from_string(txt), klass)
113 def assertInvalid(self, txt):
116 lambda: srcformats.get_format_from_string(txt),
119 def testFormats(self):
120 self.assertFormat('1.0', srcformats.FormatOne)
121 self.assertFormat('3.0 (native)', srcformats.FormatThree)
122 self.assertFormat('3.0 (quilt)', srcformats.FormatThreeQuilt)
124 def testInvalidFormats(self):
125 self.assertInvalid('')
126 self.assertInvalid('.')
127 self.assertInvalid('3.0 (cvs)')
128 self.assertInvalid(' 1.0 ')
129 self.assertInvalid('8.4 (hardy)')
131 if __name__ == '__main__':