def queue(queue_setup, test_db_args, test_db_database):
test_db_args['database'] = test_db_database
q = sql_step_queue.SQLStepQueue('test').connect(**test_db_args)
with q._db_conn() as conn:
conn.query('DELETE FROM %s' % q.table_name)
return q
def test_ensure_connected():
q = sql_step_queue.SQLStepQueue('bad_queue')
with pytest.raises(exceptions.NotConnected):
q.qsize()
with pytest.raises(exceptions.NotConnected):
q.enqueue({})
with pytest.raises(exceptions.NotConnected):
q.start()
def queue_setup(request, test_db_args, test_db_database):
with database.connect(**test_db_args) as conn:
conn.execute('CREATE DATABASE IF NOT EXISTS %s' % test_db_database)
test_db_args['database'] = test_db_database
q = sql_step_queue.SQLStepQueue('test').connect(**test_db_args).setup()
def cleanup():
q.destroy()
request.addfinalizer(cleanup)