400b346953
this adds tests for the pro oci model Issue: https://coreosdev.atlassian.net/browse/QUAY-632 - [ ] It works! - [ ] Comments provide sufficient explanations for the next contributor - [ ] Tests cover changes and corner cases - [ ] Follows Quay syntax patterns and format
103 lines
4.1 KiB
Python
103 lines
4.1 KiB
Python
import pytest
|
|
from endpoints.api.tag_models_interface import RepositoryTagHistory, Tag
|
|
from mock import Mock
|
|
|
|
from data import model
|
|
from endpoints.api.tag_models_pre_oci import pre_oci_model
|
|
|
|
EMPTY_REPOSITORY = 'empty_repository'
|
|
EMPTY_NAMESPACE = 'empty_namespace'
|
|
BAD_REPOSITORY_NAME = 'bad_repository_name'
|
|
BAD_NAMESPACE_NAME = 'bad_namespace_name'
|
|
|
|
|
|
@pytest.fixture
|
|
def get_monkeypatch(monkeypatch):
|
|
return monkeypatch
|
|
|
|
|
|
def mock_out_get_repository(monkeypatch, namespace_name, repository_name):
|
|
def return_none(namespace_name, repository_name):
|
|
return None
|
|
|
|
def return_repository(namespace_name, repository_name):
|
|
return 'repository'
|
|
|
|
if namespace_name == BAD_NAMESPACE_NAME or repository_name == BAD_REPOSITORY_NAME:
|
|
return_function = return_none
|
|
else:
|
|
return_function = return_repository
|
|
|
|
monkeypatch.setattr(model.repository, 'get_repository', return_function)
|
|
|
|
|
|
def create_mock_tag(name, reversion, lifetime_start_ts, lifetime_end_ts, mock_id, docker_image_id,
|
|
manifest_list):
|
|
tag_mock = Mock()
|
|
tag_mock.name = name
|
|
image_mock = Mock()
|
|
image_mock.docker_image_id = docker_image_id
|
|
tag_mock.image = image_mock
|
|
tag_mock.reversion = reversion
|
|
tag_mock.lifetime_start_ts = lifetime_start_ts
|
|
tag_mock.lifetime_end_ts = lifetime_end_ts
|
|
tag_mock.id = mock_id
|
|
tag_mock.manifest_list = manifest_list
|
|
tag = Tag(name=name, reversion=reversion, image=image_mock, docker_image_id=docker_image_id,
|
|
lifetime_start_ts=lifetime_start_ts, lifetime_end_ts=lifetime_end_ts,
|
|
manifest_list=manifest_list)
|
|
return tag_mock, tag
|
|
|
|
|
|
first_mock, first_tag = create_mock_tag('tag1', 'rev1', 'start1', 'end1', 'id1',
|
|
'docker_image_id1', [])
|
|
second_mock, second_tag = create_mock_tag('tag2', 'rev2', 'start2', 'end2', 'id2',
|
|
'docker_image_id2', ['manifest'])
|
|
|
|
|
|
def mock_out_list_repository_tag_history(monkeypatch, namespace_name, repository_name, page, size,
|
|
specific_tag):
|
|
def list_empty_tag_history(repository, page, size, specific_tag):
|
|
return [], {}, False
|
|
|
|
def list_filled_tag_history(repository, page, size, specific_tag):
|
|
tags = [first_mock, second_mock]
|
|
return tags, {
|
|
first_mock.id: first_mock.manifest_list,
|
|
second_mock.id: second_mock.manifest_list
|
|
}, len(tags) > size
|
|
|
|
def list_only_second_tag(repository, page, size, specific_tag):
|
|
tags = [second_mock]
|
|
return tags, {second_mock.id: second_mock.manifest_list}, len(tags) > size
|
|
|
|
if namespace_name == EMPTY_NAMESPACE or repository_name == EMPTY_REPOSITORY:
|
|
return_function = list_empty_tag_history
|
|
else:
|
|
if specific_tag == 'tag2':
|
|
return_function = list_only_second_tag
|
|
else:
|
|
return_function = list_filled_tag_history
|
|
|
|
monkeypatch.setattr(model.tag, 'list_repository_tag_history', return_function)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'expected, namespace_name, repository_name, page, size, specific_tag', [
|
|
(None, BAD_NAMESPACE_NAME, 'repository_name', 1, 100, None),
|
|
(None, 'namespace_name', BAD_REPOSITORY_NAME, 1, 100, None),
|
|
(RepositoryTagHistory(tags=[], more=False), EMPTY_NAMESPACE, EMPTY_REPOSITORY, 1, 100, None),
|
|
(RepositoryTagHistory(tags=[first_tag, second_tag], more=False), 'namespace', 'repository', 1,
|
|
100, None),
|
|
(RepositoryTagHistory(tags=[first_tag, second_tag], more=True), 'namespace', 'repository', 1,
|
|
1, None),
|
|
(RepositoryTagHistory(tags=[second_tag], more=False), 'namespace', 'repository', 1, 100,
|
|
'tag2'),
|
|
])
|
|
def test_list_repository_tag_history(expected, namespace_name, repository_name, page, size,
|
|
specific_tag, get_monkeypatch):
|
|
mock_out_get_repository(get_monkeypatch, namespace_name, repository_name)
|
|
mock_out_list_repository_tag_history(get_monkeypatch, namespace_name, repository_name, page,
|
|
size, specific_tag)
|
|
assert pre_oci_model.list_repository_tag_history(namespace_name, repository_name, page, size,
|
|
specific_tag) == expected
|