mirror of
https://github.com/sprockets/sprockets.clients.postgresql.git
synced 2024-12-29 11:17:11 +00:00
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
"""
|
|
Tests for the sprockets.clients.postgresql package
|
|
|
|
"""
|
|
import mock
|
|
import os
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
from sprockets.clients import postgresql
|
|
|
|
|
|
class TestGetURI(unittest.TestCase):
|
|
|
|
def test_get_uri_returns_proper_values(self):
|
|
|
|
os.environ['TEST1_HOST'] = 'test1-host'
|
|
os.environ['TEST1_PORT'] = '5436'
|
|
os.environ['TEST1_DBNAME'] = 'test1'
|
|
os.environ['TEST1_USER'] = 'foo1'
|
|
os.environ['TEST1_PASSWORD'] = 'baz1'
|
|
|
|
self.assertEqual(postgresql._get_uri('test1'),
|
|
'postgresql://foo1:baz1@test1-host:5436/test1')
|
|
|
|
|
|
class TestSession(unittest.TestCase):
|
|
|
|
@mock.patch('queries.session.Session.__init__')
|
|
def setUp(self, mock_init):
|
|
self.mock_init = mock_init
|
|
os.environ['TEST2_HOST'] = 'db1'
|
|
os.environ['TEST2_PORT'] = '5433'
|
|
os.environ['TEST2_DBNAME'] = 'bar'
|
|
os.environ['TEST2_USER'] = 'foo'
|
|
os.environ['TEST2_PASSWORD'] = 'baz'
|
|
self.session = postgresql.Session('test2')
|
|
|
|
def test_session_invokes_queries_session(self):
|
|
self.assertTrue(self.mock_init.called)
|
|
|
|
|
|
class TestTornadoSession(unittest.TestCase):
|
|
|
|
@mock.patch('queries.tornado_session.TornadoSession.__init__')
|
|
def setUp(self, mock_init):
|
|
self.mock_init = mock_init
|
|
os.environ['TEST3_HOST'] = 'db1'
|
|
os.environ['TEST3_PORT'] = '5434'
|
|
os.environ['TEST3_DBNAME'] = 'bar'
|
|
os.environ['TEST3_USER'] = 'foo'
|
|
os.environ['TEST3_PASSWORD'] = 'baz'
|
|
self.session = postgresql.TornadoSession('test3')
|
|
|
|
def test_session_invokes_queries_session(self):
|
|
self.assertTrue(self.mock_init.called)
|