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)
