[link-me] Support SSH remote URLs as well

Author: woojiahao

link_me/tests/specs/invalid_ssh.yml

@@ -0,0 +1,9 @@
+initialization:
+  steps:
+    - type: commit
+      empty: true
+      message: Empty commit
+      id: start
+    - type: remote
+      remote-url: git@github.com/git-mastery/link-me.git
+      remote-name: upstream

link_me/tests/specs/valid.yml

@@ -5,5 +5,5 @@ initialization:
       message: Empty commit
       id: start
     - type: remote
-      remote-url: https://github.com/git-mastery/link-me
+      remote-url: https://github.com/git-mastery/link-me.git
       remote-name: upstream

link_me/tests/specs/valid_ssh.yml

@@ -0,0 +1,9 @@
+initialization:
+  steps:
+    - type: commit
+      empty: true
+      message: Empty commit
+      id: start
+    - type: remote
+      remote-url: git@github.com:git-mastery/link-me.git
+      remote-name: upstream

link_me/tests/test_verify.py

@@ -12,6 +12,16 @@ def test_valid():
         assert_output(output, GitAutograderStatus.SUCCESSFUL)
 
 
+def test_valid_ssh():
+    with loader.load("specs/valid_ssh.yml", "start") as output:
+        assert_output(output, GitAutograderStatus.SUCCESSFUL)
+
+
+def test_invalid_ssh():
+    with loader.load("specs/invalid_ssh.yml", "start") as output:
+        assert_output(output, GitAutograderStatus.UNSUCCESSFUL, [WRONG_UPSTREAM_URL])
+
+
 def test_wrong_url():
     with loader.load("specs/wrong_url.yml", "start") as output:
         assert_output(output, GitAutograderStatus.UNSUCCESSFUL, [WRONG_UPSTREAM_URL])

link_me/verify.py

@@ -13,7 +13,7 @@ def verify(exercise: GitAutograderExercise) -> GitAutograderOutput:
         raise exercise.wrong_answer([MISSING_UPSTREAM_REMOTE])
 
     upstream = exercise.repo.remotes.remote("upstream")
-    if not upstream.remote.url.startswith("https://github.com/git-mastery/link-me"):
+    if not upstream.is_for_repo("git-mastery", "link-me"):
         raise exercise.wrong_answer([WRONG_UPSTREAM_URL])
 
     return exercise.to_output(