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
|
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import mock
from testtools.matchers import MatchesRegex
from heat.engine.clients.os import swift
from heat.tests.common import HeatTestCase
from heat.tests import utils
class SwiftClientPluginTestCase(HeatTestCase):
def setUp(self):
super(SwiftClientPluginTestCase, self).setUp()
self.swift_client = mock.Mock()
con = utils.dummy_context()
c = con.clients
self.swift_plugin = c.client_plugin('swift')
self.swift_plugin._client = self.swift_client
class SwiftUtilsTests(SwiftClientPluginTestCase):
def test_is_valid_temp_url_path(self):
sc = swift.SwiftClientPlugin
valids = [
"/v1/AUTH_demo/c/o",
"/v1/AUTH_demo/c/o/",
"/v1/TEST_demo/c/o",
"/v1/AUTH_demo/c/pseudo_folder/o",
]
for url in valids:
self.assertTrue(sc.is_valid_temp_url_path(url))
invalids = [
"/v2/AUTH_demo/c/o",
"/v1/AUTH_demo/c//",
"/v1/AUTH_demo/c/",
"/AUTH_demo/c//",
"//AUTH_demo/c/o",
"//v1/AUTH_demo/c/o",
"/v1/AUTH_demo/o",
"/v1/AUTH_demo//o",
"/v1//c/o",
"/v1/c/o",
]
for url in invalids:
self.assertFalse(sc.is_valid_temp_url_path(url))
def test_get_temp_url(self):
self.swift_client.url = ("http://fake-host.com:8080/v1/"
"AUTH_test_tenant_id")
self.swift_client.head_account = mock.Mock(return_value={
'x-account-meta-temp-url-key': '123456'})
self.swift_client.post_account = mock.Mock()
container_name = '1234' # from stack.id
stack_name = 'test'
handle_name = 'foo'
obj_name = '%s-%s' % (stack_name, handle_name)
url = self.swift_plugin.get_temp_url(container_name, obj_name)
self.assertFalse(self.swift_client.post_account.called)
regexp = ("http://fake-host.com:8080/v1/AUTH_test_tenant_id/%s"
"/%s\?temp_url_sig=[0-9a-f]{40}&"
"temp_url_expires=[0-9]{10}" %
(container_name, obj_name))
self.assertThat(url, MatchesRegex(regexp))
timeout = int(url.split('=')[-1])
self.assertTrue(timeout < swift.MAX_EPOCH)
def test_get_temp_url_no_account_key(self):
self.swift_client.url = ("http://fake-host.com:8080/v1/"
"AUTH_demo")
head_account = {}
def post_account(data):
head_account.update(data)
self.swift_client.head_account = mock.Mock(return_value=head_account)
self.swift_client.post_account = post_account
container_name = '1234' # from stack.id
stack_name = 'test'
handle_name = 'foo'
obj_name = '%s-%s' % (stack_name, handle_name)
self.assertNotIn('x-account-meta-temp-url-key', head_account)
self.swift_plugin.get_temp_url(container_name, obj_name)
self.assertIn('x-account-meta-temp-url-key', head_account)
def test_get_signal_url(self):
self.swift_client.url = ("http://fake-host.com:8080/v1/"
"AUTH_test_tenant_id")
self.swift_client.head_account = mock.Mock(return_value={
'x-account-meta-temp-url-key': '123456'})
self.swift_client.post_account = mock.Mock()
container_name = '1234' # from stack.id
stack_name = 'test'
handle_name = 'foo'
obj_name = '%s-%s' % (stack_name, handle_name)
url = self.swift_plugin.get_signal_url(container_name, obj_name)
self.assertTrue(self.swift_client.put_container.called)
self.assertTrue(self.swift_client.put_object.called)
regexp = ("http://fake-host.com:8080/v1/AUTH_test_tenant_id/%s"
"/%s\?temp_url_sig=[0-9a-f]{40}&"
"temp_url_expires=[0-9]{10}" %
(container_name, obj_name))
self.assertThat(url, MatchesRegex(regexp))
|
