def test_main(self):
"""Test the high level environment initialization logic."""
self.mox.StubOutWithMock(initalize_environment, 'create_user')
self.mox.StubOutWithMock(initalize_environment, 'create_db_engine')
self.mox.StubOutWithMock(initalize_environment, 'get_env_config')
self.mox.StubOutWithMock(initalize_environment, 'create_db')
self.mox.StubOutWithMock(initalize_environment, 'create_db_lang')
self.mox.StubOutWithMock(initalize_environment, 'install_postgis')
self.mox.StubOutWithMock(database, 'init_db')
mock_connection = MockDBConnection()
mock_engine = MockEngine(mock_connection)
initalize_environment.create_user(DATABASE_USER,
DATABASE_PASSWORD).AndReturn(None)
initalize_environment.create_db_engine(DATABASE_USER,
DATABASE_PASSWORD).AndReturn(mock_engine)
initalize_environment.get_env_config().AndReturn(TEST_CONFIG_SETTINGS)
initalize_environment.create_db(mock_connection,
DATABASE_NAME).AndReturn(None)
initalize_environment.create_db_lang(DATABASE_NAME).AndReturn(None)
initalize_environment.install_postgis(DATABASE_NAME).AndReturn(None)
database.init_db().AndReturn(mock_connection)
self.mox.ReplayAll()
initalize_environment.main(DATABASE_TYPE)
def test_create_db_engine(self):
"""Test creating the sqlalchemy DB engine."""
self.mox.StubOutWithMock(sqlalchemy, 'create_engine')
sqlalchemy.create_engine(ENGINE_URL).AndReturn(True)
self.mox.ReplayAll()
engine = initalize_environment.create_db_engine(DATABASE_USER,
DATABASE_PASSWORD)
self.assertTrue(engine, True)