def test_can_create_test_db(blank_postgis_session): engine = blank_postgis_session.get_bind() create_test_database(engine) create_test_data(engine) # Get some data q = blank_postgis_session.query(Contact) assert len(q.all()) == 2
import sys import logging from argparse import ArgumentParser from pkg_resources import resource_filename from sqlalchemy import create_engine from pycds.util import create_test_database, create_test_data if __name__ == '__main__': parser = ArgumentParser(description="Script to create a test CRMP database and write it to a test database. DSN strings are of form:\n\tdialect+driver://username:password@host:port/database\nExamples:\n\tpostgresql://scott:tiger@localhost/mydatabase\n\tpostgresql+psycopg2://scott:tiger@localhost/mydatabase\n\tpostgresql+pg8000://scott:tiger@localhost/mydatabase") parser.add_argument("-d", "--dsn", help="Source database DSN from which to read ") parser.add_argument("-t", "--testdsn", help="Destination DSN to write to") args = parser.parse_args() logger = logging.getLogger(__name__) logging.basicConfig(stream=sys.stdout, level=logging.INFO) read_engine = create_engine(args.dsn) write_engine = create_engine(args.testdsn) create_test_database(write_engine) create_test_data(write_engine)