[Improvement] improve github loader (#984)

This commit is contained in:
Deven Patel
2023-12-01 11:24:13 -08:00
committed by GitHub
parent b02e8feeda
commit 36af1a7615
8 changed files with 487 additions and 91 deletions

View File

@@ -0,0 +1,33 @@
import pytest
from embedchain.loaders.github import GithubLoader
@pytest.fixture
def mock_github_loader_config():
return {
"token": "your_mock_token",
}
@pytest.fixture
def mock_github_loader(mocker, mock_github_loader_config):
mock_github = mocker.patch("github.Github")
_ = mock_github.return_value
return GithubLoader(config=mock_github_loader_config)
def test_github_loader_init(mocker, mock_github_loader_config):
mock_github = mocker.patch("github.Github")
GithubLoader(config=mock_github_loader_config)
mock_github.assert_called_once_with("your_mock_token")
def test_github_loader_init_empty_config(mocker):
with pytest.raises(ValueError, match="requires a personal access token"):
GithubLoader()
def test_github_loader_init_missing_token():
with pytest.raises(ValueError, match="requires a personal access token"):
GithubLoader(config={})