119 lines
4.1 KiB
Python
119 lines
4.1 KiB
Python
# Copyright (c) 2013 Red Hat, Inc.
|
|
#
|
|
# 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 datetime
|
|
|
|
import sqlalchemy as sa
|
|
|
|
from zaqar.queues.storage import pooling
|
|
from zaqar.queues.storage import sqlalchemy
|
|
from zaqar.queues.storage.sqlalchemy import controllers
|
|
from zaqar.queues.storage.sqlalchemy import tables
|
|
from zaqar.queues.storage.sqlalchemy import utils
|
|
from zaqar import tests as testing
|
|
from zaqar.tests.queues.storage import base
|
|
|
|
|
|
class SqlalchemyTableTests(testing.TestBase):
|
|
|
|
def setUp(self):
|
|
super(SqlalchemyTableTests, self).setUp()
|
|
self.engine = sa.create_engine('sqlite:///:memory:')
|
|
tables.metadata.create_all(self.engine, checkfirst=True)
|
|
|
|
def test_table_queries(self):
|
|
self.engine.execute(tables.Queues.insert(), id=1, project='test',
|
|
name='zaqar', metadata=utils.json_encode('aaaa'))
|
|
self.engine.execute(tables.Messages.insert(), id=1, qid=1, ttl=10,
|
|
body=utils.json_encode('bbbb'), client='a',
|
|
created=datetime.datetime.now())
|
|
self.engine.execute(tables.Claims.insert(), id=1, qid=1, ttl=10,
|
|
created=datetime.datetime.now())
|
|
|
|
rs = self.engine.execute(tables.Claims.select())
|
|
row = rs.fetchone()
|
|
|
|
self.assertEqual(row.id, 1)
|
|
self.assertEqual(row.qid, 1)
|
|
self.assertEqual(row.ttl, 10)
|
|
|
|
self.engine.execute(tables.Claims.delete(tables.Claims.c.id == 1))
|
|
rs = self.engine.execute(tables.Claims.select())
|
|
row = rs.fetchone()
|
|
|
|
self.assertIsNone(row)
|
|
|
|
|
|
class SqlalchemyQueueTests(base.QueueControllerTest):
|
|
driver_class = sqlalchemy.DataDriver
|
|
controller_class = controllers.QueueController
|
|
|
|
|
|
class SqlalchemyMessageTests(base.MessageControllerTest):
|
|
driver_class = sqlalchemy.DataDriver
|
|
controller_class = controllers.MessageController
|
|
|
|
|
|
class SqlalchemyClaimTests(base.ClaimControllerTest):
|
|
driver_class = sqlalchemy.DataDriver
|
|
controller_class = controllers.ClaimController
|
|
|
|
|
|
class SqlalchemyPoolsTest(base.PoolsControllerTest):
|
|
driver_class = sqlalchemy.ControlDriver
|
|
controller_class = controllers.PoolsController
|
|
|
|
def setUp(self):
|
|
super(SqlalchemyPoolsTest, self).setUp()
|
|
self.load_conf('wsgi_sqlalchemy.conf')
|
|
|
|
def tearDown(self):
|
|
super(SqlalchemyPoolsTest, self).tearDown()
|
|
|
|
|
|
class SqlalchemyCatalogueTest(base.CatalogueControllerTest):
|
|
driver_class = sqlalchemy.ControlDriver
|
|
controller_class = controllers.CatalogueController
|
|
|
|
def setUp(self):
|
|
super(SqlalchemyCatalogueTest, self).setUp()
|
|
self.load_conf('wsgi_sqlalchemy.conf')
|
|
|
|
def tearDown(self):
|
|
super(SqlalchemyCatalogueTest, self).tearDown()
|
|
|
|
|
|
class PooledMessageTests(base.MessageControllerTest):
|
|
config_file = 'wsgi_sqlalchemy_pooled.conf'
|
|
controller_class = pooling.MessageController
|
|
driver_class = pooling.DataDriver
|
|
control_driver_class = sqlalchemy.ControlDriver
|
|
controller_base_class = pooling.RoutingController
|
|
|
|
|
|
class PooledClaimsTests(base.ClaimControllerTest):
|
|
config_file = 'wsgi_sqlalchemy_pooled.conf'
|
|
controller_class = pooling.ClaimController
|
|
driver_class = pooling.DataDriver
|
|
control_driver_class = sqlalchemy.ControlDriver
|
|
controller_base_class = pooling.RoutingController
|
|
|
|
|
|
class PooledQueueTests(base.QueueControllerTest):
|
|
config_file = 'wsgi_sqlalchemy_pooled.conf'
|
|
controller_class = pooling.QueueController
|
|
driver_class = pooling.DataDriver
|
|
control_driver_class = sqlalchemy.ControlDriver
|
|
controller_base_class = pooling.RoutingController
|