118 lines
3.9 KiB
Python
118 lines
3.9 KiB
Python
# Copyright 2017 Catalyst IT Ltd
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import os
|
|
import tempfile
|
|
import zipfile
|
|
|
|
from tempest.lib.common.utils import data_utils
|
|
from tempest.lib import decorators
|
|
|
|
from qinling_tempest_plugin.tests import base
|
|
|
|
|
|
class FunctionsTest(base.BaseQinlingTest):
|
|
name_prefix = 'FunctionsTest'
|
|
|
|
@classmethod
|
|
def resource_setup(cls):
|
|
super(FunctionsTest, cls).resource_setup()
|
|
|
|
cls.runtime_id = None
|
|
|
|
# Create runtime for function tests
|
|
name = data_utils.rand_name('runtime', prefix=cls.name_prefix)
|
|
_, body = cls.admin_client.create_runtime(
|
|
'openstackqinling/python-runtime', name
|
|
)
|
|
cls.runtime_id = body['id']
|
|
|
|
@classmethod
|
|
def resource_cleanup(cls):
|
|
if cls.runtime_id:
|
|
cls.admin_client.delete_resource('runtimes', cls.runtime_id)
|
|
|
|
super(FunctionsTest, cls).resource_cleanup()
|
|
|
|
def setUp(self):
|
|
super(FunctionsTest, self).setUp()
|
|
|
|
# Wait until runtime is available
|
|
self.await_runtime_available(self.runtime_id)
|
|
|
|
python_file_path = os.path.abspath(
|
|
os.path.join(
|
|
os.path.dirname(__file__),
|
|
os.pardir,
|
|
os.pardir,
|
|
'functions/python_test.py'
|
|
)
|
|
)
|
|
|
|
base_name, extention = os.path.splitext(python_file_path)
|
|
self.base_name = os.path.basename(base_name)
|
|
self.python_zip_file = os.path.join(
|
|
tempfile.gettempdir(),
|
|
'%s.zip' % self.base_name
|
|
)
|
|
|
|
if not os.path.isfile(self.python_zip_file):
|
|
zf = zipfile.ZipFile(self.python_zip_file, mode='w')
|
|
try:
|
|
# Use default compression mode, may change in future.
|
|
zf.write(
|
|
python_file_path,
|
|
'%s%s' % (self.base_name, extention),
|
|
compress_type=zipfile.ZIP_STORED
|
|
)
|
|
finally:
|
|
zf.close()
|
|
|
|
@decorators.idempotent_id('9c36ac64-9a44-4c44-9e44-241dcc6b0933')
|
|
def test_crud_function(self):
|
|
# Create function
|
|
function_name = data_utils.rand_name('function',
|
|
prefix=self.name_prefix)
|
|
with open(self.python_zip_file, 'rb') as package_data:
|
|
resp, body = self.client.create_function(
|
|
{"source": "package"},
|
|
self.runtime_id,
|
|
name=function_name,
|
|
package_data=package_data,
|
|
entry='%s.main' % self.base_name
|
|
)
|
|
function_id = body['id']
|
|
|
|
self.assertEqual(201, resp.status_code)
|
|
self.addCleanup(self.client.delete_resource, 'functions',
|
|
function_id, ignore_notfound=True)
|
|
|
|
# Get functions
|
|
resp, body = self.client.get_resources('functions')
|
|
|
|
self.assertEqual(200, resp.status)
|
|
self.assertIn(
|
|
function_id,
|
|
[function['id'] for function in body['functions']]
|
|
)
|
|
|
|
# Download function package
|
|
resp, data = self.client.download_function(function_id)
|
|
self.assertEqual(200, resp.status)
|
|
self.assertEqual(os.path.getsize(self.python_zip_file), len(data))
|
|
|
|
# Delete function
|
|
resp = self.client.delete_resource('functions', function_id)
|
|
|
|
self.assertEqual(204, resp.status)
|