def case_get_by_ip(self, refresh_db_before): """ Get server row with ip keyword.""" server_name = "TestServer" server_second_name = "TestServerTwo" server_status = "TestStatus" server_type = "TestType" ServerStatusOp.add(server_status) ServerTypeOp.add(server_type) ips = ["11.11.11.11", "22.22.22.22"] for ip in ips: IpOp.add(ip) srv_one = ServerOp.add(server_name, server_status, server_type, ips=[ips[0]]) srv_two = ServerOp.add(server_second_name, server_status, server_type, ips=[ips[1]]) get_first_ip = ServerOp.get(ip=ips[0]) self.assertTrue(len(get_first_ip) is 1) self.assertEqual(get_first_ip[0], srv_one) self.assertNotEqual(get_first_ip[0], srv_two) self.assertTrue(len(get_first_ip[0].ips) is 1) self.assertEqual(get_first_ip[0].ips[0].address, ips[0])
def case_add_with_ips(self, refresh_db_before): """ Create new Server row with ips many-to-many relation.""" server_name = "TestServer" server_status = "TestStatus" server_type = "TestType" ips = ["11.11.11.11", "22.22.22.22", "33.33.33.33"] ServerStatusOp.add(server_status) ServerTypeOp.add(server_type) for ip in ips: IpOp.add(ip) new_server = ServerOp.add(server_name, server_status, server_type, ips=ips) get_servers = ServerOp.get() self.assertTrue(len(get_servers) is 1) self.assertEqual(get_servers[0].description, None) self.assertEqual(get_servers[0].id, 1) self.assertEqual(get_servers[0].name, server_name) self.assertEqual(get_servers[0], new_server) for srv_ip, exp_ip in zip(get_servers[0].ips, ips): self.assertEqual(srv_ip.address, exp_ip)
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 case_resolve_ip_positive(self, refresh_db_before): """ Try to resolve Ip row.""" ips = ["11.11.11.11", "22.22.22.22", "33.33.33.33"] for ip in ips: IpOp.add(address=ip) for id, ip in enumerate(ips): ip_obj = ServerOp.resolve_ip(ip) self.assertEqual(ip_obj.id, id + 1) self.assertEqual(ip_obj.address, ip) self.assertTrue(isinstance(ip_obj, Ip))
def case_delete_records(self, refresh_db_before): """ Create new record in Ip table and then delete it.""" new_address = "0.0.0.0" IpOp.add(new_address) ip_obj = IpOp.get(address=new_address) self.assertTrue(len(ip_obj) is 1) self.assertEqual(ip_obj[0].address, new_address) IpOp.delete(ip_obj[0]) ip_obj = IpOp.get(address=new_address) self.assertFalse(ip_obj)
def case_add_records(self, refresh_db_before): """ Add new records to Ip table.""" new_addresses = ["0.0.0.0", "11.11.11.11", "222.222.222.222"] for address in new_addresses: IpOp.add(address) check_records = IpOp.get() self.assertEqual(len(check_records), len(new_addresses)) for record, exp_name in zip(check_records, new_addresses): self.assertTrue(isinstance(record, Ip)) self.assertEqual(record.address, exp_name)
def case_get_by_address_one_result(self, refresh_db_before): """ Create test records in Ip table and get them by name.""" addresses_with_id = { "0.0.0.0": 1, "11.11.11.11": 2, "222.222.222.222": 3 } for address in addresses_with_id.keys(): IpOp.add(address) for exp_address, exp_id in addresses_with_id.items(): ip_obj = IpOp.get(address=exp_address) self.assertTrue(len(ip_obj) is 1) self.assertEqual(exp_address, ip_obj[0].address) self.assertEqual(exp_id, ip_obj[0].id)
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_record(self, refresh_db_before): """ Create test record in Ip table and then update it.""" original_address = "0.0.0.0" update_address = "11.11.11.11" original_ip = IpOp.add(original_address) updated_ip_local = IpOp.update(original_ip, update_address) self.assertEqual(updated_ip_local.address, update_address) updated_ip_get = IpOp.get(address=update_address) self.assertTrue(len(updated_ip_get) is 1) self.assertEqual(updated_ip_get[0].address, update_address) self.assertEqual(updated_ip_get[0].id, original_ip.id)
def test_add(self, mock_ip, mock_val_address, mock_db): """ Assumptions: - add method run """ new_name = "Test Name" new_ip = IpOp.add(new_name) exp_calls = [mock.call(new_name)] mock_ip.assert_has_calls(exp_calls) self.assertEqual(new_ip, mock_ip()) self.assertTrue(mock_val_address.called) db_exp_calls = [ mock.call.session.add(new_ip), mock.call.session.commit() ] mock_db.assert_has_calls(db_exp_calls)
def case_get_by_all(self, refresh_db_before): """ Get server row with both srv_status srv_type keywords. """ server_name = "TestServer" server_second_name = "TestServerTwo" server_third_name = "TestServerThree" server_fourth_name = "TestServerFour" ServerStatusOp.add("StatusOne") ServerStatusOp.add("StatusTwo") ServerTypeOp.add("TypeOne") ServerTypeOp.add("TypeTwo") IpOp.add("11.11.11.11") IpOp.add("22.22.22.22") TagOp.add("Tag one") TagOp.add("Tag two") AdminOp.add("Admin One") AdminOp.add("Admin Two") srv_one = ServerOp.add( server_name, "StatusOne", "TypeOne", ips=["11.11.11.11"], tags=["Tag one"], admins=["Admin One"], ) ServerOp.add( server_second_name, "StatusTwo", "TypeTwo", ips=["22.22.22.22"], tags=["Tag two"], admins=["Admin Two"], ) ServerOp.add( server_third_name, "StatusOne", "TypeTwo", ips=["22.22.22.22"], tags=["Tag two"], admins=["Admin Two"], ) srv_four = ServerOp.add( server_fourth_name, "StatusOne", "TypeOne", ips=["11.11.11.11"], tags=["Tag one"], admins=["Admin One"], ) get_by_all = ServerOp.get( srv_status="StatusOne", srv_type="TypeOne", ip="11.11.11.11", tags=["Tag one"], admins=["Admin One"], ) self.assertTrue(len(get_by_all) is 2) self.assertEqual(get_by_all[0].id, 1) self.assertEqual(get_by_all[1].id, 4) self.assertEqual(get_by_all[0], srv_one) self.assertEqual(get_by_all[1], srv_four)