def test_block_remove(self, stdout, Request, urlopen): request = Request.return_value urlopen.return_value.getcode.return_value = 200 self.set_envs() self.addCleanup(self.delete_envs) plugin.block(['remove', '-s', 'myservice', '-i', 'myinst', '-b', 'http']) Request.assert_called_with(self.target + "services/myservice/proxy/myinst?" + "callback=/resources/myinst/block/http") request.add_header.assert_any_call("Authorization", "bearer " + self.token) request.add_header.assert_any_call("Content-Type", "application/x-www-form-urlencoded") self.assertEqual(request.get_method(), 'DELETE') urlopen.assert_called_with(request) stdout.write.assert_called_with("block successfully removed\n")
def test_block_add_file_content(self, stdout, Request, urlopen): request = Request.return_value urlopen.return_value.getcode.return_value = 200 self.set_envs() self.addCleanup(self.delete_envs) path = os.path.join(os.path.dirname(__file__), "testdata", "block_http") plugin.block(['add', '-s', 'myservice', '-i', 'myinst', '-b', 'http', '-c', '@'+path]) Request.assert_called_with(self.target + "services/myservice/proxy/myinst?" + "callback=/resources/myinst/block") request.add_header.assert_any_call("Authorization", "bearer " + self.token) request.add_header.assert_any_call("Content-Type", "application/x-www-form-urlencoded") request.add_data.assert_called_with("content=content%0A&block_name=http") self.assertEqual(request.get_method(), 'POST') urlopen.assert_called_with(request) stdout.write.assert_called_with("block successfully added\n")
def test_block_list(self, stdout, Request, urlopen): request = Request.return_value urlopen.return_value.getcode.return_value = 200 urlopen.return_value.read.return_value = '{"_id": "myinst", ' + \ '"blocks": [{"block_name": "http", "content": "my content"}, ' + \ '{"block_name": "server", "content": "server content"}]}' self.set_envs() self.addCleanup(self.delete_envs) plugin.block(['list', '-s', 'myservice', '-i', 'myinst']) Request.assert_called_with(self.target + "services/myservice/proxy/myinst?" + "callback=/resources/myinst/block") request.add_header.assert_any_call("Authorization", "bearer " + self.token) self.assertEqual(request.get_method(), 'GET') urlopen.assert_called_with(request) stdout.write.assert_called_with("""block_name: content http: my content server: server content """)