Add tests for repo_add method()

Grokzen · Grokzen · commit f0babcc98cba · 2020-02-10T17:08:43.000+01:00
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -5,6 +5,7 @@
 
 # sgit imports
 from sgit.core import Sgit, DEFAULT_REPO_CONTENT
+from sgit.exceptions import *
 
 # 3rd party imports
 import pytest
@@ -83,3 +84,35 @@ def test_repo_list(sgit):
     ## The default configfile after init_repo is empty and usable here
     retcode = sgit.repo_list()
     assert retcode == 1
+
+
+def test_repo_add(sgit):
+    retcode = sgit.init_repo()
+    assert retcode == None
+
+    ## Test exception failure if we send in the wrong data
+    with pytest.raises(SgitConfigException) as pytest_wrapped_e:
+        sgit.repo_add(None, None, None)
+
+    assert pytest_wrapped_e.type == SgitConfigException
+
+    ## Do a valid add of a new repo
+    name = 'foobar'
+    gitrepo = 'git@github.com/sgit'
+    revision = 'master'
+
+    sgit.repo_add(name, gitrepo, revision)
+
+    saved_data = sgit._get_config_file()
+    assert saved_data == {
+        'repos': {
+            name: {
+                'clone-url': gitrepo,
+                'revision': revision
+            }
+        }
+    }
+
+    ## If rerunning the same config then it should cause an error
+    retcode = sgit.repo_add(name, gitrepo, revision)
+    assert retcode == 1
