Kartik
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
"""Tests for _project.py — project_id + branch resolver."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import subprocess
|
|
|
|
|
|
def test_resolve_project_id_from_https_remote(tmp_git_repo):
|
|
from _project import resolve_project_id
|
|
|
|
pid = resolve_project_id(str(tmp_git_repo))
|
|
assert pid == "mem0ai-mem0"
|
|
|
|
|
|
def test_resolve_project_id_from_ssh_remote(tmp_git_repo_ssh):
|
|
from _project import resolve_project_id
|
|
|
|
pid = resolve_project_id(str(tmp_git_repo_ssh))
|
|
assert pid == "acme-cool-project"
|
|
|
|
|
|
def test_resolve_project_id_fallback_basename(tmp_no_git):
|
|
from _project import resolve_project_id
|
|
|
|
pid = resolve_project_id(str(tmp_no_git))
|
|
assert pid == os.path.basename(str(tmp_no_git))
|
|
|
|
|
|
def test_resolve_project_id_from_env(tmp_no_git, monkeypatch):
|
|
from _project import resolve_project_id
|
|
|
|
monkeypatch.setenv("MEM0_PROJECT_ID", "my-override")
|
|
pid = resolve_project_id(str(tmp_no_git))
|
|
assert pid == "my-override"
|
|
|
|
|
|
def test_resolve_project_id_from_project_map(tmp_no_git):
|
|
from _project import resolve_project_id, save_project_mapping
|
|
|
|
save_project_mapping(str(tmp_no_git), "custom-project")
|
|
pid = resolve_project_id(str(tmp_no_git))
|
|
assert pid == "custom-project"
|
|
|
|
|
|
def test_save_project_mapping_creates_file(tmp_no_git):
|
|
from _project import save_project_mapping
|
|
|
|
save_project_mapping(str(tmp_no_git), "test-proj")
|
|
map_path = os.path.expanduser("~/.mem0/project_map.json")
|
|
assert os.path.isfile(map_path)
|
|
with open(map_path) as f:
|
|
data = json.load(f)
|
|
assert data[str(tmp_no_git)] == "test-proj"
|
|
|
|
|
|
def test_resolve_branch_in_git_repo(tmp_git_repo):
|
|
from _project import resolve_branch
|
|
|
|
subprocess.run(
|
|
["git", "checkout", "-b", "feat/test-branch"],
|
|
cwd=tmp_git_repo,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
branch = resolve_branch(str(tmp_git_repo))
|
|
assert branch == "feat/test-branch"
|
|
|
|
|
|
def test_resolve_branch_no_git(tmp_no_git):
|
|
from _project import resolve_branch
|
|
|
|
branch = resolve_branch(str(tmp_no_git))
|
|
assert branch == "unknown"
|
|
|
|
|
|
def test_remote_url_to_slug_various_formats():
|
|
from _project import _remote_url_to_slug
|
|
|
|
assert _remote_url_to_slug("https://github.com/mem0ai/mem0.git") == "mem0ai-mem0"
|
|
assert _remote_url_to_slug("git@github.com:mem0ai/mem0.git") == "mem0ai-mem0"
|
|
assert _remote_url_to_slug("ssh://git@github.com/acme/app.git") == "acme-app"
|
|
assert _remote_url_to_slug("https://gitlab.com/org/sub/repo.git") == "sub-repo"
|
|
assert _remote_url_to_slug("git@bitbucket.org:team/project.git") == "team-project"
|
|
|
|
|
|
def test_remote_url_to_slug_no_git_suffix():
|
|
from _project import _remote_url_to_slug
|
|
|
|
assert _remote_url_to_slug("https://github.com/foo/bar") == "foo-bar"
|
|
|
|
|
|
def test_resolve_project_id_priority_order(tmp_git_repo, monkeypatch):
|
|
"""Env var > project_map > git remote > basename."""
|
|
from _project import resolve_project_id, save_project_mapping
|
|
|
|
# Git remote gives "mem0ai-mem0"
|
|
assert resolve_project_id(str(tmp_git_repo)) == "mem0ai-mem0"
|
|
|
|
# project_map overrides git remote
|
|
save_project_mapping(str(tmp_git_repo), "from-map")
|
|
assert resolve_project_id(str(tmp_git_repo)) == "from-map"
|
|
|
|
# Env var overrides everything
|
|
monkeypatch.setenv("MEM0_PROJECT_ID", "from-env")
|
|
assert resolve_project_id(str(tmp_git_repo)) == "from-env"
|