feat: added test

cyclotruc · Jan 1, 2025 · 747d1f8 · 747d1f8
1 parent acfd915
commit 747d1f8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/gitingest/parse_query.py b/src/gitingest/parse_query.py
@@ -116,6 +116,7 @@ def _parse_url(url: str) -> dict[str, Any]:
     _id = str(uuid.uuid4())
     slug = f"{user_name}-{repo_name}"
 
+    final_url = f"https://{domain}/{user_name}/{repo_name}" if url.startswith("https://") else f"http://{domain}/{user_name}/{repo_name}"
     parsed = {
         "user_name": user_name,
         "repo_name": repo_name,
@@ -124,7 +125,7 @@ def _parse_url(url: str) -> dict[str, Any]:
         "commit": None,
         "subpath": "/",
         "local_path": f"{TMP_BASE_PATH}/{_id}/{slug}",
-        "url": f"https://{domain}/{user_name}/{repo_name}",
+        "url": final_url,
         "slug": slug,
         "id": _id,
     }

diff --git a/src/gitingest/tests/test_parse_query.py b/src/gitingest/tests/test_parse_query.py
@@ -9,6 +9,7 @@ def test_parse_url_valid() -> None:
         "https://github.com/user/repo",
         "https://gitlab.com/user/repo",
         "https://bitbucket.org/user/repo",
+        "http://github.com/user/repo",
     ]
     for url in test_cases:
         result = _parse_url(url)