feat: add Sha256Str type
This commit is contained in:
1 parent
49b84acf1f
commit
dc5b09048d
4 files changed
+110
No files matched your search
@@ -0,0 +1,45 @@
|
||||
import pytest
|
||||
from pydantic import BaseModel, ValidationError
|
||||
|
||||
from pydantic_extra_types.sha256 import Sha256Str
|
||||
|
||||
EMPTY_SHA256 = 'e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'
|
||||
|
||||
|
||||
class Artifact(BaseModel):
|
||||
checksum: Sha256Str
|
||||
|
||||
|
||||
@pytest.mark.parametrize('value', [EMPTY_SHA256, EMPTY_SHA256.upper(), f' {EMPTY_SHA256}\n'])
|
||||
def test_valid_sha256(value: str) -> None:
|
||||
checksum = Artifact(checksum=value).checksum
|
||||
assert checksum == EMPTY_SHA256
|
||||
assert isinstance(checksum, Sha256Str)
|
||||
|
||||
|
||||
def test_sha256_from_json() -> None:
|
||||
artifact = Artifact.model_validate_json(f'{{"checksum": "{EMPTY_SHA256.upper()}"}}')
|
||||
assert artifact.checksum == EMPTY_SHA256
|
||||
assert artifact.model_dump_json() == f'{{"checksum":"{EMPTY_SHA256}"}}'
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
'value',
|
||||
[
|
||||
'',
|
||||
EMPTY_SHA256[:-1],
|
||||
EMPTY_SHA256 + '0',
|
||||
'g' + EMPTY_SHA256[1:],
|
||||
'0x' + EMPTY_SHA256[2:],
|
||||
EMPTY_SHA256[:30] + ' ' + EMPTY_SHA256[31:],
|
||||
EMPTY_SHA256[:-1] + '\u0663',
|
||||
],
|
||||
)
|
||||
def test_invalid_sha256(value: str) -> None:
|
||||
with pytest.raises(ValidationError, match='string_pattern_mismatch'):
|
||||
Artifact(checksum=value)
|
||||
|
||||
|
||||
def test_sha256_not_a_string() -> None:
|
||||
with pytest.raises(ValidationError, match='string_type'):
|
||||
Artifact(checksum=1234)
|
||||
Reference in new issue
Block a user