def case_delete_records(self, refresh_db_before): """ Create new record in ServerType table and then delete it.""" new_type = "New type" ServerTypeOp.add(new_type) type_obj = ServerTypeOp.get(name=new_type) self.assertTrue(len(type_obj) is 1) self.assertEqual(type_obj[0].name, new_type) ServerTypeOp.delete(type_obj[0]) type_obj = ServerTypeOp.get(name=new_type) self.assertFalse(type_obj)
def test_get_name(self, mock_servtype, mock_val_id, mock_val_name): """ Assumptions: - name argument given """ test_name = "test_name" ServerTypeOp.get(name=test_name) self.assertFalse(mock_val_id.called) self.assertTrue(mock_val_name.called) self.assertTrue(mock_servtype.query.filter_by.called) exp_calls = [mock.call(name=test_name), mock.call().all()] mock_servtype.query.filter_by.assert_has_calls(exp_calls)
def case_get_by_name_one_result(self, refresh_db_before): """ Create test records in ServerType table and get them by name.""" types_with_id = {"Type one": 1, "Type two": 2} for serv_type in types_with_id.keys(): ServerTypeOp.add(serv_type) for exp_type, exp_id in types_with_id.items(): type_obj = ServerTypeOp.get(name=exp_type) self.assertTrue(len(type_obj) is 1) self.assertEqual(exp_type, type_obj[0].name) self.assertEqual(exp_id, type_obj[0].id)
def case_add_records(self, refresh_db_before): """ Add new records to ServerType table.""" new_types = ["Type one", "Type two"] for serv_type in new_types: ServerTypeOp.add(serv_type) check_records = ServerTypeOp.get() self.assertEqual(len(check_records), len(new_types)) for record, exp_status in zip(check_records, new_types): self.assertTrue(isinstance(record, ServerType)) self.assertEqual(record.name, exp_status)
def case_update_record(self, refresh_db_before): """ Create test record in ServerType table and then update it.""" original_type = "TypeOrig" update_type = "TypeUpdate" original_type_obj = ServerTypeOp.add(original_type) updated_type_local = ServerTypeOp.update(original_type_obj, update_type) self.assertEqual(updated_type_local.name, update_type) updated_type_get = ServerTypeOp.get(name=update_type) self.assertTrue(len(updated_type_get) is 1) self.assertEqual(updated_type_get[0].name, update_type) self.assertEqual(updated_type_get[0].id, original_type_obj.id)
def resolve_type(cls, type_name): """ Find ServerType record according to given name. Return ID of type row. Args: type_name(str): ServerType record name Returns: type_id(int): ServerType record ID """ type_results = ServerTypeOp.get(name=type_name) if len(type_results) is not 1: raise ServerTypeNotFoundError(f'Not found type name: "{type_name}".') type_id = type_results[0].id return type_id