add tests for truncated files
[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       RecoverCorruptPayloadSingleTest \
27     , RecoverCorruptPayloadMultiTest \
28     , RecoverCorruptPayloadGZSingleTest \
29     , RecoverCorruptPayloadGZMultiTest \
30     , RecoverCorruptPayloadGZAESSingleTest \
31     , RecoverCorruptPayloadGZAESMultiTest \
32     , RecoverCorruptHeaderSingleTest \
33     , RecoverCorruptHeaderMultiTest \
34     , RecoverCorruptHeaderGZSingleTest \
35     , RecoverCorruptHeaderGZMultiTest \
36     , RecoverCorruptHeaderGZAESSingleTest \
37     , RecoverCorruptHeaderGZAESMultiTest \
38     , RecoverCorruptTruncateTest \
39     , RecoverCorruptTruncateGZTest \
40     , RecoverCorruptTruncateGZAESTest \
41     , RescueCorruptHeaderCTSizeGZAESTest \
42     , RecoverCorruptEntireHeaderSingleTest \
43     , RecoverCorruptEntireHeaderMultiTest \
44     , RecoverCorruptEntireHeaderGZSingleTest \
45     , RecoverCorruptEntireHeaderGZMultiTest \
46     , RecoverCorruptEntireHeaderGZAESSingleTest \
47     , RecoverCorruptEntireHeaderGZAESMultiTest \
48     , RescueCorruptLeadingGarbageSingleTest \
49     , RescueCorruptLeadingGarbageMultiTest \
50     , RecoverCorruptTrailingDataSingleTest \
51     , RecoverCorruptTrailingDataMultiTest \
52     , RecoverCorruptTrailingDataGZSingleTest \
53     , RecoverCorruptTrailingDataGZMultiTest \
54     , RecoverCorruptTrailingDataGZAESSingleTest \
55     , RecoverCorruptTrailingDataGZAESMultiTest \
56     , RecoverCorruptVolumeTest \
57     , RecoverCorruptVolumeGZTest \
58     , RecoverCorruptVolumeGZAESTest \
59     , RecoverCorruptHoleTest \
60     , RecoverCorruptHoleGZTest \
61     , RecoverCorruptHoleGZAESTest \
62     , RescueCorruptTruncateTest \
63     , RescueCorruptTruncateGZTest \
64     , RescueCorruptTruncateGZAESTest \
65     , RescueCorruptHoleTest \
66     , RescueCorruptHoleGZTest \
67     , RescueCorruptHoleGZAESTest \
68     , GenIndexIntactSingleTest \
69     , GenIndexIntactSingleGZTest \
70     , GenIndexIntactSingleGZAESTest \
71     , GenIndexIntactMultiTest \
72     , GenIndexIntactMultiGZTest \
73     , GenIndexIntactMultiGZAESTest \
74     , GenIndexCorruptTruncateTest \
75     , GenIndexCorruptTruncateGZTest \
76     , GenIndexCorruptTruncateGZAESTest \
77     , GenIndexCorruptHoleTest \
78     , GenIndexCorruptHoleGZTest \
79     , GenIndexCorruptHoleGZAESTest \
80     , GenIndexCorruptEntireHeaderTest \
81     , GenIndexCorruptEntireHeaderGZTest \
82     , GenIndexCorruptEntireHeaderGZAESTest
83
84 from testing.test_rescue_tar import RescueTarTest
85 from testing.test_encryption import EncryptionTest
86 from testing.test_deltatar import (DeltaTarTest, DeltaTar2Test,
87     DeltaTarStreamTest, DeltaTarGzipTest, DeltaTarGzipStreamTest,
88     DeltaTarGzipConcatTest, DeltaTarGzipAes128ConcatTest,
89     DeltaTarAes128ConcatTest
90     )
91 from testing.test_compression_level import suite
92
93 if __name__ == "__main__":
94     import sys
95     if len (sys.argv) == 1:
96         unittest.TextTestRunner().run(suite())
97         unittest.main()
98     else:
99         suite = unittest.TestSuite ()
100         def add (n):
101             ret = False
102             for group in [#testing.test_multivol
103                            MultivolGnuFormatTest, MultivolPaxFormatTest
104                          # testing.test_concat_compress
105                          , ConcatCompressTest
106                          # testing.test_rescue_tar
107                          , RescueTarTest
108                          # testing.test_encryption
109                          , EncryptionTest
110                          # testing.test_deltatar
111                          , DeltaTarTest,             DeltaTar2Test
112                          , DeltaTarStreamTest,       DeltaTarGzipTest
113                          , DeltaTarGzipStreamTest,   DeltaTarGzipConcatTest
114                          , DeltaTarGzipAes128ConcatTest
115                          , DeltaTarAes128ConcatTest
116                          , HeaderTest, AESGCMTest
117
118                          # testing.test_recover
119                          , RecoverCorruptPayloadSingleTest
120                          , RecoverCorruptPayloadMultiTest
121                          , RecoverCorruptPayloadGZSingleTest
122                          , RecoverCorruptPayloadGZMultiTest
123                          , RecoverCorruptPayloadGZAESSingleTest
124                          , RecoverCorruptPayloadGZAESMultiTest
125                          , RecoverCorruptHeaderSingleTest
126                          , RecoverCorruptHeaderMultiTest
127                          , RecoverCorruptHeaderGZSingleTest
128                          , RecoverCorruptHeaderGZMultiTest
129                          , RecoverCorruptHeaderGZAESSingleTest
130                          , RecoverCorruptHeaderGZAESMultiTest
131                          , RecoverCorruptTruncateTest
132                          , RecoverCorruptTruncateGZTest
133                          , RecoverCorruptTruncateGZAESTest
134                          , RescueCorruptHeaderCTSizeGZAESTest
135                          , RecoverCorruptEntireHeaderSingleTest
136                          , RecoverCorruptEntireHeaderMultiTest
137                          , RecoverCorruptEntireHeaderGZSingleTest
138                          , RecoverCorruptEntireHeaderGZMultiTest
139                          , RecoverCorruptEntireHeaderGZAESSingleTest
140                          , RecoverCorruptEntireHeaderGZAESMultiTest
141                          , RescueCorruptLeadingGarbageSingleTest
142                          , RescueCorruptLeadingGarbageMultiTest
143                          , RecoverCorruptTrailingDataSingleTest
144                          , RecoverCorruptTrailingDataMultiTest
145                          , RecoverCorruptTrailingDataGZSingleTest
146                          , RecoverCorruptTrailingDataGZMultiTest
147                          , RecoverCorruptTrailingDataGZAESSingleTest
148                          , RecoverCorruptVolumeTest
149                          , RecoverCorruptVolumeGZTest
150                          , RecoverCorruptVolumeGZAESTest
151                          , RecoverCorruptHoleTest
152                          , RecoverCorruptHoleGZTest
153                          , RecoverCorruptHoleGZAESTest
154                          , RescueCorruptTruncateTest
155                          , RescueCorruptTruncateGZTest
156                          , RescueCorruptTruncateGZAESTest
157                          , RescueCorruptHoleTest
158                          , RescueCorruptHoleGZTest
159                          , RescueCorruptHoleGZAESTest
160                          , GenIndexIntactSingleTest
161                          , GenIndexIntactSingleGZTest
162                          , GenIndexIntactSingleGZAESTest
163                          , GenIndexIntactMultiTest
164                          , GenIndexIntactMultiGZTest
165                          , GenIndexIntactMultiGZAESTest
166                          , GenIndexCorruptTruncateTest
167                          , GenIndexCorruptTruncateGZTest
168                          , GenIndexCorruptTruncateGZAESTest
169                          , GenIndexCorruptHoleTest
170                          , GenIndexCorruptHoleGZTest
171                          , GenIndexCorruptHoleGZAESTest
172                          , GenIndexCorruptEntireHeaderTest
173                          , GenIndexCorruptEntireHeaderGZTest
174                          , GenIndexCorruptEntireHeaderGZAESTest
175                          ]:
176                 try:
177                     t = group (n)
178                 except ValueError: # no such test
179                     continue
180                 print ("including ā€œ%sā€" % n)
181                 suite.addTest(t)
182                 ret = True
183             return ret
184         n = 0
185         for arg in sys.argv [1:]:
186             if add (arg) is True:
187                 n += 1
188         if n == 0:
189             print ("ERROR: no valid test name specified, please double-check",
190                    file=sys.stderr)
191             sys.exit (-1)
192         runner = unittest.TextTestRunner ()
193         runner.run (suite)
194