sunscrapers /
djoser
| @@ 69-89 (lines=21) @@ | ||
| 66 | } |
|
| 67 | ||
| 68 | ||
| 69 | @pytest.mark.django_db(transaction=False) |
|
| 70 | def test_valid_pipeline(test_user): |
|
| 71 | request = mock.MagicMock() |
|
| 72 | request.user = test_user |
|
| 73 | request.data = {'email': 'new@localhost'} |
|
| 74 | username = getattr(test_user, User.USERNAME_FIELD) |
|
| 75 | ||
| 76 | pipeline = pipelines.user_update.Pipeline(request) |
|
| 77 | with catch_signal(signals.user_updated) as handler: |
|
| 78 | result = pipeline.run() |
|
| 79 | ||
| 80 | handler.assert_called_once_with( |
|
| 81 | sender=mock.ANY, |
|
| 82 | signal=signals.user_updated, |
|
| 83 | user=result['user'], |
|
| 84 | request=request |
|
| 85 | ) |
|
| 86 | ||
| 87 | assert test_user.email == request.data['email'] |
|
| 88 | assert 'response_data' in result |
|
| 89 | assert result['response_data'] == { |
|
| 90 | 'id': 1, |
|
| 91 | 'email': test_user.email, |
|
| 92 | User.USERNAME_FIELD: username, |
|
| @@ 96-113 (lines=18) @@ | ||
| 93 | assert result['response_data'] == {User.USERNAME_FIELD: username, 'id': 1} |
|
| 94 | ||
| 95 | ||
| 96 | @pytest.mark.django_db(transaction=False) |
|
| 97 | def test_valid_pipeline(): |
|
| 98 | request = mock.MagicMock() |
|
| 99 | request.data = {User.USERNAME_FIELD: 'test', 'password': 'testing123'} |
|
| 100 | ||
| 101 | pipeline = pipelines.user_create.Pipeline(request) |
|
| 102 | with catch_signal(signals.user_created) as handler: |
|
| 103 | result = pipeline.run() |
|
| 104 | ||
| 105 | handler.assert_called_once_with( |
|
| 106 | sender=mock.ANY, |
|
| 107 | signal=signals.user_created, |
|
| 108 | user=result['user'], |
|
| 109 | request=request |
|
| 110 | ) |
|
| 111 | ||
| 112 | assert 'response_data' in result |
|
| 113 | assert result['response_data'] == { |
|
| 114 | User.USERNAME_FIELD: request.data[User.USERNAME_FIELD], 'id': 1 |
|
| 115 | } |
|
| 116 | ||
