mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
skshetry
cc4816f652
This is a simple refactoring which makes the caller to choose appropriate token instead of the studio.notify_refs making a decision for them.
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
from urllib.parse import urljoin
|
|
|
|
import pytest
|
|
from requests import Response
|
|
|
|
from dvc.utils.studio import STUDIO_URL, notify_refs
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"status_code",
|
|
[
|
|
200, # success
|
|
401, # should not fail on client errors
|
|
500, # should not fail even on server errors
|
|
],
|
|
)
|
|
def test_notify_refs(mocker, status_code):
|
|
response = Response()
|
|
response.status_code = status_code
|
|
|
|
mock_post = mocker.patch("requests.Session.post", return_value=response)
|
|
|
|
notify_refs(
|
|
"[email protected]:iterative/dvc.git",
|
|
"TOKEN",
|
|
pushed=["p1", "p2"],
|
|
removed=["r1", "r2"],
|
|
)
|
|
|
|
assert mock_post.called
|
|
assert mock_post.call_args == mocker.call(
|
|
urljoin(STUDIO_URL, "/webhook/dvc"),
|
|
json={
|
|
"repo_url": "[email protected]:iterative/dvc.git",
|
|
"client": "dvc",
|
|
"refs": {
|
|
"pushed": ["p1", "p2"],
|
|
"removed": ["r1", "r2"],
|
|
},
|
|
},
|
|
headers={"Authorization": "token TOKEN"},
|
|
timeout=5,
|
|
)
|