1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import time
import unittest
from pathlib import Path
import pytest
from huggingface_hub.constants import ENDPOINT
from huggingface_hub.utils import run_interactive_subprocess, run_subprocess
from huggingface_hub.utils._git_credential import (
_parse_credential_output,
list_credential_helpers,
set_git_credential,
unset_git_credential,
)
STORE_AND_CACHE_HELPERS_CONFIG = """
[credential]
helper = store
helper = cache --timeout 30000
helper = git-credential-manager
helper = /usr/libexec/git-core/git-credential-libsecret
"""
@pytest.mark.usefixtures("fx_cache_dir")
class TestGitCredentials(unittest.TestCase):
cache_dir: Path
def setUp(self):
"""Initialize and configure a local repo.
Avoid to configure git helpers globally on a contributor's machine.
"""
run_subprocess("git init", folder=self.cache_dir)
with (self.cache_dir / ".git" / "config").open("w") as f:
f.write(STORE_AND_CACHE_HELPERS_CONFIG)
def test_list_credential_helpers(self) -> None:
helpers = list_credential_helpers(folder=self.cache_dir)
self.assertIn("cache", helpers)
self.assertIn("store", helpers)
self.assertIn("git-credential-manager", helpers)
self.assertIn("/usr/libexec/git-core/git-credential-libsecret", helpers)
def test_set_and_unset_git_credential(self) -> None:
username = "hf_test_user_" + str(round(time.time())) # make username unique
# Set credentials
set_git_credential(token="hf_test_token", username=username, folder=self.cache_dir)
# Check credentials are stored
with run_interactive_subprocess("git credential fill", folder=self.cache_dir) as (stdin, stdout):
stdin.write(f"url={ENDPOINT}\nusername={username}\n\n")
stdin.flush()
output = stdout.read()
self.assertIn("password=hf_test_token", output)
# Unset credentials
unset_git_credential(username=username, folder=self.cache_dir)
# Check credentials are NOT stored
# Cannot check with `git credential fill` as it would hang forever: only
# checking `store` helper instead.
with run_interactive_subprocess("git credential-store get", folder=self.cache_dir) as (stdin, stdout):
stdin.write(f"url={ENDPOINT}\nusername={username}\n\n")
stdin.flush()
output = stdout.read()
self.assertEqual("", output)
def test_git_credential_parsing_regex(self) -> None:
output = """
credential.helper = store
credential.helper = cache --timeout 30000
credential.helper = osxkeychain"""
assert _parse_credential_output(output) == ["cache", "osxkeychain", "store"]
|
