add tests

This commit is contained in:
Kurt Hutten IrevDev
2022-06-14 21:35:20 +10:00
parent 374d5bd238
commit 3237e02c10

View File

@ -3,8 +3,8 @@ import pytest
import asyncio
from .client import ClientFromEnv
from .models import FileConversion, FileOutputFormat, FileSourceFormat, User, Pong, APICallStatus
from .api.file import create_file_conversion_with_base64_helper
from .models import FileConversion, FileOutputFormat, FileSourceFormat, User, Pong, APICallStatus, FileMass, FileVolume
from .api.file import create_file_conversion_with_base64_helper, create_file_mass, create_file_volume
from .api.meta import ping
from .api.users import get_user_self
@ -106,3 +106,56 @@ async def test_file_convert_stl_async():
assert fc.id is not None
print(f"FileConversion: {fc}")
def test_file_mass():
# Create our client.
client = ClientFromEnv()
dir_path = os.path.dirname(os.path.realpath(__file__))
file = open(os.path.join(dir_path, "../assets/testing.stl"), "rb")
content = file.read()
file.close()
# Get the fc.
fm: FileMass = create_file_mass.sync(
client=client,
body=content,
src_format=FileSourceFormat.STL,
material_density=1.0)
assert fm is not None
print(f"FileConversion: {fm}")
assert fm.id is not None
assert fm.mass is not None
assert fm.status == APICallStatus.COMPLETED
print(f"FileConversion: {fm}")
def test_file_volume():
# Create our client.
client = ClientFromEnv()
dir_path = os.path.dirname(os.path.realpath(__file__))
file = open(os.path.join(dir_path, "../assets/testing.stl"), "rb")
content = file.read()
file.close()
# Get the fc.
fv: FileVolume = create_file_volume.sync(
client=client,
body=content,
src_format=FileSourceFormat.STL)
assert fv is not None
print(f"FileConversion: {fv}")
assert fv.id is not None
assert fv.volume is not None
assert fv.status == APICallStatus.COMPLETED
print(f"FileConversion: {fv}")