tests/test_srcformats.py

   1 #!/usr/bin/env python
   2
   3 from base_test import DakTestCase
   4
   5 from daklib import srcformats
   6 from collections import defaultdict
   7 from daklib.formats import parse_format
   8 from daklib.dak_exceptions import UnknownFormatError
   9
  10 class SourceFormatTestCase(DakTestCase):
  11     def get_rejects(self, has_vars):
  12         has = defaultdict(lambda: 0)
  13         has.update(has_vars)
  14         return list(self.fmt.reject_msgs(has))
  15
  16     def assertAccepted(self, has):
  17         self.assertEqual(self.get_rejects(has), [])
  18
  19     def assertRejected(self, has):
  20         self.assertNotEqual(self.get_rejects(has), [])
  21
  22 class FormatOneTestCase(SourceFormatTestCase):
  23     fmt = srcformats.FormatOne
  24
  25     def testEmpty(self):
  26         self.assertRejected({})
  27
  28     def testNative(self):
  29         self.assertAccepted({'native_tar': 1, 'native_tar_gz': 1})
  30
  31     def testStandard(self):
  32         self.assertAccepted({
  33             'orig_tar': 1,
  34             'orig_tar_gz': 1,
  35             'debian_diff': 1,
  36         })
  37
  38     def testDisallowed(self):
  39         self.assertRejected({
  40             'native_tar': 1,
  41             'native_tar_gz': 1,
  42             'debian_tar': 1,
  43         })
  44         self.assertRejected({
  45             'orig_tar': 1,
  46             'orig_tar_gz': 1,
  47             'debian_diff': 0,
  48         })
  49         self.assertRejected({
  50             'native_tar': 1,
  51             'native_tar_gz': 1,
  52             'more_orig_tar': 1,
  53         })
  54         self.assertRejected({
  55             'native_tar': 1,
  56             'native_tar_gz': 1,
  57             'debian_diff': 1,
  58         })
  59         self.assertRejected({
  60             'orig_tar': 1,
  61             'orig_tar_gz': 1,
  62             'debian_diff': 1,
  63             'orig_tar_sig': 1,
  64         })
  65
  66 class FormatTreeTestCase(SourceFormatTestCase):
  67     fmt = srcformats.FormatThree
  68
  69     def testEmpty(self):
  70         self.assertRejected({})
  71
  72     def testSimple(self):
  73         self.assertAccepted({'native_tar': 1})
  74
  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})
  80
  81 class FormatTreeQuiltTestCase(SourceFormatTestCase):
  82     fmt = srcformats.FormatThreeQuilt
  83
  84     def testEmpty(self):
  85         self.assertRejected({})
  86
  87     def testSimple(self):
  88         self.assertAccepted({'orig_tar': 1, 'debian_tar': 1})
  89
  90     def testMultipleTarballs(self):
  91         self.assertAccepted({
  92             'orig_tar': 1,
  93             'debian_tar': 1,
  94             'more_orig_tar': 42,
  95         })
  96
  97     def testDisallowed(self):
  98         self.assertRejected({
  99             'orig_tar': 1,
 100             'debian_tar': 1,
 101             'debian_diff': 1
 102         })
 103         self.assertRejected({
 104             'orig_tar': 1,
 105             'debian_tar': 1,
 106             'native_tar': 1,
 107         })
 108
 109 class FormatFromStringTestCase(DakTestCase):
 110     def assertFormat(self, txt, klass):
 111         self.assertEqual(srcformats.get_format_from_string(txt), klass)
 112
 113     def assertInvalid(self, txt):
 114         self.assertRaises(
 115             UnknownFormatError,
 116             lambda: srcformats.get_format_from_string(txt),
 117         )
 118
 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)
 123
 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)')
 130
 131 if __name__ == '__main__':
 132     import unittest
 133     unittest.main()