6 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
8 from collections import defaultdict
10 from daklib import srcformats
11 from daklib.formats import parse_format
12 from daklib.dak_exceptions import UnknownFormatError
14 class SourceFormatTestCase(unittest.TestCase):
15 def get_rejects(self, has_vars):
16 has = defaultdict(lambda: 0)
18 return list(self.fmt.reject_msgs(has))
20 def assertAccepted(self, has):
21 self.assertEqual(self.get_rejects(has), [])
23 def assertRejected(self, has):
24 self.assertNotEqual(self.get_rejects(has), [])
26 class FormatOneTestCase(SourceFormatTestCase):
27 fmt = srcformats.FormatOne
30 self.assertRejected({})
33 self.assertAccepted({'native_tar': 1, 'native_tar_gz': 1})
35 def testStandard(self):
42 def testDisallowed(self):
64 class FormatTreeTestCase(SourceFormatTestCase):
65 fmt = srcformats.FormatThree
68 self.assertRejected({})
71 self.assertAccepted({'native_tar': 1})
73 def testDisallowed(self):
74 self.assertRejected({'native_tar': 1, 'orig_tar': 1})
75 self.assertRejected({'native_tar': 1, 'debian_diff': 1})
76 self.assertRejected({'native_tar': 1, 'debian_tar': 1})
77 self.assertRejected({'native_tar': 1, 'more_orig_tar': 1})
79 class FormatTreeQuiltTestCase(SourceFormatTestCase):
80 fmt = srcformats.FormatThreeQuilt
83 self.assertRejected({})
86 self.assertAccepted({'orig_tar': 1, 'debian_tar': 1})
88 def testMultipleTarballs(self):
95 def testDisallowed(self):
101 self.assertRejected({
109 class ParseFormatTestCase(unittest.TestCase):
110 def assertParse(self, format, expected):
111 self.assertEqual(parse_format(format), expected)
113 def assertParseFail(self, format):
116 lambda: parse_format(format)
120 self.assertParse('1.0', (1, 0))
123 self.assertParseFail('')
124 self.assertParseFail(' ')
125 self.assertParseFail(' ')
128 self.assertParse('1.2 (three)', (1, 2, 'three'))
129 self.assertParseFail('0.0 ()')
131 class ValidateFormatTestCase(unittest.TestCase):
132 def assertValid(self, format, **kwargs):
133 kwargs['is_a_dsc'] = kwargs.get('is_a_dsc', True)
134 self.fmt.validate_format(format, **kwargs)
136 def assertInvalid(self, *args, **kwargs):
139 lambda: self.assertValid(*args, **kwargs),
142 class ValidateFormatOneTestCase(ValidateFormatTestCase):
143 fmt = srcformats.FormatOne
146 self.assertValid((1, 0))
148 def testInvalid(self):
149 self.assertInvalid((0, 1))
150 self.assertInvalid((3, 0, 'quilt'))
154 def testBinary(self):
155 self.assertValid((1, 5), is_a_dsc=False)
156 self.assertInvalid((1, 0), is_a_dsc=False)
159 self.assertInvalid((1, 3), is_a_dsc=False)
160 self.assertValid((1, 5), is_a_dsc=False)
161 self.assertValid((1, 8), is_a_dsc=False)
162 self.assertInvalid((1, 9), is_a_dsc=False)
164 def testFilesField(self):
165 self.assertInvalid((1, 7), is_a_dsc=False, field='notfiles')
166 self.assertValid((1, 8), is_a_dsc=False, field='notfiles')
168 class ValidateFormatThreeTestCase(ValidateFormatTestCase):
169 fmt = srcformats.FormatThree
172 self.assertValid((3, 0, 'native'))
174 def testInvalid(self):
175 self.assertInvalid((1, 0))
176 self.assertInvalid((0, 0))
177 self.assertInvalid((3, 0, 'quilt'))
179 class ValidateFormatThreeQuiltTestCase(ValidateFormatTestCase):
180 fmt = srcformats.FormatThreeQuilt
183 self.assertValid((3, 0, 'quilt'))
185 def testInvalid(self):
186 self.assertInvalid((1, 0))
187 self.assertInvalid((0, 0))
188 self.assertInvalid((3, 0, 'native'))
190 class FormatFromStringTestCase(unittest.TestCase):
191 def assertFormat(self, txt, klass):
192 self.assertEqual(srcformats.get_format_from_string(txt), klass)
194 def assertInvalid(self, txt):
197 lambda: srcformats.get_format_from_string(txt),
200 def testFormats(self):
201 self.assertFormat('1.0', srcformats.FormatOne)
202 self.assertFormat('3.0 (native)', srcformats.FormatThree)
203 self.assertFormat('3.0 (quilt)', srcformats.FormatThreeQuilt)
205 def testInvalidFormats(self):
206 self.assertInvalid('')
207 self.assertInvalid('.')
208 self.assertInvalid('3.0 (cvs)')
209 self.assertInvalid(' 1.0 ')
210 self.assertInvalid('8.4 (hardy)')
212 if __name__ == '__main__':