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])