tailbone/tests/__init__.py
2014-11-18 23:00:24 -06:00

33 lines
634 B
Python

import unittest
from mock import Mock
from pyramid import testing
class TestCase(unittest.TestCase):
"""
Base class for all test suites.
"""
def setUp(self):
self.config = testing.setUp()
def tearDown(self):
testing.tearDown()
def mock_query():
"""
Mock object used to simulate a ``sqlalchemy.Query`` instance.
"""
query = Mock()
query.return_value = query
query.outerjoin.return_value = query
query.join.return_value = query
query.filter.return_value = query
query.distinct.return_value = query
query.order_by.return_value = query
return query