Make swift post output an error message when failing
swiftclient.shell.st_post was ignoring the result dict returned from SwiftService.post and therefore giving no indication when posts failed e.g. due to invalid auth credentials. This patch ensures that SwiftService always returns a result_dict from post() and then checks the result dict. On failure the shell now outputs the ClientException message and exits with error code 1. Also adds unit tests and cleans up some unnecessary mocked return values in existing tests. Closes-Bug: 1383243 Change-Id: I111150eb3d026c8d01c2cac13d3613ca7304e5b9
This commit is contained in:
parent
871a3e40f3
commit
95056d1ef4
@ -595,7 +595,6 @@ class SwiftService(object):
|
||||
'response_dict': response_dict
|
||||
})
|
||||
return res
|
||||
else:
|
||||
if not objects:
|
||||
res["action"] = "post_container"
|
||||
response_dict = {}
|
||||
|
@ -543,8 +543,7 @@ If the container is not found, it will be created automatically.
|
||||
|
||||
Positional arguments:
|
||||
[container] Name of container to post to.
|
||||
[object] Name of object to post. Specify multiple times
|
||||
for multiple objects.
|
||||
[object] Name of object to post.
|
||||
|
||||
Optional arguments:
|
||||
--read-acl <acl> Read ACL for containers. Quick summary of ACL syntax:
|
||||
@ -596,7 +595,7 @@ def st_post(parser, args, output_manager):
|
||||
with SwiftService(options=_opts) as swift:
|
||||
try:
|
||||
if not args:
|
||||
swift.post()
|
||||
result = swift.post()
|
||||
else:
|
||||
container = args[0]
|
||||
if '/' in container:
|
||||
@ -612,15 +611,16 @@ def st_post(parser, args, output_manager):
|
||||
results_iterator = swift.post(
|
||||
container=container, objects=objects
|
||||
)
|
||||
for result in results_iterator: # only 1 result
|
||||
if not result["success"]:
|
||||
raise(result["error"])
|
||||
result = next(results_iterator)
|
||||
else:
|
||||
output_manager.error(
|
||||
'Usage: %s post %s\n%s', BASENAME,
|
||||
st_post_options, st_post_help)
|
||||
return
|
||||
else:
|
||||
swift.post(container=container)
|
||||
result = swift.post(container=container)
|
||||
if not result["success"]:
|
||||
raise(result["error"])
|
||||
|
||||
except SwiftError as e:
|
||||
output_manager.error(e.value)
|
||||
|
@ -386,26 +386,68 @@ class TestShell(unittest.TestCase):
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_account(self, connection):
|
||||
argv = ["", "post"]
|
||||
connection.return_value.head_object.return_value = {}
|
||||
swiftclient.shell.main(argv)
|
||||
connection.return_value.post_account.assert_called_with(
|
||||
headers={}, response_dict={})
|
||||
|
||||
@mock.patch('swiftclient.shell.OutputManager.error')
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_account_bad_auth(self, connection, error):
|
||||
argv = ["", "post"]
|
||||
connection.return_value.post_account.side_effect = \
|
||||
swiftclient.ClientException('bad auth')
|
||||
swiftclient.shell.main(argv)
|
||||
error.assert_called_with('bad auth')
|
||||
|
||||
@mock.patch('swiftclient.shell.OutputManager.error')
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_account_not_found(self, connection, error):
|
||||
argv = ["", "post"]
|
||||
connection.return_value.post_account.side_effect = \
|
||||
swiftclient.ClientException('test', http_status=404)
|
||||
swiftclient.shell.main(argv)
|
||||
error.assert_called_with('Account not found')
|
||||
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_container(self, connection):
|
||||
argv = ["", "post", "container"]
|
||||
connection.return_value.head_object.return_value = {}
|
||||
swiftclient.shell.main(argv)
|
||||
connection.return_value.post_container.assert_called_with(
|
||||
'container', headers={}, response_dict={})
|
||||
|
||||
@mock.patch('swiftclient.shell.OutputManager.error')
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_container(self, connection):
|
||||
def test_post_container_bad_auth(self, connection, error):
|
||||
argv = ["", "post", "container"]
|
||||
connection.return_value.post_container.side_effect = \
|
||||
swiftclient.ClientException('bad auth')
|
||||
swiftclient.shell.main(argv)
|
||||
error.assert_called_with('bad auth')
|
||||
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_container_not_found_causes_put(self, connection):
|
||||
argv = ["", "post", "container"]
|
||||
connection.return_value.post_container.side_effect = \
|
||||
swiftclient.ClientException('test', http_status=404)
|
||||
swiftclient.shell.main(argv)
|
||||
self.assertEqual('container',
|
||||
connection.return_value.put_container.call_args[0][0])
|
||||
|
||||
@mock.patch('swiftclient.shell.OutputManager.error')
|
||||
def test_post_container_with_bad_name(self, error):
|
||||
argv = ["", "post", "conta/iner"]
|
||||
swiftclient.shell.main(argv)
|
||||
self.assertTrue(error.called)
|
||||
self.assertTrue(error.call_args[0][0].startswith('WARNING: / in'))
|
||||
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_container_with_options(self, connection):
|
||||
argv = ["", "post", "container",
|
||||
"--read-acl", "test2:tester2",
|
||||
"--write-acl", "test3:tester3 test4",
|
||||
"--sync-to", "othersite",
|
||||
"--sync-key", "secret",
|
||||
]
|
||||
connection.return_value.head_object.return_value = {}
|
||||
swiftclient.shell.main(argv)
|
||||
connection.return_value.post_container.assert_called_with(
|
||||
'container', headers={
|
||||
@ -420,13 +462,28 @@ class TestShell(unittest.TestCase):
|
||||
"--meta", "Color:Blue",
|
||||
"--header", "content-type:text/plain"
|
||||
]
|
||||
connection.return_value.head_object.return_value = {}
|
||||
swiftclient.shell.main(argv)
|
||||
connection.return_value.post_object.assert_called_with(
|
||||
'container', 'object', headers={
|
||||
'Content-Type': 'text/plain',
|
||||
'X-Object-Meta-Color': 'Blue'}, response_dict={})
|
||||
|
||||
@mock.patch('swiftclient.shell.OutputManager.error')
|
||||
@mock.patch('swiftclient.service.Connection')
|
||||
def test_post_object_bad_auth(self, connection, error):
|
||||
argv = ["", "post", "container", "object"]
|
||||
connection.return_value.post_object.side_effect = \
|
||||
swiftclient.ClientException("bad auth")
|
||||
swiftclient.shell.main(argv)
|
||||
error.assert_called_with('bad auth')
|
||||
|
||||
@mock.patch('swiftclient.shell.OutputManager.error')
|
||||
def test_post_object_too_many_args(self, error):
|
||||
argv = ["", "post", "container", "object", "bad_arg"]
|
||||
swiftclient.shell.main(argv)
|
||||
self.assertTrue(error.called)
|
||||
self.assertTrue(error.call_args[0][0].startswith('Usage'))
|
||||
|
||||
@mock.patch('swiftclient.shell.generate_temp_url')
|
||||
def test_temp_url(self, temp_url):
|
||||
argv = ["", "tempurl", "GET", "60", "/v1/AUTH_account/c/o",
|
||||
|
Loading…
x
Reference in New Issue
Block a user