summaryrefslogtreecommitdiff
path: root/tests/test_insert.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_insert.py')
-rw-r--r--tests/test_insert.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/test_insert.py b/tests/test_insert.py
index caaf7ff..384ae49 100644
--- a/tests/test_insert.py
+++ b/tests/test_insert.py
@@ -1,4 +1,16 @@
+from src.program import Program
+from src.exceptions import IncompatibleRowLengthError
+import pytest
+
+
def test_insert(capsys, insert_program):
insert_program.run()
- assert capsys.readouterr()[0] == 'Row added!\n' \ No newline at end of file
+ assert capsys.readouterr()[0] == 'Row added!\n'
+
+
+def test_out_of_bounds_insert(capsys):
+ Program(args=['-f', './tests/test_table.csv', '-c', 'insert', '1,2,3,4']).run()
+
+ assert capsys.readouterr()[0] == 'Incompatible row length, got 4, expected 3\n'
+