Как я могу проверить, кроме ошибок, используя side_effect в макете Python Unittest? - PullRequest
Новая
       24

Как я могу проверить, кроме ошибок, используя side_effect в макете Python Unittest?

0 голосов
/ 29 апреля 2019

Я младший разработчик и пытаюсь написать несколько юнит-тестов для наших конечных точек API. Ниже приведен класс, который я тестирую, и сам тест, который выполняется без каких-либо проблем. (Но я все еще сомневаюсь, что это бьет по моим методам). Мой вопрос: как я могу улучшить свой тест, а также убедиться, что он покрывает исключение (в данном случае ValueError, SystemError, Exception), используя side_effects (или любые другие лучшие предложения) из макета Python? Я читаю документацию по макету Python, но до сих пор не могу понять, как улучшить и протестировать исключения.

мы используем фотовспышку, python3.x,

--- Класс, который я тестирую: 

@USER_MOD.route('', methods=[HttpMethods.HTTP_POST])
@jwt_required
def create_user():
    """
    Private end point for creating users of the system
    :return: json
    """

    response = ""

    try:

        logger_obj.info("Request : POST : Create User: {} ".format(request.base_url))
        # validating input request
        if ValidationUtils.validate_request(request, "CREATE_USER"):
            logger_obj.debug("Request Validation Successful")
            response = UserService().create_new_user(request)
            logger_obj.debug("User Created: {} ".format(response))
            return jsonify(response), HTTPStatus.OK

    except ValueError as error:
        logger_obj.error("ValueError create_user() : {}".format(error))
        response = ApplicationUtils.create_api_response(status=ResponseConstants.MSG_ERROR, message=str(error))
        return jsonify(response), HTTPStatus.BAD_REQUEST
    except SystemError as error:
        logger_obj.error("SystemError create_user() : {}".format(error))
        response = ApplicationUtils.create_api_response(status=ResponseConstants.MSG_ERROR, message=str(error))
        return jsonify(response), HTTPStatus.INTERNAL_SERVER_ERROR
    except Exception as error:
        logger_obj.error("Exception create_user() : {}".format(error))
        response = ApplicationUtils.create_api_response(status=ResponseConstants.MSG_ERROR, message=str(error))
        return jsonify(response), HTTPStatus.UNAUTHORIZED
    finally:
        logger_obj.info("Response : POST : Create User: {} : {}".format(request.base_url, response))



--- Test for above class:

class UserApiTests(DataTestCase):  //(or i can use unittest.TestCase)

    def setUp(self):
        os.environ.update(self.config.to_env())
        self.flask_app = make_flask_app()
        self.client = self.flask_app.test_client()
        self.flask_app.testing = True

    @patch('usermgmt.app.services.user_service.UserService.create_new_user')
    def test_create_user(self, mock_create_new_user):

        # Given
        mock_create_new_user.return_value.status_code = 200

        mock_create_new_user.return_value = {
            "status": "SUCCESS"
        }
        data_to_post = {
            "name": "Test2",
            "email": "new-user2@entity.com",
            "entity_id": 1,
            "is_active": True,
            "product_roles": [
                {"product_id": 1, "role_id": 4},
                {"product_id": 2, "role_id": 4}
            ],
        }

        # When
        response = self.client.post('/api/usermgmt/users', data=json.dumps(data_to_post), headers={
            "Authorization": "Bearer {}".format(get_jwt(identity=self), "Content-Type: application/json")
        })

        data = response.data
        json_data = json.loads(data)

        # Then
        self.assertEqual(response.status_code, 200)
        self.assertEqual(json_data['status'], "SUCCESS")

1 Ответ

0 голосов
/ 30 апреля 2019

Во время практики я нашел этот подход довольно хорошим. 

class TestCreateUser(TestCase):
    def test_works_in_correct_case(self):
        # equal to your test_create_user

    @patch("usermgmt.app.services.user_service.UserService.create_new_user")
    def test_handles_value_error_in_create_new_user(self, mock_create_new_user):
        mock_create_new_user.side_effect = ValueError

        # Your preparation flow here

        response = self.client.post('/api/usermgmt/users') # simplified call here is just an example
        self.assertEqual(response.status_code, 400)
        # check if response body is what you've expected, etc.
...