From 197aa2fc55e64c9b24c7b8d0a79157f51848e3cc Mon Sep 17 00:00:00 2001 From: Jess Frazelle Date: Tue, 5 Jul 2022 15:38:49 -0700 Subject: [PATCH] fix tests Signed-off-by: Jess Frazelle --- Makefile | 1 + kittycad/client_test.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/Makefile b/Makefile index a7daedc8c..57aebe465 100644 --- a/Makefile +++ b/Makefile @@ -22,6 +22,7 @@ generate: docker-image ## Generate the api client. shell: docker-image ## Pop into a shell in the docker image. docker run --rm -i $(DOCKER_FLAGS) \ --name python-generator-shell \ + -e KITTYCAD_API_TOKEN \ --disable-content-trust \ -v $(CURDIR):/usr/src \ --workdir /usr/src \ diff --git a/kittycad/client_test.py b/kittycad/client_test.py index ead487e57..7935b93eb 100644 --- a/kittycad/client_test.py +++ b/kittycad/client_test.py @@ -3,7 +3,7 @@ import pytest import asyncio from .client import ClientFromEnv -from .models import FileConversion, FileOutputFormat, FileSourceFormat, User, Pong, APICallStatus, FileMass, FileVolume +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 @@ -81,7 +81,7 @@ def test_file_convert_stl(): assert fc.id is not None - assert fc.status == APICallStatus.COMPLETED + assert fc.status == ApiCallStatus.COMPLETED print(f"FileConversion: {fc}") @@ -133,7 +133,7 @@ def test_file_mass(): assert fm.to_dict() is not None - assert fm.status == APICallStatus.COMPLETED + assert fm.status == ApiCallStatus.COMPLETED def test_file_volume(): @@ -160,4 +160,4 @@ def test_file_volume(): assert fv.to_dict() is not None - assert fv.status == APICallStatus.COMPLETED + assert fv.status == ApiCallStatus.COMPLETED