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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
import os
import shutil
import subprocess
import tempfile
from testtools import TestCase
from mock import (
MagicMock,
patch,
)
import six
if six.PY3:
from urllib.parse import urlparse
else:
from urlparse import urlparse
try:
from charmhelpers.fetch import (
bzrurl,
UnhandledSource,
)
except ImportError:
bzrurl = None
UnhandledSource = None
class BzrUrlFetchHandlerTest(TestCase):
def setUp(self):
super(BzrUrlFetchHandlerTest, self).setUp()
self.valid_urls = (
"bzr+ssh://example.com/branch-name",
"bzr+ssh://example.com/branch-name/",
"lp:lp-branch-name",
"lp:example/lp-branch-name",
)
self.invalid_urls = (
"http://example.com/foo.tar.gz",
"http://example.com/foo.tgz",
"http://example.com/foo.tar.bz2",
"http://example.com/foo.tbz2",
"http://example.com/foo.zip",
"http://example.com/foo.zip?bar=baz&x=y#whee",
"ftp://example.com/foo.tar.gz",
"https://example.com/foo.tgz",
"file://example.com/foo.tar.bz2",
"git://example.com/foo.tar.gz",
"http://example.com/foo",
"http://example.com/foobar=baz&x=y#tar.gz",
"http://example.com/foobar?h=baz.zip",
"abc:example",
"file//example.com/foo.tar.bz2",
"garbage",
)
self.fh = bzrurl.BzrUrlFetchHandler()
def test_handles_bzr_urls(self):
for url in self.valid_urls:
result = self.fh.can_handle(url)
self.assertEqual(result, True, url)
for url in self.invalid_urls:
result = self.fh.can_handle(url)
self.assertNotEqual(result, True, url)
@patch('charmhelpers.fetch.bzrurl.check_call')
def test_branch(self, check_call):
dest_path = "/destination/path"
for url in self.valid_urls:
self.fh.remote_branch = MagicMock()
self.fh.load_plugins = MagicMock()
self.fh.branch(url, dest_path)
check_call.assert_called_with(['bzr', 'branch', url, dest_path])
for url in self.invalid_urls:
with patch.dict('os.environ', {'CHARM_DIR': 'foo'}):
self.assertRaises(UnhandledSource, self.fh.branch,
url, dest_path)
@patch('charmhelpers.fetch.bzrurl.check_call')
def test_branch_revno(self, check_call):
dest_path = "/destination/path"
for url in self.valid_urls:
self.fh.remote_branch = MagicMock()
self.fh.load_plugins = MagicMock()
self.fh.branch(url, dest_path, revno=42)
check_call.assert_called_with(['bzr', 'branch', '-r', '42',
url, dest_path])
for url in self.invalid_urls:
with patch.dict('os.environ', {'CHARM_DIR': 'foo'}):
self.assertRaises(UnhandledSource, self.fh.branch, url,
dest_path)
def test_branch_functional(self):
src = None
dst = None
try:
src = tempfile.mkdtemp()
subprocess.check_call(['bzr', 'init', src])
dst = tempfile.mkdtemp()
os.rmdir(dst)
self.fh.branch(src, dst)
assert os.path.exists(os.path.join(dst, '.bzr'))
self.fh.branch(src, dst) # idempotent
assert os.path.exists(os.path.join(dst, '.bzr'))
finally:
if src:
shutil.rmtree(src, ignore_errors=True)
if dst:
shutil.rmtree(dst, ignore_errors=True)
def test_installs(self):
self.fh.branch = MagicMock()
for url in self.valid_urls:
branch_name = urlparse(url).path.strip("/").split("/")[-1]
dest = os.path.join('foo', 'fetched')
dest_dir = os.path.join(dest, os.path.basename(branch_name))
with patch.dict('os.environ', {'CHARM_DIR': 'foo'}):
where = self.fh.install(url)
self.assertEqual(where, dest_dir)
@patch('charmhelpers.fetch.bzrurl.mkdir')
def test_installs_dir(self, _mkdir):
self.fh.branch = MagicMock()
for url in self.valid_urls:
branch_name = urlparse(url).path.strip("/").split("/")[-1]
dest = os.path.join('opt', 'f')
dest_dir = os.path.join(dest, os.path.basename(branch_name))
with patch.dict('os.environ', {'CHARM_DIR': 'foo'}):
where = self.fh.install(url, dest)
self.assertEqual(where, dest_dir)
_mkdir.assert_called_with(dest, perms=0o755)
|