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)