archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

test_sdk: load BlenderDesktopLogo.png from the correct path

Browse Source
This commit is contained in:
Sybren A. Stüvel
2016-08-23 14:32:59 +02:00
parent 9d1181330b
commit 2e0ba4c6cd
1 changed files with 8 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_sdk.py
View File

@@ -1,11 +1,16 @@
"""Tests for the FlaskInternal SDK.""" """Tests for the FlaskInternal SDK."""
from os.path import join, dirname, abspath, exists
from flask import url_for from flask import url_for
import pillarsdk import pillarsdk
from pillar.tests import AbstractPillarTest from pillar.tests import AbstractPillarTest
from pillar.sdk import FlaskInternalApi from pillar.sdk import FlaskInternalApi
blender_desktop_logo_path = join(dirname(abspath(__file__)), 'BlenderDesktopLogo.png')
assert exists(blender_desktop_logo_path)
class FlaskInternalApiTest(AbstractPillarTest): class FlaskInternalApiTest(AbstractPillarTest):
def setUp(self, **kwargs): def setUp(self, **kwargs):
@@ -48,16 +53,11 @@ class FlaskInternalApiTest(AbstractPillarTest):
asset.delete(api=self.sdk_api) asset.delete(api=self.sdk_api)
def test_upload_file_to_project(self): def test_upload_file_to_project(self):
import test_api
from os.path import join, dirname, abspath
file_path = join(dirname(abspath(test_api.__file__)), 'BlenderDesktopLogo.png')
with self.app.test_request_context(): with self.app.test_request_context():
resp = pillarsdk.File.upload_to_project( resp = pillarsdk.File.upload_to_project(
self.project_id, self.project_id,
'image/png', 'image/png',
file_path, blender_desktop_logo_path,
api=self.sdk_api api=self.sdk_api
) )
file_id = resp['file_id'] file_id = resp['file_id']
@@ -70,11 +70,6 @@ class FlaskInternalApiTest(AbstractPillarTest):
self.assertEqual('BlenderDesktopLogo.png', file_doc['filename']) self.assertEqual('BlenderDesktopLogo.png', file_doc['filename'])
def test_create_asset_from_file(self): def test_create_asset_from_file(self):
import test_api
from os.path import join, dirname, abspath
file_path = join(dirname(abspath(test_api.__file__)), 'BlenderDesktopLogo.png')
# Create a group node to serve as parent. # Create a group node to serve as parent.
with self.app.test_request_context(): with self.app.test_request_context():
resp = self.post(url_for('nodes|resource'), auth_token='token', resp = self.post(url_for('nodes|resource'), auth_token='token',
@@ -87,12 +82,12 @@ class FlaskInternalApiTest(AbstractPillarTest):
expected_status=201) expected_status=201)
parent_id = resp.json()['_id'] parent_id = resp.json()['_id']
with self.app.test_request_context(), open(file_path, 'rb') as fileobj: with self.app.test_request_context(), open(blender_desktop_logo_path, 'rb') as fileobj:
resp = pillarsdk.Node.create_asset_from_file( resp = pillarsdk.Node.create_asset_from_file(
unicode(self.project_id), unicode(self.project_id),
unicode(parent_id), unicode(parent_id),
'image', 'image',
file_path, blender_desktop_logo_path,
mimetype='image/jpeg', mimetype='image/jpeg',
always_create_new_node=False, always_create_new_node=False,
fileobj=fileobj, fileobj=fileobj,