def test_expand_environment_variables_in_connection():
os.environ["DATABASE_URL"] = "postgresql:///shakes"
assert parse("$DATABASE_URL SELECT * FROM work", empty_config) == {
"connection": "postgresql:///shakes",
"sql": "SELECT * FROM work",
"result_var": None,
}
def test_parse_with_sql():
assert parse(
"postgresql://*****:*****@localhost/shakes SELECT * FROM work",
empty_config,
) == {
"connection": "postgresql://*****:*****@localhost/shakes",
"sql": "SELECT * FROM work",
"result_var": None,
}
def test_parse_no_sql():
assert parse("will:longliveliz@localhost/shakes", empty_config) == \
{'connection': "will:longliveliz@localhost/shakes",
'sql': '',
'result_var': None}
def test_expand_environment_variables_in_connection():
os.environ['DATABASE_URL'] = 'postgresql:///shakes'
assert parse("$DATABASE_URL SELECT * FROM work", empty_config) == \
{'connection': "postgresql:///shakes",
'sql': 'SELECT * FROM work',
'flags': default_flags}
def test_parse_postgresql_socket_connection():
assert parse("postgresql:///shakes SELECT * FROM work", empty_config) == \
{'connection': "postgresql:///shakes",
'sql': 'SELECT * FROM work'}
def test_parse_with_sql():
assert parse("postgresql://*****:*****@localhost/shakes SELECT * FROM work",
empty_config) == \
{'connection': "postgresql://*****:*****@localhost/shakes",
'sql': 'SELECT * FROM work'}
def test_parse_shovel_operator():
assert parse("dest << SELECT * FROM work", empty_config) == \
{'connection': "",
'sql': 'SELECT * FROM work',
'result_var': "dest"}
def test_parse_postgresql_socket_connection():
assert parse("postgresql:///shakes SELECT * FROM work", empty_config) == \
{'connection': "postgresql:///shakes",
'sql': 'SELECT * FROM work',
'result_var': None}
def test_parse_sql_only():
assert parse("SELECT * FROM work") == \
{'connection': "",
'sql': 'SELECT * FROM work'}
def test_parse_connect_plus_shovel():
assert parse("sqlite:// dest << SELECT * FROM work", empty_config) == {
"connection": "sqlite://",
"sql": "SELECT * FROM work",
"result_var": "dest",
}
def test_parse_shovel_operator():
assert parse("dest << SELECT * FROM work", empty_config) == {
"connection": "",
"sql": "SELECT * FROM work",
"result_var": "dest",
}
def test_parse_postgresql_socket_connection():
assert parse("postgresql:///shakes SELECT * FROM work", empty_config) == {
"connection": "postgresql:///shakes",
"sql": "SELECT * FROM work",
"result_var": None,
}
def test_parse_sql_only():
assert parse("SELECT * FROM work", empty_config) == {
"connection": "",
"sql": "SELECT * FROM work",
"result_var": None,
}
def test_parse_with_sql():
assert parse("postgresql://*****:*****@localhost/shakes SELECT * FROM work",
empty_config) == \
{'connection': "postgresql://*****:*****@localhost/shakes",
'sql': 'SELECT * FROM work',
'result_var': None}
def test_parse_postgresql_socket_connection():
assert parse("postgresql:///shakes SELECT * FROM work") == \
{'connection': "postgresql:///shakes",
'sql': 'SELECT * FROM work'}
def test_parse_sql_only():
assert parse("SELECT * FROM work", empty_config) == \
{'connection': "",
'sql': 'SELECT * FROM work',
'result_var': None}
def test_parse_no_sql():
assert parse("will:longliveliz@localhost/shakes") == \
{'connection': "will:longliveliz@localhost/shakes",
'sql': ''}
def test_expand_environment_variables_in_connection():
os.environ['DATABASE_URL'] = 'postgresql:///shakes'
assert parse("$DATABASE_URL SELECT * FROM work", empty_config) == \
{'connection': "postgresql:///shakes",
'sql': 'SELECT * FROM work',
'result_var': None}
def test_parse_with_sql():
assert parse("postgresql://*****:*****@localhost/shakes SELECT * FROM work") == \
{'connection': "postgresql://*****:*****@localhost/shakes",
'sql': 'SELECT * FROM work'}
def test_parse_connect_plus_shovel():
assert parse("sqlite:// dest << SELECT * FROM work", empty_config) == \
{'connection': "sqlite://",
'sql': 'SELECT * FROM work',
'result_var': "dest"}
def test_parse_no_sql():
assert parse("will:longliveliz@localhost/shakes", empty_config) == \
{'connection': "will:longliveliz@localhost/shakes",
'sql': '',
'flags': default_flags}
def test_parse_sql_only():
assert parse("SELECT * FROM work", empty_config) == \
{'connection': "",
'sql': 'SELECT * FROM work'}
def test_parse_sql_only():
assert parse("SELECT * FROM work", empty_config) == \
{'connection': "",
'sql': 'SELECT * FROM work',
'flags': default_flags}
def test_parse_no_sql():
assert parse("will:longliveliz@localhost/shakes", empty_config) == \
{'connection': "will:longliveliz@localhost/shakes",
'sql': ''}
def test_parse_no_sql():
assert parse("will:longliveliz@localhost/shakes", empty_config) == {
"connection": "will:longliveliz@localhost/shakes",
"sql": "",
"result_var": None,
}
