File size: 1,311 Bytes
6ab520d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
# import os
from io import BytesIO
import app.utils as utils
import pytest
from fastapi import UploadFile
from werkzeug.datastructures import FileStorage
def test_allowed_file():
# 💡 NOTE Run test with: pytest ./tests/test_utils.py::test_allowed_file -v
assert utils.allowed_file("cat.JPG")
assert utils.allowed_file("cat.jpeg")
assert utils.allowed_file("cat.JPEG")
assert utils.allowed_file("../../car.PNG")
assert utils.allowed_file("/usr/var/src/car.gif")
assert not utils.allowed_file("cat.JPGG")
assert not utils.allowed_file("invoice.pdf")
assert not utils.allowed_file("/usr/src/slides.odt")
assert not utils.allowed_file("/usr/src/api")
assert not utils.allowed_file("/usr/src/api/")
assert not utils.allowed_file("/usr/src/dog.")
assert not utils.allowed_file("/usr/src/dog./")
@pytest.mark.asyncio
async def test_get_file_hash():
# 💡 NOTE Run test with: pytest ./tests/test_utils.py::test_get_file_hash -v
filename = "tests/dog.jpeg"
md5_filename = "0a7c757a80f2c5b13fa7a2a47a683593.jpeg"
with open(filename, "rb") as fp:
file = FileStorage(fp)
file = UploadFile(file=BytesIO(file.read()), filename="dog.jpeg")
new_filename = await utils.get_file_hash(file)
assert md5_filename == new_filename
|