mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
50 lines
2.2 KiB
Python
50 lines
2.2 KiB
Python
import pytest
|
|
from mock import patch
|
|
|
|
from endpoints.api.subscribe_models_pre_oci import data_model
|
|
|
|
|
|
@pytest.mark.parametrize("username,repo_count", [("devtable", 3)])
|
|
def test_get_private_repo_count(username, repo_count):
|
|
with patch(
|
|
"endpoints.api.subscribe_models_pre_oci.get_private_repo_count"
|
|
) as mock_get_private_reop_count:
|
|
mock_get_private_reop_count.return_value = repo_count
|
|
count = data_model.get_private_repo_count(username)
|
|
|
|
mock_get_private_reop_count.assert_called_once_with(username)
|
|
assert count == repo_count
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"kind_name,target_username,metadata",
|
|
[("over_private_usage", "devtable", {"namespace": "devtable"})],
|
|
)
|
|
def test_create_unique_notification(kind_name, target_username, metadata):
|
|
with patch("endpoints.api.subscribe_models_pre_oci.get_user_or_org") as mock_get_user_or_org:
|
|
mock_get_user_or_org.return_value = {"username": target_username}
|
|
with patch(
|
|
"endpoints.api.subscribe_models_pre_oci.create_unique_notification"
|
|
) as mock_create_unique_notification:
|
|
data_model.create_unique_notification(kind_name, target_username, metadata)
|
|
|
|
mock_get_user_or_org.assert_called_once_with(target_username)
|
|
mock_create_unique_notification.assert_called_once_with(
|
|
kind_name, mock_get_user_or_org.return_value, metadata
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("target_username,kind_name", [("devtable", "over_private_usage")])
|
|
def test_delete_notifications_by_kind(target_username, kind_name):
|
|
with patch("endpoints.api.subscribe_models_pre_oci.get_user_or_org") as mock_get_user_or_org:
|
|
mock_get_user_or_org.return_value = {"username": target_username}
|
|
with patch(
|
|
"endpoints.api.subscribe_models_pre_oci.delete_notifications_by_kind"
|
|
) as mock_delete_notifications_by_kind:
|
|
data_model.delete_notifications_by_kind(target_username, kind_name)
|
|
|
|
mock_get_user_or_org.assert_called_once_with(target_username)
|
|
mock_delete_notifications_by_kind.assert_called_once_with(
|
|
mock_get_user_or_org.return_value, kind_name
|
|
)
|