Move remaining util tests into pytest
This commit is contained in:
parent
fa354d695a
commit
69406c6aec
4 changed files with 198 additions and 226 deletions
98
util/registry/test/test_generatorfile.py
Normal file
98
util/registry/test/test_generatorfile.py
Normal file
|
@ -0,0 +1,98 @@
|
|||
from _pyio import BufferedReader
|
||||
|
||||
import magic
|
||||
|
||||
from util.registry.generatorfile import GeneratorFile
|
||||
|
||||
def sample_generator():
|
||||
yield 'this'
|
||||
yield 'is'
|
||||
yield 'a'
|
||||
yield 'test'
|
||||
|
||||
def test_basic_generator():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
assert f.tell() == 0
|
||||
assert f.read() == "thisisatest"
|
||||
assert f.tell() == len("thisisatest")
|
||||
|
||||
def test_same_lengths():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
assert f.read(4) == "this"
|
||||
assert f.tell() == 4
|
||||
|
||||
assert f.read(2) == "is"
|
||||
assert f.tell() == 6
|
||||
|
||||
assert f.read(1) == "a"
|
||||
assert f.tell() == 7
|
||||
|
||||
assert f.read(4) == "test"
|
||||
assert f.tell() == 11
|
||||
|
||||
def test_indexed_lengths():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
assert f.read(6) == "thisis"
|
||||
assert f.tell() == 6
|
||||
|
||||
assert f.read(5) == "atest"
|
||||
assert f.tell() == 11
|
||||
|
||||
def test_misindexed_lengths():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
assert f.read(6) == "thisis"
|
||||
assert f.tell() == 6
|
||||
|
||||
assert f.read(3) == "ate"
|
||||
assert f.tell() == 9
|
||||
|
||||
assert f.read(2) == "st"
|
||||
assert f.tell() == 11
|
||||
|
||||
assert f.read(2) == ""
|
||||
assert f.tell() == 11
|
||||
|
||||
def test_misindexed_lengths_2():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
assert f.read(8) == "thisisat"
|
||||
assert f.tell() == 8
|
||||
|
||||
assert f.read(1) == "e"
|
||||
assert f.tell() == 9
|
||||
|
||||
assert f.read(2) == "st"
|
||||
assert f.tell() == 11
|
||||
|
||||
assert f.read(2) == ""
|
||||
assert f.tell() == 11
|
||||
|
||||
def test_overly_long():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
assert f.read(60) == "thisisatest"
|
||||
assert f.tell() == 11
|
||||
|
||||
def test_with_bufferedreader():
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
buffered = BufferedReader(f)
|
||||
assert buffered.peek(10) == "thisisatest"
|
||||
assert buffered.read(10) == "thisisates"
|
||||
|
||||
def mimed_html_generator():
|
||||
yield '<html>'
|
||||
yield '<body>'
|
||||
yield 'sometext' * 1024
|
||||
yield '</body>'
|
||||
yield '</html>'
|
||||
|
||||
def test_magic():
|
||||
mgc = magic.Magic(mime=True)
|
||||
|
||||
with GeneratorFile(mimed_html_generator()) as f:
|
||||
buffered = BufferedReader(f)
|
||||
file_header_bytes = buffered.peek(1024)
|
||||
assert mgc.from_buffer(file_header_bytes) == "text/html"
|
||||
|
||||
with GeneratorFile(sample_generator()) as f:
|
||||
buffered = BufferedReader(f)
|
||||
file_header_bytes = buffered.peek(1024)
|
||||
assert mgc.from_buffer(file_header_bytes) == "text/plain"
|
20
util/test/test_util.py
Normal file
20
util/test/test_util.py
Normal file
|
@ -0,0 +1,20 @@
|
|||
import pytest
|
||||
|
||||
from util import slash_join
|
||||
|
||||
@pytest.mark.parametrize('pieces, expected', [
|
||||
(['https://github.com', '/coreos-inc/' 'quay/pull/1092/files'],
|
||||
'https://github.com/coreos-inc/quay/pull/1092/files'),
|
||||
|
||||
(['https://', 'github.com/', '/coreos-inc', '/quay/pull/1092/files/'],
|
||||
'https://github.com/coreos-inc/quay/pull/1092/files'),
|
||||
|
||||
(['https://somegithub.com/', '/api/v3/'],
|
||||
'https://somegithub.com/api/v3'),
|
||||
|
||||
(['https://github.somedomain.com/', '/api/v3/'],
|
||||
'https://github.somedomain.com/api/v3'),
|
||||
])
|
||||
def test_slash_join(pieces, expected):
|
||||
joined_url = slash_join(*pieces)
|
||||
assert joined_url == expected
|
|
@ -1,6 +1,32 @@
|
|||
from itertools import islice
|
||||
|
||||
import pytest
|
||||
|
||||
from util.validation import is_json, validate_label_key
|
||||
from util.validation import is_json, validate_label_key, generate_valid_usernames, validate_username
|
||||
|
||||
@pytest.mark.parametrize('username, is_valid', [
|
||||
('ja', True),
|
||||
('jak', True),
|
||||
('jake', True),
|
||||
('ja_ke', True),
|
||||
('te-st', True),
|
||||
('te.st', True),
|
||||
|
||||
('z' * 30, True),
|
||||
('z' * 255, True),
|
||||
|
||||
('j', False),
|
||||
|
||||
('z' * 256, False),
|
||||
|
||||
('_test', False),
|
||||
('Test', False),
|
||||
('hello world', False),
|
||||
('te---st', False),
|
||||
])
|
||||
def test_validate_username(username, is_valid):
|
||||
valid, _ = validate_username(username)
|
||||
assert valid == is_valid
|
||||
|
||||
|
||||
@pytest.mark.parametrize('string_value,expected', [
|
||||
|
@ -48,3 +74,56 @@ def test_is_json(string_value, expected):
|
|||
])
|
||||
def test_validate_label_key(key, is_valid):
|
||||
assert validate_label_key(key) == is_valid
|
||||
|
||||
|
||||
@pytest.mark.parametrize('input_username, expected_output', [
|
||||
('jake', 'jake'),
|
||||
('frank', 'frank'),
|
||||
('fra-nk', 'fra_nk'),
|
||||
|
||||
('Jake', 'jake'),
|
||||
('FranK', 'frank'),
|
||||
|
||||
('ja__ke', 'ja_ke'),
|
||||
('ja___ke', 'ja_ke'),
|
||||
|
||||
('ja__', 'ja'),
|
||||
('jake__', 'jake'),
|
||||
|
||||
('_jake', 'jake'),
|
||||
|
||||
('a', 'a0'),
|
||||
('ab', 'ab'),
|
||||
('abc', 'abc'),
|
||||
|
||||
('abcdefghijklmnopqrstuvwxyz1234567890', 'abcdefghijklmnopqrstuvwxyz1234567890'),
|
||||
('c' * 256, 'c' * 255),
|
||||
|
||||
(u'\xc6neid', 'aeneid'),
|
||||
(u'\xe9tude', 'etude'),
|
||||
(u'\u5317\u4eb0', 'bei_jing'),
|
||||
(u'\u1515\u14c7\u14c7', 'shanana'),
|
||||
(u'\u13d4\u13b5\u13c6', 'taliqua'),
|
||||
(u'\u0726\u071b\u073d\u0710\u073a', 'ptu_i'),
|
||||
(u'\u0905\u092d\u093f\u091c\u0940\u0924', 'abhijiit'),
|
||||
(u'\u0985\u09ad\u09bf\u099c\u09c0\u09a4', 'abhijiit'),
|
||||
(u'\u0d05\u0d2d\u0d3f\u0d1c\u0d40\u0d24', 'abhijiit'),
|
||||
(u'\u0d2e\u0d32\u0d2f\u0d3e\u0d32\u0d2e\u0d4d', 'mlyaalm'),
|
||||
(u'\ue000', '00'),
|
||||
(u'\u03ff', '00'),
|
||||
|
||||
(u'\u0d2e\u0d32\u03ff\u03ff\u0d2e\u0d32', 'mlml'),
|
||||
])
|
||||
def test_generate_valid_usernames(input_username, expected_output):
|
||||
name_gen = generate_valid_usernames(input_username)
|
||||
generated_output = list(islice(name_gen, 1))[0]
|
||||
assert generated_output == expected_output
|
||||
|
||||
|
||||
def test_multiple_suggestions():
|
||||
name_gen = generate_valid_usernames('a')
|
||||
generated_output = list(islice(name_gen, 4))
|
||||
assert generated_output[0] == 'a0'
|
||||
assert generated_output[1] == 'a1'
|
||||
assert generated_output[2] == 'a2'
|
||||
assert generated_output[3] == 'a3'
|
||||
|
|
Reference in a new issue