add recover tests for completely damaged headers
[python-delta-tar] / runtests.py
1 #!/usr/bin/env python3
2
3 # Copyright (C) 2013 Intra2net AG
4 #
5 # This program is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU Lesser General Public License as published
7 # by the Free Software Foundation; either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU Lesser General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program.  If not, see
17 # <http://www.gnu.org/licenses/lgpl-3.0.html>
18
19
20 import unittest
21
22 from testing.test_crypto import HeaderTest, AESGCMTest
23 from testing.test_multivol import MultivolGnuFormatTest, MultivolPaxFormatTest
24 from testing.test_concat_compress import ConcatCompressTest
25 from testing.test_recover import \
26       RecoverCorruptPayloadTest \
27     , RecoverCorruptPayloadGZTest \
28     , RecoverCorruptPayloadGZAESTest \
29     , RecoverCorruptHeaderTest \
30     , RecoverCorruptHeaderGZTest \
31     , RecoverCorruptHeaderGZAESTest \
32     , RecoverCorruptEntireHeaderTest \
33     , RecoverCorruptEntireHeaderGZTest \
34     , RecoverCorruptEntireHeaderGZAESTest
35 from testing.test_rescue_tar import RescueTarTest
36 from testing.test_encryption import EncryptionTest
37 from testing.test_deltatar import (DeltaTarTest, DeltaTar2Test,
38     DeltaTarStreamTest, DeltaTarGzipTest, DeltaTarGzipStreamTest,
39     DeltaTarGzipConcatTest, DeltaTarGzipAes128ConcatTest,
40     DeltaTarAes128ConcatTest
41     )
42 from testing.test_compression_level import suite
43
44 if __name__ == "__main__":
45     import sys
46     if len (sys.argv) == 1:
47         unittest.TextTestRunner().run(suite())
48         unittest.main()
49     else:
50         suite = unittest.TestSuite ()
51         def add (n):
52             ret = False
53             for group in [#testing.test_multivol
54                            MultivolGnuFormatTest, MultivolPaxFormatTest
55                          # testing.test_concat_compress
56                          , ConcatCompressTest
57                          # testing.test_rescue_tar
58                          , RescueTarTest
59                          # testing.test_encryption
60                          , EncryptionTest
61                          # testing.test_deltatar
62                          , DeltaTarTest,             DeltaTar2Test
63                          , DeltaTarStreamTest,       DeltaTarGzipTest
64                          , DeltaTarGzipStreamTest,   DeltaTarGzipConcatTest
65                          , DeltaTarGzipAes128ConcatTest
66                          , DeltaTarAes128ConcatTest
67                          , HeaderTest, AESGCMTest
68                          # testing.test_recover
69                          , RecoverCorruptPayloadTest
70                          , RecoverCorruptPayloadGZTest
71                          , RecoverCorruptPayloadGZAESTest
72                          , RecoverCorruptHeaderTest
73                          , RecoverCorruptHeaderGZTest
74                          , RecoverCorruptHeaderGZAESTest
75                          , RecoverCorruptEntireHeaderTest
76                          , RecoverCorruptEntireHeaderGZTest
77                          , RecoverCorruptEntireHeaderGZAESTest
78                          ]:
79                 try:
80                     t = group (n)
81                 except ValueError: # no such test
82                     continue
83                 print ("including ā€œ%sā€" % n)
84                 suite.addTest(t)
85                 ret = True
86             return ret
87         n = 0
88         for arg in sys.argv [1:]:
89             if add (arg) is True:
90                 n += 1
91         if n == 0:
92             print ("ERROR: no valid test name specified, please double-check",
93                    file=sys.stderr)
94             sys.exit (-1)
95         runner = unittest.TextTestRunner ()
96         runner.run (suite)
97