1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
# vim: set fileencoding=utf-8 :
# (C) 2012 Guido Günther <agx@sigxcpu.org>
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, please see
# <http://www.gnu.org/licenses/>
"""Test L{gbp.pq}"""
from . import context
from . import testutils
import gbp.log
import gbp.scripts.import_dscs as import_dscs
from gbp.errors import GbpError
class StubGitImportDsc(object):
"""
A Stub for GitImportDsc.
"""
def __init__(self, args):
self.failfile = None
for arg in args:
if arg.startswith('--failfile'):
self.failfile = "%s.dsc" % arg.split('=')[1]
def importdsc(self, dsc):
"""
Stub the dsc import and fail if we were told to do
so by the --failfile option.
"""
return 1 if dsc.filename == self.failfile else 0
class DscStub(object):
def __init__(self, filename, version):
self.filename = filename
self.version = version
self.dscfile = filename
@classmethod
def parse(cls, filename):
# filename is like file1.dsc, file2.dsc, use
# the digit as version number
version = filename[4]
return cls(filename, version)
class TestImportDscs(testutils.DebianGitTestRepo):
"""Test L{gbp.scripts.import_dscs}'s """
def setUp(self):
testutils.DebianGitTestRepo.setUp(self)
context.chdir(self.repo.path)
self.orig_err = gbp.log.err
gbp.log.err = self._check_err_msg
self.safed_GitImportDsc = import_dscs.GitImportDsc
self.safed_DscFile = import_dscs.DscFile
import_dscs.GitImportDsc = StubGitImportDsc
import_dscs.DscFile = DscStub
def _check_err_msg(self, err):
self.assertIsInstance(err, GbpError)
self.assertIn("Failed to import", str(err))
def test_import_success(self):
"""Test importing success with stub"""
ret = import_dscs.main(['argv0', 'file1.dsc', 'file2.dsc'])
self.assertEqual(ret, 0)
def test_import_fail_first(self):
ret = import_dscs.main(['argv0',
'--failfile=file1',
'file1.dsc'])
self.assertEqual(ret, 1)
def test_import_fail_second(self):
ret = import_dscs.main(['argv0',
'--failfile=file1',
'--failfile=file2',
'file1.dsc',
'file2.dsc'])
self.assertEqual(ret, 1)
def tearDown(self):
gbp.log.err = self.orig_err
testutils.DebianGitTestRepo.tearDown(self)
context.teardown()
import_dscs.GitImportDsc = self.safed_GitImportDsc
import_dscs.DscFile = self.safed_DscFile
|
