fix strip_sha256

This commit is contained in:
Antoine Legrand 2017-03-24 19:43:23 +01:00
parent 4a64ddc86e
commit 22c1a29892
2 changed files with 13 additions and 1 deletions

View file

@ -429,6 +429,9 @@ class OCIAppModel(AppRegistryDataInterface):
def _strip_sha256_header(digest):
return digest.lstrip('sha256:')
if digest.startswith('sha256:'):
return digest.split('sha256:')[1]
return digest
oci_app_model = OCIAppModel()

View file

@ -0,0 +1,9 @@
import pytest
from data.interfaces.appr import _strip_sha256_header
@pytest.mark.parametrize('digest,expected', [
('sha256:251b6897608fb18b8a91ac9abac686e2e95245d5a041f2d1e78fe7a815e6480a', '251b6897608fb18b8a91ac9abac686e2e95245d5a041f2d1e78fe7a815e6480a'),
('251b6897608fb18b8a91ac9abac686e2e95245d5a041f2d1e78fe7a815e6480a', '251b6897608fb18b8a91ac9abac686e2e95245d5a041f2d1e78fe7a815e6480a'),])
def test_stip_sha256(digest, expected):
assert _strip_sha256_header(digest) == expected