Alexander Kislitsky 9390d0ad2b Fix for levels mismatch in set request and environment
Error is rased now if nonexistent level specified in the set
resource value request. For instance we have env levels: ['a'].
Request to set resource value with levels ['a', 'b'] raises error
now.

Change-Id: I6ce8dc99b288196092a7e4cf69be9756d5b0f6dd
Closes-Bug: #1624271
Closes-Bug: #1614551
2016-09-19 15:38:10 +03:00

112 lines
4.5 KiB
Python

# 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.
from tuning_box.app import db
from tuning_box import errors
from tuning_box import library
from tuning_box.tests.test_app import BaseTest
class TestLibrary(BaseTest):
def add_res_def_to_another_env(self, res_name):
component_data = {
'name': 'component2',
'resource_definitions': [{
'name': res_name,
'content': {'key': 'value'}
}]
}
res = self.client.post('/components', data=component_data)
self.assertEqual(201, res.status_code)
component_id = res.json['id']
env_data = {
'components': [component_id],
'hierarchy_levels': [],
}
res = self.client.post('/environments', data=env_data)
self.assertEqual(201, res.status_code)
def test_get_resource_definition(self):
self._fixture()
res_name = 'resdef1'
res_id = 5
environment_id = 9
component_id = 7
# Creating resource definition with the same name in another
# environment
self.add_res_def_to_another_env(res_name)
res = self.client.get('/resource_definitions')
self.assertEqual(200, res.status_code)
self.assertTrue(all(res_def['name'] == res_name
for res_def in res.json))
with self.app.app_context():
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_definition, res_id, None)
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_definition, res_name, None)
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_definition, '',
environment_id)
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_definition, None, None)
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_definition, None,
environment_id)
actual_res = library.get_resource_definition(res_id,
environment_id)
self.assertEqual(res_id, actual_res.id)
self.assertEqual(res_name, actual_res.name)
self.assertEqual(component_id, actual_res.component_id)
actual_res = library.get_resource_definition(res_id,
environment_id)
self.assertEqual(res_id, actual_res.id)
self.assertEqual(res_name, actual_res.name)
self.assertEqual(component_id, actual_res.component_id)
def test_get_resource_values(self):
self._fixture()
res_def_id = 5
environment_id = 9
values = {'k': 'v'}
levels = (('lvl1', 'val1'), ('lvl2', 'val2'))
self._add_resource_values(environment_id, res_def_id, levels, values)
with self.app.app_context(), db.db.session.begin():
environment = db.Environment.query.get(environment_id)
res_def = db.ResourceDefinition.query.get(res_def_id)
res_values = library.get_resource_values(
environment, levels, res_def)
self.assertEqual(values, res_values.values)
def test_get_resource_values_not_found(self):
self._fixture()
res_def_id = 5
environment_id = 9
levels = (('lvl1', 'val1'), ('lvl2', 'val2'))
with self.app.app_context(), db.db.session.begin():
environment = db.Environment.query.get(environment_id)
res_def = db.ResourceDefinition.query.get(res_def_id)
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_values, environment,
levels, res_def)
# Test for empty levels
self.assertRaises(errors.TuningboxNotFound,
library.get_resource_values, environment,
(), res_def)