147 lines
4.3 KiB
Python
147 lines
4.3 KiB
Python
# Copyright 2012 Cloudscaling Group, Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 collections
|
|
|
|
from oslotest import base as oslo_test
|
|
from sqlalchemy import Column
|
|
from sqlalchemy import Integer, String
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
|
from oslo.db.sqlalchemy import models
|
|
from oslo.db.sqlalchemy import test_base
|
|
|
|
|
|
BASE = declarative_base()
|
|
|
|
|
|
class ModelBaseTest(test_base.DbTestCase):
|
|
def setUp(self):
|
|
super(ModelBaseTest, self).setUp()
|
|
self.mb = models.ModelBase()
|
|
self.ekm = ExtraKeysModel()
|
|
|
|
def test_modelbase_has_dict_methods(self):
|
|
dict_methods = ('__getitem__',
|
|
'__setitem__',
|
|
'__contains__',
|
|
'get',
|
|
'update',
|
|
'save',
|
|
'iteritems')
|
|
for method in dict_methods:
|
|
self.assertTrue(hasattr(models.ModelBase, method),
|
|
"Method %s() is not found" % method)
|
|
|
|
def test_modelbase_is_iterable(self):
|
|
self.assertTrue(issubclass(models.ModelBase, collections.Iterable))
|
|
|
|
def test_modelbase_set(self):
|
|
self.mb['world'] = 'hello'
|
|
self.assertEqual(self.mb['world'], 'hello')
|
|
|
|
def test_modelbase_update(self):
|
|
h = {'a': '1', 'b': '2'}
|
|
self.mb.update(h)
|
|
for key in h.keys():
|
|
self.assertEqual(self.mb[key], h[key])
|
|
|
|
def test_modelbase_contains(self):
|
|
mb = models.ModelBase()
|
|
h = {'a': '1', 'b': '2'}
|
|
mb.update(h)
|
|
for key in h.keys():
|
|
# Test 'in' syntax (instead of using .assertIn)
|
|
self.assertTrue(key in mb)
|
|
|
|
self.assertFalse('non-existent-key' in mb)
|
|
|
|
def test_modelbase_iteritems(self):
|
|
h = {'a': '1', 'b': '2'}
|
|
expected = {
|
|
'id': None,
|
|
'smth': None,
|
|
'name': 'NAME',
|
|
'a': '1',
|
|
'b': '2',
|
|
}
|
|
self.ekm.update(h)
|
|
self.assertEqual(dict(self.ekm.iteritems()), expected)
|
|
|
|
def test_modelbase_iter(self):
|
|
expected = {
|
|
'id': None,
|
|
'smth': None,
|
|
'name': 'NAME',
|
|
}
|
|
i = iter(self.ekm)
|
|
found_items = 0
|
|
while True:
|
|
r = next(i, None)
|
|
if r is None:
|
|
break
|
|
self.assertEqual(expected[r[0]], r[1])
|
|
found_items += 1
|
|
|
|
self.assertEqual(len(expected), found_items)
|
|
|
|
def test_modelbase_several_iters(self):
|
|
mb = ExtraKeysModel()
|
|
it1 = iter(mb)
|
|
it2 = iter(mb)
|
|
|
|
self.assertFalse(it1 is it2)
|
|
self.assertEqual(dict(it1), dict(mb))
|
|
self.assertEqual(dict(it2), dict(mb))
|
|
|
|
def test_extra_keys_empty(self):
|
|
"""Test verifies that by default extra_keys return empty list."""
|
|
self.assertEqual(self.mb._extra_keys, [])
|
|
|
|
def test_extra_keys_defined(self):
|
|
"""Property _extra_keys will return list with attributes names."""
|
|
self.assertEqual(self.ekm._extra_keys, ['name'])
|
|
|
|
def test_model_with_extra_keys(self):
|
|
data = dict(self.ekm)
|
|
self.assertEqual(data, {'smth': None,
|
|
'id': None,
|
|
'name': 'NAME'})
|
|
|
|
|
|
class ExtraKeysModel(BASE, models.ModelBase):
|
|
__tablename__ = 'test_model'
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
smth = Column(String(255))
|
|
|
|
@property
|
|
def name(self):
|
|
return 'NAME'
|
|
|
|
@property
|
|
def _extra_keys(self):
|
|
return ['name']
|
|
|
|
|
|
class TimestampMixinTest(oslo_test.BaseTestCase):
|
|
|
|
def test_timestampmixin_attr(self):
|
|
methods = ('created_at',
|
|
'updated_at')
|
|
for method in methods:
|
|
self.assertTrue(hasattr(models.TimestampMixin, method),
|
|
"Method %s() is not found" % method)
|