def test_command_create_failed(self):
instance = base.FakeResource(name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(['fakeresource', 'create', path])
self.assertRaises(Exception, self.branch.commands['create'].run,
args)
finally:
os.close(fd)
os.unlink(path)
def test_command_update_id_mismatch(self):
instance = base.FakeResource(id="789", name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
["fakeresource", "update", "123", path])
self.assertRaises(Exception, self.branch.commands["update"].run,
args)
finally:
os.close(fd)
os.unlink(path)
def test_command_create(self):
instance = base.FakeResource(name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(['fakeresource', 'create', path])
self.assertEqual(args.func,
self.branch.commands['create'].run_and_print)
instance = self.branch.commands['create'].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
finally:
os.close(fd)
os.unlink(path)
def test_command_update(self):
instance = base.FakeResource(id="123", name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
["fakeresource", "update", "123", path])
self.assertEqual(args.func,
self.branch.commands["update"].run_and_print)
instance = self.branch.commands["update"].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
finally:
os.close(fd)
os.unlink(path)
class TestResourceCommand(unittest2.TestCase):
def __init__(self, *args, **kwargs):
super(TestResourceCommand, self).__init__(*args, **kwargs)
self.parser = argparse.ArgumentParser()
self.subparsers = self.parser.add_subparsers()
self.branch = resource.ResourceBranch(
base.FakeResource, 'Test Command', base.FakeApp(), self.subparsers)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps(base.RESOURCES), 200, 'OK')))
def test_command_list(self):
args = self.parser.parse_args(['fakeresource', 'list'])
self.assertEqual(args.func, self.branch.commands['list'].run_and_print)
instances = self.branch.commands['list'].run(args)
actual = [instance.serialize() for instance in instances]
expected = json.loads(json.dumps(base.RESOURCES))
self.assertListEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse('', 500, 'INTERNAL SERVER ERROR')))
def test_command_list_failed(self):
args = self.parser.parse_args(['fakeresource', 'list'])
self.assertRaises(Exception, self.branch.commands['list'].run, args)
@mock.patch.object(
models.ResourceManager, 'get_by_name',
mock.MagicMock(return_value=None))
@mock.patch.object(
models.ResourceManager, 'get_by_id',
mock.MagicMock(return_value=base.FakeResource(**base.RESOURCES[0])))
def test_command_get_by_id(self):
args = self.parser.parse_args(['fakeresource', 'get', '123'])
self.assertEqual(args.func, self.branch.commands['get'].run_and_print)
instance = self.branch.commands['get'].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps(base.RESOURCES[0]), 200, 'OK')))
def test_command_get(self):
args = self.parser.parse_args(['fakeresource', 'get', 'abc'])
self.assertEqual(args.func, self.branch.commands['get'].run_and_print)
instance = self.branch.commands['get'].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse('', 404, 'NOT FOUND')))
def test_command_get_404(self):
args = self.parser.parse_args(['fakeresource', 'get', 'cba'])
self.assertEqual(args.func, self.branch.commands['get'].run_and_print)
self.assertRaises(resource.ResourceNotFoundError,
self.branch.commands['get'].run,
args)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse('', 500, 'INTERNAL SERVER ERROR')))
def test_command_get_failed(self):
args = self.parser.parse_args(['fakeresource', 'get', 'cba'])
self.assertRaises(Exception, self.branch.commands['get'].run, args)
@mock.patch.object(
httpclient.HTTPClient, 'post',
mock.MagicMock(return_value=base.FakeResponse(json.dumps(base.RESOURCES[0]), 200, 'OK')))
def test_command_create(self):
instance = base.FakeResource(name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(['fakeresource', 'create', path])
self.assertEqual(args.func,
self.branch.commands['create'].run_and_print)
instance = self.branch.commands['create'].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient, 'post',
mock.MagicMock(return_value=base.FakeResponse('', 500, 'INTERNAL SERVER ERROR')))
def test_command_create_failed(self):
instance = base.FakeResource(name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(['fakeresource', 'create', path])
self.assertRaises(Exception,
self.branch.commands['create'].run,
args)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps([base.RESOURCES[0]]), 200, 'OK',
{})))
@mock.patch.object(
httpclient.HTTPClient, 'put',
mock.MagicMock(return_value=base.FakeResponse(json.dumps(base.RESOURCES[0]), 200, 'OK')))
def test_command_update(self):
instance = base.FakeResource(id='123', name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
['fakeresource', 'update', '123', path])
self.assertEqual(args.func,
self.branch.commands['update'].run_and_print)
instance = self.branch.commands['update'].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps([base.RESOURCES[0]]), 200, 'OK')))
@mock.patch.object(
httpclient.HTTPClient, 'put',
mock.MagicMock(return_value=base.FakeResponse('', 500, 'INTERNAL SERVER ERROR')))
def test_command_update_failed(self):
instance = base.FakeResource(id='123', name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
['fakeresource', 'update', '123', path])
self.assertRaises(Exception,
self.branch.commands['update'].run,
args)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps([base.RESOURCES[0]]), 200, 'OK')))
def test_command_update_id_mismatch(self):
instance = base.FakeResource(id='789', name='abc')
fd, path = tempfile.mkstemp(suffix='.json')
try:
with open(path, 'a') as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
['fakeresource', 'update', '123', path])
self.assertRaises(Exception,
self.branch.commands['update'].run,
args)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps([base.RESOURCES[0]]), 200, 'OK',
{})))
@mock.patch.object(
httpclient.HTTPClient, 'delete',
mock.MagicMock(return_value=base.FakeResponse('', 204, 'NO CONTENT')))
def test_command_delete(self):
args = self.parser.parse_args(['fakeresource', 'delete', 'abc'])
self.assertEqual(args.func,
self.branch.commands['delete'].run_and_print)
self.branch.commands['delete'].run(args)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse('', 404, 'NOT FOUND')))
def test_command_delete_404(self):
args = self.parser.parse_args(['fakeresource', 'delete', 'cba'])
self.assertEqual(args.func,
self.branch.commands['delete'].run_and_print)
self.assertRaises(resource.ResourceNotFoundError,
self.branch.commands['delete'].run,
args)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps([base.RESOURCES[0]]), 200, 'OK')))
@mock.patch.object(
httpclient.HTTPClient, 'delete',
mock.MagicMock(return_value=base.FakeResponse('', 500, 'INTERNAL SERVER ERROR')))
def test_command_delete_failed(self):
args = self.parser.parse_args(['fakeresource', 'delete', 'cba'])
self.assertRaises(Exception, self.branch.commands['delete'].run, args)
def test_resource_serialize(self):
instance = base.FakeResource(id='123', name='abc')
self.assertDictEqual(instance.serialize(), base.RESOURCES[0])
class TestResourceCommand(unittest2.TestCase):
def __init__(self, *args, **kwargs):
super(TestResourceCommand, self).__init__(*args, **kwargs)
self.parser = argparse.ArgumentParser()
self.subparsers = self.parser.add_subparsers()
self.branch = resource.ResourceBranch(base.FakeResource,
"Test Command", base.FakeApp(),
self.subparsers)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps(base.RESOURCES), 200, "OK")),
)
def test_command_list(self):
args = self.parser.parse_args(["fakeresource", "list"])
self.assertEqual(args.func, self.branch.commands["list"].run_and_print)
instances = self.branch.commands["list"].run(args)
actual = [instance.serialize() for instance in instances]
expected = json.loads(json.dumps(base.RESOURCES))
self.assertListEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(
return_value=base.FakeResponse("", 500, "INTERNAL SERVER ERROR")),
)
def test_command_list_failed(self):
args = self.parser.parse_args(["fakeresource", "list"])
self.assertRaises(Exception, self.branch.commands["list"].run, args)
@mock.patch.object(models.ResourceManager, "get_by_name",
mock.MagicMock(return_value=None))
@mock.patch.object(
models.ResourceManager,
"get_by_id",
mock.MagicMock(return_value=base.FakeResource(**base.RESOURCES[0])),
)
def test_command_get_by_id(self):
args = self.parser.parse_args(["fakeresource", "get", "123"])
self.assertEqual(args.func, self.branch.commands["get"].run_and_print)
instances = self.branch.commands["get"].run(args)
actual = instances[0].serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps(base.RESOURCES[0]), 200, "OK")),
)
def test_command_get_single(self):
args = self.parser.parse_args(["fakeresource", "get", "abc"])
self.assertEqual(args.func, self.branch.commands["get"].run_and_print)
instances = self.branch.commands["get"].run(args)
self.assertEqual(len(instances), 1)
instance = instances[0]
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(side_effect=[
base.FakeResponse(json.dumps(base.RESOURCES[0]), 200, "OK"),
base.FakeResponse(json.dumps(base.RESOURCES[1]), 200, "OK"),
]),
)
def test_command_get_multiple(self):
args = self.parser.parse_args(["fakeresource", "get", "abc", "def"])
self.assertEqual(args.func, self.branch.commands["get"].run_and_print)
instances = self.branch.commands["get"].run(args)
self.assertEqual(len(instances), 2)
actual = instances[0].serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
actual = instances[1].serialize()
expected = json.loads(json.dumps(base.RESOURCES[1]))
self.assertEqual(actual, expected)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse("", 404, "NOT FOUND")),
)
def test_command_get_404(self):
args = self.parser.parse_args(["fakeresource", "get", "cba"])
self.assertEqual(args.func, self.branch.commands["get"].run_and_print)
self.assertRaises(resource.ResourceNotFoundError,
self.branch.commands["get"].run, args)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(
return_value=base.FakeResponse("", 500, "INTERNAL SERVER ERROR")),
)
def test_command_get_failed(self):
args = self.parser.parse_args(["fakeresource", "get", "cba"])
self.assertRaises(Exception, self.branch.commands["get"].run, args)
@mock.patch.object(
httpclient.HTTPClient,
"post",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps(base.RESOURCES[0]), 200, "OK")),
)
def test_command_create(self):
instance = base.FakeResource(name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(["fakeresource", "create", path])
self.assertEqual(args.func,
self.branch.commands["create"].run_and_print)
instance = self.branch.commands["create"].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient,
"post",
mock.MagicMock(
return_value=base.FakeResponse("", 500, "INTERNAL SERVER ERROR")),
)
def test_command_create_failed(self):
instance = base.FakeResource(name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(["fakeresource", "create", path])
self.assertRaises(Exception, self.branch.commands["create"].run,
args)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps([base.RESOURCES[0]]), 200, "OK", {})),
)
@mock.patch.object(
httpclient.HTTPClient,
"put",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps(base.RESOURCES[0]), 200, "OK")),
)
def test_command_update(self):
instance = base.FakeResource(id="123", name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
["fakeresource", "update", "123", path])
self.assertEqual(args.func,
self.branch.commands["update"].run_and_print)
instance = self.branch.commands["update"].run(args)
actual = instance.serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps([base.RESOURCES[0]]), 200, "OK")),
)
@mock.patch.object(
httpclient.HTTPClient,
"put",
mock.MagicMock(
return_value=base.FakeResponse("", 500, "INTERNAL SERVER ERROR")),
)
def test_command_update_failed(self):
instance = base.FakeResource(id="123", name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
["fakeresource", "update", "123", path])
self.assertRaises(Exception, self.branch.commands["update"].run,
args)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps([base.RESOURCES[0]]), 200, "OK")),
)
def test_command_update_id_mismatch(self):
instance = base.FakeResource(id="789", name="abc")
fd, path = tempfile.mkstemp(suffix=".json")
try:
with open(path, "a") as f:
f.write(json.dumps(instance.serialize(), indent=4))
args = self.parser.parse_args(
["fakeresource", "update", "123", path])
self.assertRaises(Exception, self.branch.commands["update"].run,
args)
finally:
os.close(fd)
os.unlink(path)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps([base.RESOURCES[0]]), 200, "OK", {})),
)
@mock.patch.object(
httpclient.HTTPClient,
"delete",
mock.MagicMock(return_value=base.FakeResponse("", 204, "NO CONTENT")),
)
def test_command_delete(self):
args = self.parser.parse_args(["fakeresource", "delete", "abc"])
self.assertEqual(args.func,
self.branch.commands["delete"].run_and_print)
self.branch.commands["delete"].run(args)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse("", 404, "NOT FOUND")),
)
def test_command_delete_404(self):
args = self.parser.parse_args(["fakeresource", "delete", "cba"])
self.assertEqual(args.func,
self.branch.commands["delete"].run_and_print)
self.assertRaises(resource.ResourceNotFoundError,
self.branch.commands["delete"].run, args)
@mock.patch.object(
httpclient.HTTPClient,
"get",
mock.MagicMock(return_value=base.FakeResponse(
json.dumps([base.RESOURCES[0]]), 200, "OK")),
)
@mock.patch.object(
httpclient.HTTPClient,
"delete",
mock.MagicMock(
return_value=base.FakeResponse("", 500, "INTERNAL SERVER ERROR")),
)
def test_command_delete_failed(self):
args = self.parser.parse_args(["fakeresource", "delete", "cba"])
self.assertRaises(Exception, self.branch.commands["delete"].run, args)
@mock.patch.object(
models.ResourceManager,
"get_by_id",
mock.MagicMock(return_value=base.FakeResource(**base.RESOURCES[0])),
)
def test_command_get_unicode_primary_key(self):
args = self.parser.parse_args(
["fakeresource", "get", "examples.test_rule_utf8_náme"])
self.assertEqual(args.func, self.branch.commands["get"].run_and_print)
instances = self.branch.commands["get"].run(args)
actual = instances[0].serialize()
expected = json.loads(json.dumps(base.RESOURCES[0]))
self.assertEqual(actual, expected)
