def test_process_runaway(cursor, output):
block = textwrap.dedent("""
```sql
SELECT 1+1;
""")
process(cursor, block, output)
assert "Warning" in output.getvalue()
def test_process_unknown_args(cursor, output):
block = textwrap.dedent("""
```sql
--captain="hello"
SELECT 1+1;
```
""")
process(cursor, block, output)
assert "Unknown" in output.getvalue()
def test_process_show_caption(cursor, output):
block = textwrap.dedent("""
```sql
--caption="hello"
SELECT 1+1;
```
""")
process(cursor, block, output)
assert "hello" in output.getvalue()
def test_process_hide_headers(cursor, output):
block = textwrap.dedent("""
```sql
--hide-headers
SELECT 1+1;
```
""")
process(cursor, block, output)
assert "2" in output.getvalue()
def test_block_process(cursor, output):
block = textwrap.dedent("""
```sql
CREATE TABLE Foo (id INTEGER PRIMARY KEY, data INTEGER);
INSERT INTO Foo (data) VALUES (1);
```
""")
process(cursor, block, output)
cursor.execute("SELECT * FROM Foo")
assert len(cursor.fetchall()) == 1
def test_process_failure(cursor, output):
block = textwrap.dedent("""
```sql
SELCT 1+1;
```
""")
process(cursor, block, output)
print("test_process_failure")
print(output.getvalue())
print("test_process_failure end")
assert "Error" in output.getvalue()
def process_file(filename: str, database: str, file=sys.stdout):
blocks = open(filename).read().split('\n\n')
cursor = get_cursor(database)
for block in blocks:
process(cursor, block, file=file)
def test_process_normal_text(cursor, output):
block = textwrap.dedent("""
Hello, World!
""")
process(cursor, block, output)
assert "Hello" in output.getvalue()