def __init__(self): from inbox.server.models import new_db_session, init_db, engine from inbox.server.models.tables import IMAPAccount, Namespace, User # create all tables init_db() self.db_session = new_db_session() self.engine = engine # add stub test data user = User(id=1, name="Test User") imapaccount = IMAPAccount(id=1, user_id=1, email_address='*****@*****.**', provider='Gmail', save_raw_messages=1, o_token_issued_to='786647191490.apps.googleusercontent.com', o_user_id='115086935419017912828', o_access_token='ya29.AHES6ZTosKXaQPL5gJxJa16d3r_iclakq6ci_M2LW8dWeZAA63THAA', o_id_token='eyJhbGciOiJSUzI1NiIsImtpZCI6ImU1NTkzYmQ2NTliOTNlOWZiZGQ4OTQ1NDIzNGVhMmQ1YWE2Y2MzYWMifQ.eyJpc3MiOiJhY2NvdW50cy5nb29nbGUuY29tIiwidmVyaWZpZWRfZW1haWwiOiJ0cnVlIiwiZW1haWxfdmVyaWZpZWQiOiJ0cnVlIiwiY2lkIjoiNzg2NjQ3MTkxNDkwLmFwcHMuZ29vZ2xldXNlcmNvbnRlbnQuY29tIiwiYXpwIjoiNzg2NjQ3MTkxNDkwLmFwcHMuZ29vZ2xldXNlcmNvbnRlbnQuY29tIiwidG9rZW5faGFzaCI6IlpabmgzaDZwSlQxX29qRGQ1LU5HNWciLCJhdF9oYXNoIjoiWlpuaDNoNnBKVDFfb2pEZDUtTkc1ZyIsImVtYWlsIjoiaW5ib3hhcHB0ZXN0QGdtYWlsLmNvbSIsImlkIjoiMTE1MDg2OTM1NDE5MDE3OTEyODI4Iiwic3ViIjoiMTE1MDg2OTM1NDE5MDE3OTEyODI4IiwiYXVkIjoiNzg2NjQ3MTkxNDkwLmFwcHMuZ29vZ2xldXNlcmNvbnRlbnQuY29tIiwiaWF0IjoxMzgyNzE4MTE1LCJleHAiOjEzODI3MjIwMTV9.FemBnV73fdeh4zZEbP8NCltsIeNEyrc6wUxX97OourI9eJHdw_RWrsE5QqRthuK9Rg2_UslCCE3daL1S9bOsW-gz7S0XS3fY6-FFSCu77R08PWqRmbTsbqqG4DYaEK3S3uYBfYUqJZBl6hm5BRGQ43BPQqHgHnNTCjduED64Mrs', o_expires_in=3599, o_access_type='offline', o_token_type='Bearer', o_audience='786647191490.apps.googleusercontent.com', o_scope='https://www.googleapis.com/auth/userinfo.email https://www.googleapis.com/auth/userinfo.profile https://mail.google.com/ https://www.google.com/m8/feeds https://www.googleapis.com/auth/calendar', o_refresh_token='1/qY_uxOu0I9RCT7HUisqB3MKVhhb-Hojn6Q2F6QTRwuw', o_verified_email=1, date='2013-10-25 16:26:56') namespace = Namespace(id=1, imapaccount_id=1) self.db_session.add_all([imapaccount, namespace, user])
def __init__(self): from inbox.server.models import new_db_session, init_db, engine # Set up test database init_db() self.db_session = new_db_session() self.engine = engine # Populate with test data self.populate()
from inbox.server.models import init_db, engine_uri from inbox.server.models import engine as inbox_db_engine from inbox.sqlalchemy.util import ForceStrictMode basic_engine = sqlalchemy.create_engine(engine_uri(), listeners=[ForceStrictMode()], echo=False) print 'Creating database: test' basic_engine.execute("CREATE DATABASE IF NOT EXISTS test DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci") basic_engine.execute("GRANT ALL PRIVILEGES ON test.* TO inboxtest@localhost IDENTIFIED BY 'inboxtest'") print 'Creating database: inbox' basic_engine.execute("CREATE DATABASE IF NOT EXISTS inbox DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;") basic_engine.execute("GRANT ALL PRIVILEGES ON inbox.* TO inbox@localhost IDENTIFIED BY 'inbox'") table_mod_for = init_db() # Stamp initial alembic revision if inbox_db_engine.dialect.has_table(inbox_db_engine, "alembic_version"): res = inbox_db_engine.execute("SELECT version_num from alembic_version") current_revision = [r for r in res][0][0] assert current_revision, "Need current revision in alembic_version table..." print 'Already revisioned by alembic %s' % current_revision else: alembic_ini_filename = 'alembic.ini' # top-level, with setup.sh assert os.path.isfile(alembic_ini_filename), "Must have alembic.ini file at %s" % alembic_ini_filename alembic_cfg = alembic_config(alembic_ini_filename) print 'Stamping with alembic revision' alembic_command.stamp(alembic_cfg, "head")