def step_see_data(context, which): """Select data from expanded test table.""" if which == "expanded": wrappers.expect_pager( context, dedent("""\ -[ RECORD 1 ]-------------------------\r x | 1\r y | 1.0\r z | 1.0000\r SELECT 1\r """), timeout=1, ) else: wrappers.expect_pager( context, dedent("""\ +---+-----+--------+\r | x | y | z |\r |---+-----+--------|\r | 1 | 1.0 | 1.0000 |\r +---+-----+--------+\r SELECT 1\r """), timeout=1, )
def step_see_large_results(context): wrappers.expect_pager( context, dedent( """\ -[ RECORD 1 ]-------------------------\r ?column? | 1\r ?column? | 2\r ?column? | 3\r ?column? | 4\r ?column? | 5\r ?column? | 6\r ?column? | 7\r ?column? | 8\r ?column? | 9\r ?column? | 10\r ?column? | 11\r ?column? | 12\r ?column? | 13\r ?column? | 14\r ?column? | 15\r ?column? | 16\r ?column? | 17\r ?column? | 18\r ?column? | 19\r ?column? | 20\r ?column? | 21\r ?column? | 22\r ?column? | 23\r ?column? | 24\r ?column? | 25\r ?column? | 26\r ?column? | 27\r ?column? | 28\r ?column? | 29\r ?column? | 30\r ?column? | 31\r ?column? | 32\r ?column? | 33\r ?column? | 34\r ?column? | 35\r ?column? | 36\r ?column? | 37\r ?column? | 38\r ?column? | 39\r ?column? | 40\r ?column? | 41\r ?column? | 42\r ?column? | 43\r ?column? | 44\r ?column? | 45\r ?column? | 46\r ?column? | 47\r ?column? | 48\r ?column? | 49\r SELECT 1\r """ ), timeout=5, )
def step_see_data(context, which): """Select data from expanded test table.""" if which == 'expanded': wrappers.expect_pager( context, dedent('''\ -[ RECORD 1 ]-------------------------\r x | 1\r y | 1.0\r z | 1.0000\r SELECT 1\r '''), timeout=1) else: wrappers.expect_pager( context, dedent('''\ +-----+-----+--------+\r | x | y | z |\r |-----+-----+--------|\r | 1 | 1.0 | 1.0000 |\r +-----+-----+--------+\r SELECT 1\r '''), timeout=1)
def step_tee_ouptut(context): context.tee_file_name = '../tee_file_{0}.sql'.format(context.conf['vi']) if os.path.exists(context.tee_file_name): os.remove(context.tee_file_name) context.cli.sendline('tee {0}'.format( os.path.basename(context.tee_file_name))) wrappers.expect_pager(context, "\r\n", timeout=5)
def step_see_refresh_started(context): """ Wait to see refresh output. """ wrappers.expect_pager( context, 'Auto-completion refresh started in the background.\r\n', timeout=2)
def step_see_large_results(context): rows = ['{n:3}| {n}'.format(n=str(n)) for n in range(1, 50)] expected = ('***************************[ 1. row ]' '***************************\r\n' + '{}\r\n'.format('\r\n'.join(rows) + '\r\n')) wrappers.expect_pager(context, expected, timeout=5) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_large_results(context): rows = ['{n:3}| {n}'.format(n=str(n)) for n in range(1, 50)] expected = ('***************************[ 1. row ]' '***************************\r\n' + '{}\r\n'.format('\r\n'.join(rows) + '\r\n')) wrappers.expect_pager(context, expected, timeout=10) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_small_results(context): wrappers.expect_pager(context, dedent("""\ +---+\r | 1 |\r +---+\r | 1 |\r +---+\r """), timeout=5) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_small_results(context): wrappers.expect_pager(context, dedent("""\ +------------+\r | ?column? |\r |------------|\r | 1 |\r +------------+\r SELECT 1\r """), timeout=5)
def step_prepare_data(context): """Create table, insert a record.""" context.cli.sendline('drop table if exists a;') wrappers.wait_prompt(context) context.cli.sendline( 'create table a(x integer, y real, z numeric(10, 4));') wrappers.expect_pager(context, 'CREATE TABLE\r\n', timeout=2) context.cli.sendline('''insert into a(x, y, z) values(1, 1.0, 1.0);''') wrappers.expect_pager(context, 'INSERT 0 1\r\n', timeout=2)
def step_prepare_data(context): """Create table, insert a record.""" context.cli.sendline('drop table if exists a;') wrappers.wait_prompt(context) context.cli.sendline( 'create table a(x integer, y real, z numeric(10, 4));') wrappers.expect_pager(context, 'CREATE TABLE\r\n', timeout=2) context.cli.sendline('''insert into a(x, y, z) values(1, 1.0, 1.0);''') wrappers.expect_pager(context, 'INSERT 0 1\r\n', timeout=2)
def step_query_select_123456(context): context.cli.sendline('select 123456') wrappers.expect_pager(context, dedent("""\ +--------+\r | 123456 |\r +--------+\r | 123456 |\r +--------+\r 1 row in set\r """), timeout=5)
def step_query_select_123456(context): context.cli.sendline('select 123456') wrappers.expect_pager(context, dedent("""\ +--------+\r | 123456 |\r +--------+\r | 123456 |\r +--------+\r """), timeout=5) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_null_selected(context): """Wait to see null output.""" wrappers.expect_pager( context, dedent("""\ +--------+\r | NULL |\r +--------+\r | <null> |\r +--------+\r """), timeout=1) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_data_selected(context): """Wait to see select output.""" wrappers.expect_pager( context, dedent("""\ +-----+\r | x |\r +-----+\r | yyy |\r +-----+\r """), timeout=2) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_null_selected(context): """Wait to see null output.""" wrappers.expect_pager( context, dedent("""\ +--------+\r | NULL |\r +--------+\r | <null> |\r +--------+\r """), timeout=2) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_small_results(context): wrappers.expect_pager(context, dedent("""\ +---+\r | 1 |\r +---+\r | 1 |\r +---+\r """), timeout=5) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_see_large_results(context): wrappers.expect_pager(context, dedent("""\ -[ RECORD 1 ]-------------------------\r ?column? | 1\r ?column? | 2\r ?column? | 3\r ?column? | 4\r ?column? | 5\r ?column? | 6\r ?column? | 7\r ?column? | 8\r ?column? | 9\r ?column? | 10\r ?column? | 11\r ?column? | 12\r ?column? | 13\r ?column? | 14\r ?column? | 15\r ?column? | 16\r ?column? | 17\r ?column? | 18\r ?column? | 19\r ?column? | 20\r ?column? | 21\r ?column? | 22\r ?column? | 23\r ?column? | 24\r ?column? | 25\r ?column? | 26\r ?column? | 27\r ?column? | 28\r ?column? | 29\r ?column? | 30\r ?column? | 31\r ?column? | 32\r ?column? | 33\r ?column? | 34\r ?column? | 35\r ?column? | 36\r ?column? | 37\r ?column? | 38\r ?column? | 39\r ?column? | 40\r ?column? | 41\r ?column? | 42\r ?column? | 43\r ?column? | 44\r ?column? | 45\r ?column? | 46\r ?column? | 47\r ?column? | 48\r ?column? | 49\r SELECT 1\r """), timeout=5)
def step_see_data_selected(context): """Wait to see select output.""" wrappers.expect_pager( context, dedent("""\ +-----+\r | x |\r +-----+\r | yyy |\r +-----+\r """), timeout=1) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_query_select_number(context, param): context.cli.sendline(u'select {}'.format(param)) wrappers.expect_pager(context, dedent(u"""\ +{dashes}+\r | {param} |\r +{dashes}+\r | {param} |\r +{dashes}+\r \r """.format(param=param, dashes='-' * (len(param) + 2)) ), timeout=5) wrappers.expect_exact(context, '1 row in set', timeout=2)
def step_prepare_data(context): """Create table, insert a record.""" context.cli.sendline('drop table if exists a;') wrappers.expect_exact( context, 'You\'re about to run a destructive command.\r\nDo you want to proceed? (y/n):', timeout=2) context.cli.sendline('y') wrappers.wait_prompt(context) context.cli.sendline( 'create table a(x integer, y real, z numeric(10, 4));') wrappers.expect_pager(context, 'CREATE TABLE\r\n', timeout=2) context.cli.sendline('''insert into a(x, y, z) values(1, 1.0, 1.0);''') wrappers.expect_pager(context, 'INSERT 0 1\r\n', timeout=2)
def step_see_data_selected(context): """ Wait to see select output. """ wrappers.expect_pager(context, dedent('''\ +-----+\r | x |\r |-----|\r | yyy |\r +-----+\r SELECT 1\r '''), timeout=1)
def step_see_data_selected(context): """ Wait to see select output. """ wrappers.expect_pager( context, dedent('''\ +-----+\r | x |\r |-----|\r | yyy |\r +-----+\r SELECT 1\r '''), timeout=1)
def step_see_record_updated(context): """ Wait to see update output. """ wrappers.expect_pager(context, 'UPDATE 1\r\n', timeout=2)
def step_see_record_inserted(context): """ Wait to see insert output. """ wrappers.expect_pager(context, 'INSERT 0 1\r\n', timeout=2)
def step_see_db_created(context): """ Wait to see create database output. """ wrappers.expect_pager(context, 'CREATE DATABASE\r\n', timeout=5)
def step_see_db_created(context): """Wait to see create database output.""" wrappers.expect_pager(context, 'Query OK, 1 row affected\r\n', timeout=2)
def step_see_record_inserted(context): """ Wait to see insert output. """ wrappers.expect_pager(context, 'INSERT 0 1\r\n', timeout=2)
def step_see_table_dropped(context): """ Wait to see drop output. """ wrappers.expect_pager(context, 'DROP TABLE\r\n', timeout=2)
def step_see_named_query_saved(context): """ Wait to see query saved. """ wrappers.expect_pager(context, 'Saved.\r\n', timeout=1)
def step_see_data_deleted(context): """Wait to see delete output.""" wrappers.expect_pager(context, 'Query OK, 1 row affected\r\n', timeout=2)
def step_see_record_updated(context): """Wait to see update output.""" wrappers.expect_pager(context, 'Query OK, 1 row affected\r\n', timeout=2)
def step_see_table_created(context): """Wait to see create table output.""" wrappers.expect_pager(context, 'Query OK, 0 rows affected\r\n', timeout=2)
def step_see_db_dropped(context): """ Wait to see drop database output. """ wrappers.expect_pager(context, "DROP DATABASE\r\n", timeout=2)
def step_see_refresh_started(context): """ Wait to see refresh output. """ wrappers.expect_pager( context, 'Auto-completion refresh started in the background.\r\n', timeout=2)
def step_see_named_query_deleted(context): """ Wait to see query deleted. """ wrappers.expect_pager(context, 'foo: Deleted\r\n', timeout=1)
def step_see_data_deleted(context): """ Wait to see delete output. """ wrappers.expect_pager(context, 'DELETE 1\r\n', timeout=2)
def step_see_db_created(context): """ Wait to see create database output. """ wrappers.expect_pager(context, "CREATE DATABASE\r\n", timeout=5)
def step_see_table_created(context): """ Wait to see create table output. """ wrappers.expect_pager(context, 'CREATE TABLE\r\n', timeout=2)
def step_see_data_deleted(context): """ Wait to see delete output. """ wrappers.expect_pager(context, 'DELETE 1\r\n', timeout=2)
def step_see_record_updated(context): """ Wait to see update output. """ wrappers.expect_pager(context, 'UPDATE 1\r\n', timeout=2)
def step_see_table_dropped(context): """ Wait to see drop output. """ wrappers.expect_pager(context, 'DROP TABLE\r\n', timeout=2)
def step_see_db_dropped(context): """Wait to see drop database output.""" wrappers.expect_pager(context, 'Query OK, 0 rows affected\r\n', timeout=2)
def step_see_table_created(context): """ Wait to see create table output. """ wrappers.expect_pager(context, 'CREATE TABLE\r\n', timeout=2)
def step_notee_output(context): context.cli.sendline('notee') wrappers.expect_pager(context, "\r\n", timeout=5)
def step_see_db_dropped(context): """ Wait to see drop database output. """ wrappers.expect_pager(context, 'DROP DATABASE\r\n', timeout=2)