Move blob model tests to pytest
This commit is contained in:
parent
31bfa697f3
commit
dda083abb1
2 changed files with 32 additions and 54 deletions
32
data/model/test/test_model_blob.py
Normal file
32
data/model/test/test_model_blob.py
Normal file
|
@ -0,0 +1,32 @@
|
|||
from data import model, database
|
||||
|
||||
from test.fixtures import *
|
||||
|
||||
ADMIN_ACCESS_USER = 'devtable'
|
||||
REPO = 'simple'
|
||||
|
||||
def test_store_blob(initialized_db):
|
||||
location = database.ImageStorageLocation.select().get()
|
||||
|
||||
# Create a new blob at a unique digest.
|
||||
digest = 'somecooldigest'
|
||||
blob_storage = model.blob.store_blob_record_and_temp_link(ADMIN_ACCESS_USER, REPO, digest,
|
||||
location, 1024, 0, 5000)
|
||||
assert blob_storage.content_checksum == digest
|
||||
assert blob_storage.image_size == 1024
|
||||
assert blob_storage.uncompressed_size == 5000
|
||||
|
||||
# Link to the same digest.
|
||||
blob_storage2 = model.blob.store_blob_record_and_temp_link(ADMIN_ACCESS_USER, REPO, digest,
|
||||
location, 2048, 0, 6000)
|
||||
assert blob_storage2.id == blob_storage.id
|
||||
assert blob_storage2.image_size == 2048
|
||||
assert blob_storage2.uncompressed_size == 6000
|
||||
|
||||
# Add a new digest, ensure it has a new record.
|
||||
otherdigest = 'anotherdigest'
|
||||
blob_storage3 = model.blob.store_blob_record_and_temp_link(ADMIN_ACCESS_USER, REPO, otherdigest,
|
||||
location, 1234, 0, 5678)
|
||||
assert blob_storage3.id != blob_storage.id
|
||||
assert blob_storage3.image_size == 1234
|
||||
assert blob_storage3.uncompressed_size == 5678
|
Reference in a new issue