1 files changed, 73 insertions, 0 deletions
diff --git a/src/vgstash/test_vgstash_cli.py b/src/vgstash/test_vgstash_cli.py
new file mode 100644
index 0000000..ac123ef
--- /dev/null
+++ b/src/vgstash/test_vgstash_cli.py
@@ -0,0 +1,73 @@
+import click
+import os
+import pytest
+import vgstash
+import vgstash_cli
+
+from click.testing import CliRunner
+
+verbose = False
+
+def test_init():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['init'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Initializing the database...\nSchema created.\n"
+
+def test_add_minimum():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['add', 'Super Mario Bros.', 'NES'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Added Super Mario Bros. for NES. You physically own it and are playing it.\n"
+
+def test_add_ownership():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['add', 'The Legend of Zelda', 'NES', 'd'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Added The Legend of Zelda for NES. You digitally own it and are playing it.\n"
+
+def test_add_typical():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['add', 'Sonic the Hedgehog 2', 'Genesis', '0', '3'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Added Sonic the Hedgehog 2 for Genesis. You do not own it and have beaten it.\n"
+
+def test_add_full():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['add', 'Vectorman', 'Genesis', 'u', 'b', 'beep'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Added Vectorman for Genesis. You do not own it and have beaten it. It also has notes.\n"
+
+def test_list():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.list_games)
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == '\n'.join((
+        'Sonic the Hedgehog 2|Genesis|0|3|',
+        'Vectorman|Genesis|0|3|beep',
+        'Super Mario Bros.|NES|1|2|',
+        'The Legend of Zelda|NES|2|2|\n',
+    ))
+
+def test_list_filter():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['list', 'playlog'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == '\n'.join((
+        'Super Mario Bros.|NES|1|2|',
+        'The Legend of Zelda|NES|2|2|\n',
+    ))