def case_resolve_type_not_found(self, refresh_db_before):
""" Try to resolve ServerType row which name does not exist."""
exception_raised = False
try:
ServerOp.resolve_type("Test type")
except ServerTypeNotFoundError:
exception_raised = True
self.assertTrue(exception_raised)
def test_resolve_type_not_found(self, mock_type_op):
""" Assumptions:
- ServerTypeOp.get() returns no records
"""
mock_type_op.get.return_value = list()
with self.assertRaisesRegex(ServerTypeNotFoundError, "Not found"):
ServerOp.resolve_type("TypeName")
mock_status_get_calls = [mock.call(name="TypeName")]
mock_type_op.get.assert_has_calls(mock_status_get_calls)
def case_resolve_type_positive(self, refresh_db_before):
""" Create ServerType rows, then resolve their names into ID."""
types = {"Type one": 1, "Type two": 2}
for srv_type in types.keys():
ServerTypeOp.add(srv_type)
for srv_type_name, srv_type_id in types.items():
resolved_id = ServerOp.resolve_type(srv_type_name)
self.assertEqual(srv_type_id, resolved_id)
def test_resolve_type_positive(self, mock_type_op):
""" Assumptions:
ServerTypeOp.get() returns one record
"""
mock_type = mock.MagicMock()
mock_type.id = 1
mock_type_op.get.return_value = [mock_type]
status_id = ServerOp.resolve_type("TypeName")
self.assertEqual(mock_type.id, status_id)
mock_status_get_calls = [mock.call(name="TypeName")]
mock_type_op.get.assert_has_calls(mock_status_get_calls)