def case_update_all(self, refresh_db_before):
""" Update all fields."""
server_name = "TestServer"
server_name_two = "TestServerTwo"
ServerStatusOp.add("Status")
ServerStatusOp.add("StatusTwo")
ServerTypeOp.add("TypeOne")
ServerTypeOp.add("TypeTwo")
desc_one = "Desc one"
desc_two = "Desc two"
ips_one = IpOp.add("11.11.11.11")
ips_two = IpOp.add("22.22.22.22")
tag_one = TagOp.add("tag one")
tag_two = TagOp.add("tag two")
admin_one = AdminOp.add("Admin One")
admin_two = AdminOp.add("Admin Two")
srv = ServerOp.add(
server_name,
"Status",
"TypeOne",
description=desc_one,
ips=["11.11.11.11"],
tags=["tag one"],
admins=["Admin One"],
)
get_before_update = ServerOp.get()
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
self.assertEqual(get_before_update[0].name, server_name)
self.assertEqual(get_before_update[0].description, desc_one)
self.assertEqual(get_before_update[0].status.name, "Status")
self.assertEqual(get_before_update[0].type.name, "TypeOne")
self.assertEqual(get_before_update[0].ips, [ips_one])
self.assertEqual(get_before_update[0].tags, [tag_one])
self.assertEqual(get_before_update[0].admins, [admin_one])
ServerOp.update(
srv,
name=server_name_two,
description=desc_two,
srv_status="StatusTwo",
srv_type="TypeTwo",
ips=["22.22.22.22"],
tags=["tag two"],
admins=["Admin Two"],
)
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
self.assertEqual(get_srv[0].name, server_name_two)
self.assertEqual(get_srv[0].description, desc_two)
self.assertEqual(get_srv[0].status.name, "StatusTwo")
self.assertEqual(get_srv[0].type.name, "TypeTwo")
self.assertEqual(get_srv[0].ips, [ips_two])
self.assertEqual(get_srv[0].tags, [tag_two])
self.assertEqual(get_srv[0].admins, [admin_two])
def test_update_admins(
self,
mock_val_name,
mock_res_status,
mock_res_type,
mock_val_desc,
mock_res_ip,
mock_res_tag,
mock_res_adm,
mock_db,
):
""" Update admins in the existing record."""
mock_srv_obj = mock.MagicMock()
mock_srv_obj.admins = ["adm"]
mock_updated = ServerOp.update(mock_srv_obj, admins=["adm"])
self.assertEqual(mock_updated.admins, [mock_res_adm()])
self.assertFalse(mock_val_name.called)
self.assertFalse(mock_res_status.called)
self.assertFalse(mock_res_type.called)
self.assertFalse(mock_val_desc.called)
self.assertFalse(mock_res_ip.called)
self.assertFalse(mock_res_tag.called)
self.assertTrue(mock_res_adm.called)
self.assertTrue(mock_db.session.add.called)
self.assertTrue(mock_db.session.commit.called)
def test_update_description(
self,
mock_val_name,
mock_res_status,
mock_res_type,
mock_val_desc,
mock_res_ip,
mock_res_tag,
mock_res_adm,
mock_db,
):
""" Update description in the existing record."""
mock_srv_obj = mock.MagicMock()
mock_srv_obj.description = "DescOne"
mock_updated = ServerOp.update(mock_srv_obj, description="DescTwo")
self.assertEqual(mock_updated.description, "DescTwo")
self.assertFalse(mock_val_name.called)
self.assertFalse(mock_res_status.called)
self.assertFalse(mock_res_type.called)
self.assertTrue(mock_val_desc.called)
self.assertFalse(mock_res_ip.called)
self.assertFalse(mock_res_tag.called)
self.assertFalse(mock_res_adm.called)
self.assertTrue(mock_db.session.add.called)
self.assertTrue(mock_db.session.commit.called)
def test_update_srv_type(
self,
mock_val_name,
mock_res_status,
mock_res_type,
mock_val_desc,
mock_res_ip,
mock_res_tag,
mock_res_adm,
mock_db,
):
""" Update srv_type in the existing record."""
mock_res_type.return_value = 222
mock_srv_obj = mock.MagicMock()
mock_srv_obj.type_id = 111
mock_updated = ServerOp.update(mock_srv_obj, srv_type="SrvType")
self.assertEqual(mock_updated.type_id, 222)
self.assertFalse(mock_val_name.called)
self.assertFalse(mock_res_status.called)
self.assertTrue(mock_res_type.called)
self.assertFalse(mock_val_desc.called)
self.assertFalse(mock_res_ip.called)
self.assertFalse(mock_res_tag.called)
self.assertFalse(mock_res_adm.called)
self.assertTrue(mock_db.session.add.called)
self.assertTrue(mock_db.session.commit.called)
def test_update_name(
self,
mock_val_name,
mock_res_status,
mock_res_type,
mock_val_desc,
mock_res_ip,
mock_res_tag,
mock_res_adm,
mock_db,
):
""" Update name in the existing record."""
mock_srv_obj = mock.MagicMock()
mock_srv_obj.name = "TestNameOne"
mock_updated = ServerOp.update(mock_srv_obj, name="TestNameTwo")
self.assertEqual(mock_updated.name, "TestNameTwo")
self.assertTrue(mock_val_name.called)
self.assertFalse(mock_res_status.called)
self.assertFalse(mock_res_type.called)
self.assertFalse(mock_val_desc.called)
self.assertFalse(mock_res_ip.called)
self.assertFalse(mock_res_tag.called)
self.assertFalse(mock_res_adm.called)
self.assertTrue(mock_db.session.add.called)
self.assertTrue(mock_db.session.commit.called)
def case_update_name(self, refresh_db_before):
""" Update server name."""
server_name = "TestServer"
server_second_name = "TestServerTwo"
ServerStatusOp.add("Status")
ServerTypeOp.add("TypeOne")
srv = ServerOp.add(server_name, "Status", "TypeOne")
get_before_update = ServerOp.get(name=server_name)
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
ServerOp.update(srv, name=server_second_name)
get_srv = ServerOp.get(name=server_second_name)
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
def case_update_srv_type(self, refresh_db_before):
""" Update server type."""
server_name = "TestServer"
ServerStatusOp.add("Status")
ServerTypeOp.add("TypeOne")
ServerTypeOp.add("TypeTwo")
srv = ServerOp.add(server_name, "Status", "TypeOne")
get_before_update = ServerOp.get()
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
self.assertEqual(get_before_update[0].type.name, "TypeOne")
ServerOp.update(srv, srv_type="TypeTwo")
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
self.assertEqual(get_srv[0].type.name, "TypeTwo")
def case_update_ips(self, refresh_db_before):
""" Update ip addresses. """
server_name = "TestServer"
ServerStatusOp.add("Status")
ServerTypeOp.add("Type")
ips_one = IpOp.add("11.11.11.11")
ips_two = IpOp.add("22.22.22.22")
srv = ServerOp.add(server_name, "Status", "Type", ips=["11.11.11.11"])
get_before_update = ServerOp.get()
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
self.assertEqual(get_before_update[0].ips, [ips_one])
ServerOp.update(srv, ips=["22.22.22.22"])
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
self.assertEqual(get_srv[0].ips, [ips_two])
def case_update_description(self, refresh_db_before):
""" Update server description."""
server_name = "TestServer"
ServerStatusOp.add("Status")
ServerTypeOp.add("TypeOne")
desc_one = "Desc one"
desc_two = "Desc two"
srv = ServerOp.add(server_name, "Status", "TypeOne", desc_one)
get_before_update = ServerOp.get()
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
self.assertEqual(get_before_update[0].description, desc_one)
ServerOp.update(srv, description=desc_two)
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
self.assertEqual(get_srv[0].description, desc_two)
def case_update_admins(self, refresh_db_before):
""" Update admins."""
server_name = "TestServer"
ServerStatusOp.add("Status")
ServerTypeOp.add("Type")
admin_one = AdminOp.add("Admin One")
admin_two = AdminOp.add("Admin Two")
srv = ServerOp.add(server_name, "Status", "Type", admins=["Admin One"])
get_before_update = ServerOp.get()
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
self.assertEqual(get_before_update[0].admins, [admin_one])
ServerOp.update(srv, admins=["Admin Two"])
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
self.assertEqual(get_srv[0].admins, [admin_two])
def case_update_tags(self, refresh_db_before):
""" Update tags."""
server_name = "TestServer"
ServerStatusOp.add("Status")
ServerTypeOp.add("Type")
tag_one = TagOp.add("tag one")
tag_two = TagOp.add("tag two")
srv = ServerOp.add(server_name, "Status", "Type", tags=["tag one"])
get_before_update = ServerOp.get()
self.assertTrue(len(get_before_update) is 1)
self.assertEqual(get_before_update[0].id, 1)
self.assertEqual(get_before_update[0].tags, [tag_one])
ServerOp.update(srv, tags=["tag two"])
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].id, 1)
self.assertEqual(get_srv[0].tags, [tag_two])