def test_show_quota_invalid_args(self, stderr):
with self.assertRaises(SystemExit) as cm:
admin_plugin.show_quota(["-s", self.service_name])
exc = cm.exception
self.assertEqual(2, exc.code)
stderr.write.assert_called_with(
"show-quota: error: argument -t/--team is required\n")
def test_show_quota_request_failure(self, stderr, Request, urlopen):
lines = []
stderr.write.side_effect = lambda data, **kw: lines.append(data)
request = mock.Mock()
Request.return_value = request
result = mock.Mock()
result.getcode.return_value = 400
result.read.return_value = "something went wrong"
urlopen.return_value = result
with self.assertRaises(SystemExit) as cm:
admin_plugin.show_quota(["-s", self.service_name, "-t", "myteam"])
exc = cm.exception
self.assertEqual(1, exc.code)
self.assertEqual("ERROR: something went wrong\n", "".join(lines))
def test_show_quota_request_failure(self, stderr, Request, urlopen):
lines = []
stderr.write.side_effect = lambda data, **kw: lines.append(data)
request = mock.Mock()
Request.return_value = request
result = mock.Mock()
result.getcode.return_value = 400
result.read.return_value = "something went wrong"
urlopen.return_value = result
with self.assertRaises(SystemExit) as cm:
admin_plugin.show_quota(["-s", self.service_name, "-t", "myteam"])
exc = cm.exception
self.assertEqual(1, exc.code)
self.assertEqual("ERROR: something went wrong\n", "".join(lines))
def test_show_quota(self, stdout, Request, urlopen):
lines = []
stdout.write.side_effect = lambda data, **kw: lines.append(data)
quota_data = u"""{"used":["inst1", "inst2", "inst3"], "quota": 10}"""
request = mock.Mock()
Request.return_value = request
result = mock.Mock()
result.getcode.return_value = 200
result.read.return_value = quota_data
urlopen.return_value = result
admin_plugin.show_quota(["-s", self.service_name, "-t", "myteam"])
Request.assert_called_with(self.target + "services/proxy/service/rpaas?" + "callback=/admin/quota/myteam")
request.add_header.assert_any_call("Authorization", "bearer " + self.token)
self.assertEqual("GET", request.get_method())
expected_output = u"""Quota usage: 3/10.\n"""
self.assertEqual(expected_output, "".join(lines))
def test_show_quota(self, stdout, Request, urlopen):
lines = []
stdout.write.side_effect = lambda data, **kw: lines.append(data)
quota_data = u"""{"used":["inst1", "inst2", "inst3"], "quota": 10}"""
request = mock.Mock()
Request.return_value = request
result = mock.Mock()
result.getcode.return_value = 200
result.read.return_value = quota_data
urlopen.return_value = result
admin_plugin.show_quota(["-s", self.service_name, "-t", "myteam"])
Request.assert_called_with(self.target +
"services/proxy/service/rpaas?" +
"callback=/admin/quota/myteam")
request.add_header.assert_any_call("Authorization", "bearer " + self.token)
self.assertEqual("GET", request.get_method())
expected_output = u"""Quota usage: 3/10.\n"""
self.assertEqual(expected_output, "".join(lines))
def test_show_quota_invalid_args(self, stderr):
with self.assertRaises(SystemExit) as cm:
admin_plugin.show_quota(["-s", self.service_name])
exc = cm.exception
self.assertEqual(2, exc.code)
stderr.write.assert_called_with("show-quota: error: argument -t/--team is required\n")
