Skip to content

Commit 645acc3

Browse files
committed
correctly mock default repo
1 parent c78c59f commit 645acc3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit/test_tasks.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -107,9 +107,9 @@ def test_save_performance_project_types():
107107
# Check projects w/ invalid properties
108108

109109

110-
@mock.patch.object(ModelRepository, 'get_repository')
110+
@mock.patch.object(ModelRepository, 'list_repositories')
111111
@mock.patch.object(ModelRepository, 'get_project')
112-
def test_register_model_403_error(get_project, get_repository):
112+
def test_register_model_403_error(get_project, list_repositories):
113113
"""Verify HTTP 403 is converted to a user-friendly error.
114114
115115
Depending on environment configuration, this can happen when attempting to
@@ -123,15 +123,15 @@ def test_register_model_403_error(get_project, get_repository):
123123
from sasctl.tasks import register_model
124124

125125
get_project.return_value = {'name': 'Project Name'}
126-
get_repository.side_effect = HTTPError(None, 403, None, None, None)
126+
list_repositories.side_effect = HTTPError(None, 403, None, None, None)
127127

128128
# HTTP 403 error when getting repository should throw a user-friendly
129129
# AuthorizationError
130130
with pytest.raises(AuthorizationError):
131131
register_model(None, 'model name', 'project name')
132132

133133
# All other errors should be bubbled up
134-
get_repository.side_effect = HTTPError(None, 404, None, None, None)
134+
list_repositories.side_effect = HTTPError(None, 404, None, None, None)
135135
with pytest.raises(HTTPError):
136136
register_model(None, 'model name', 'project name')
137137

0 commit comments

Comments
 (0)