20 lines
797 B
Python
20 lines
797 B
Python
import unittest
|
|
from unittest.mock import patch, MagicMock
|
|
from helix.db.sql_manager import SQLManager
|
|
|
|
|
|
class SQLManagerTest(unittest.TestCase):
|
|
def test_uses_url_if_vcap_environment_is_not_set(self):
|
|
create_engine = MagicMock()
|
|
|
|
with patch('sqlalchemy.create_engine', create_engine):
|
|
SQLManager.get_sql_session_maker('postgres://example.com')
|
|
create_engine.assert_called_with('postgres://example.com', echo=False)
|
|
|
|
def test_defaults_to_local_connection_if_vcap_environment_is_not_set_and_url_is_none(self):
|
|
create_engine = MagicMock()
|
|
|
|
with patch('sqlalchemy.create_engine', create_engine):
|
|
SQLManager.get_sql_session_maker(None)
|
|
create_engine.assert_called_with('postgres://pivotal:@localhost/pivotal', echo=True)
|