Changeset View
Changeset View
Standalone View
Standalone View
swh/model/hypothesis_strategies.py
| Show All 20 Lines | |||||
| from .identifiers import snapshot_identifier, identifier_to_bytes | from .identifiers import snapshot_identifier, identifier_to_bytes | ||||
| pgsql_alphabet = characters( | pgsql_alphabet = characters( | ||||
| blacklist_categories=('Cs', ), | blacklist_categories=('Cs', ), | ||||
| blacklist_characters=['\u0000']) # postgresql does not like these | blacklist_characters=['\u0000']) # postgresql does not like these | ||||
| def optional(strategy): | |||||
| return one_of(none(), strategy) | |||||
| def pgsql_text(): | def pgsql_text(): | ||||
| return text(alphabet=pgsql_alphabet) | return text(alphabet=pgsql_alphabet) | ||||
| def sha1_git(): | def sha1_git(): | ||||
| return binary(min_size=20, max_size=20) | return binary(min_size=20, max_size=20) | ||||
| Show All 39 Lines | |||||
| def origin_visits(): | def origin_visits(): | ||||
| return builds( | return builds( | ||||
| OriginVisit, | OriginVisit, | ||||
| visit=integers(0, 1000), | visit=integers(0, 1000), | ||||
| origin=urls(), | origin=urls(), | ||||
| status=sampled_from(['ongoing', 'full', 'partial']), | status=sampled_from(['ongoing', 'full', 'partial']), | ||||
| type=pgsql_text()) | type=pgsql_text(), | ||||
| snapshot=optional(sha1_git())) | |||||
| @composite | @composite | ||||
| def releases(draw): | def releases(draw): | ||||
| (date, author) = draw(one_of( | (date, author) = draw(one_of( | ||||
| tuples(none(), none()), | tuples(none(), none()), | ||||
| tuples(timestamps_with_timezone(), persons()))) | tuples(timestamps_with_timezone(), persons()))) | ||||
| rel = draw(builds( | rel = draw(builds( | ||||
| ▲ Show 20 Lines • Show All 50 Lines • ▼ Show 20 Lines | return draw(builds( | ||||
| blake2s256=binary(min_size=32, max_size=32), | blake2s256=binary(min_size=32, max_size=32), | ||||
| status=one_of(just('visible'), just('hidden')), | status=one_of(just('visible'), just('hidden')), | ||||
| data=binary(), | data=binary(), | ||||
| )) | )) | ||||
| @composite | @composite | ||||
| def skipped_contents(draw): | def skipped_contents(draw): | ||||
| def optional(strategy): | |||||
| return one_of(none(), strategy) | |||||
| return draw(builds( | return draw(builds( | ||||
| SkippedContent, | SkippedContent, | ||||
| length=integers(min_value=-1, max_value=2**63-1), | length=integers(min_value=-1, max_value=2**63-1), | ||||
| sha1=optional(sha1()), | sha1=optional(sha1()), | ||||
| sha1_git=optional(sha1_git()), | sha1_git=optional(sha1_git()), | ||||
| sha256=optional(binary(min_size=32, max_size=32)), | sha256=optional(binary(min_size=32, max_size=32)), | ||||
| blake2s256=optional(binary(min_size=32, max_size=32)), | blake2s256=optional(binary(min_size=32, max_size=32)), | ||||
| status=just('absent'), | status=just('absent'), | ||||
| ▲ Show 20 Lines • Show All 86 Lines • Show Last 20 Lines | |||||