def error_rows(rule_file,
staging_db,
submission=None,
models=None,
assert_num=None):
"""Insert the models into the database, then run the rule SQL against
those models. Return the resulting (invalid) rows"""
if submission is None:
submission = Submission(user_id=None,
reporting_start_date=datetime(2015, 10, 1),
reporting_end_date=datetime(2015, 10, 31))
if models is None:
models = []
submission_id = insert_submission(staging_db, submission)
sql = SQLLoader.readSqlStr(rule_file).format(submission_id)
for model in models:
model.submission_id = submission_id
staging_db.session.add(model)
staging_db.session.commit()
result = staging_db.connection.execute(sql).fetchall()
if assert_num is not None:
assert (len(result) == assert_num)
# clean up
for model in models:
staging_db.session.delete(model)
staging_db.session.commit()
return result
def query_columns(rule_file, staging_db):
sql = SQLLoader.readSqlStr(rule_file).format(randint(1, 9999))
return staging_db.connection.execute(sql).keys()