1 files changed, 17 insertions, 4 deletions
diff --git a/src/vgstash/test_vgstash_cli.py b/src/vgstash/test_vgstash_cli.py
index 941179a..098c39e 100644
--- a/src/vgstash/test_vgstash_cli.py
+++ b/src/vgstash/test_vgstash_cli.py
@@ -6,6 +6,8 @@ import vgstash_cli
 
 from click.testing import CliRunner
 
+# TODO: Setup a testing directory for test data and integrate with pytest
+
 verbose = True
 interactive = False
 
@@ -182,6 +184,7 @@ def test_update():
         'The Legend  |   NES    |   D |   P\n'
     ))
 
+
 def test_notes():
     runner = CliRunner()
     result = runner.invoke(vgstash_cli.cli, ['notes', 'Vectorman 2', 'Genesis'])
@@ -195,6 +198,7 @@ def test_notes():
         'boop\n'
     ))
 
+
 def test_notes_edit():
     if not interactive:
         return
@@ -215,11 +219,11 @@ def test_notes_edit():
 
 def test_import_file_yaml():
     runner = CliRunner()
-    result = runner.invoke(vgstash_cli.cli, ["import", "test.yml"])
+    result = runner.invoke(vgstash_cli.cli, ["import", "test_import.yml"])
     if verbose:
         print(result.output)
     assert result.exit_code == 0
-    assert result.output == "Successfully imported 2 games from {}.\n".format(os.path.join(os.getcwd(), "test.yml"))
+    assert result.output == "Successfully imported 2 games from {}.\n".format(os.path.join(os.getcwd(), "test_import.yml"))
 
     # List the results to make sure they match what the editor has.
     list_runner = CliRunner()
@@ -231,11 +235,11 @@ def test_import_file_yaml():
 
 def test_import_file_yaml_update():
     runner = CliRunner()
-    result = runner.invoke(vgstash_cli.cli, ["import", "test.yml", "-u"])
+    result = runner.invoke(vgstash_cli.cli, ["import", "test_import.yml", "-u"])
     if verbose:
         print(result.output)
     assert result.exit_code == 0
-    assert result.output == "Successfully imported 3 games from {}.\n".format(os.path.join(os.getcwd(), "test.yml"))
+    assert result.output == "Successfully imported 3 games from {}.\n".format(os.path.join(os.getcwd(), "test_import.yml"))
 
     # List the results to make sure they match what the editor has.
     list_runner = CliRunner()
@@ -243,3 +247,12 @@ def test_import_file_yaml_update():
     if verbose:
         print(list_result.output)
     assert list_result.exit_code == 0
+
+
+def test_export_file_yaml():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ["export", "-f", "yaml", "test_export.yml"])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Successfully exported 6 games to {}.\n".format(os.path.join(os.getcwd(), "test_export.yml"))