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)