mirror of
https://github.com/NixOS/nix.git
synced 2025-11-08 19:46:02 +01:00
0e35cd6f3e
* It is tough to contribute to a project that doesn't use a formatter, * It is extra hard to contribute to a project which has configured the formatter, but ignores it for some files * Code formatting makes it harder to hide obscure / weird bugs by accident or on purpose, Let's rip the bandaid off? Note that PRs currently in flight should be able to be merged relatively easily by applying `clang-format` to their tip prior to merge. Co-authored-by: Graham Christensen <graham@grahamc.com>
102 lines
3.5 KiB
C++
102 lines
3.5 KiB
C++
#include <nlohmann/json.hpp>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "nix/fetchers/fetchers.hh"
|
|
#include "nix/fetchers/fetch-settings.hh"
|
|
#include "nix/util/json-utils.hh"
|
|
#include "nix/util/tests/characterization.hh"
|
|
|
|
namespace nix::fetchers {
|
|
|
|
using nlohmann::json;
|
|
|
|
class AccessKeysTest : public ::testing::Test
|
|
{
|
|
protected:
|
|
|
|
public:
|
|
void SetUp() override
|
|
{
|
|
experimentalFeatureSettings.experimentalFeatures.get().insert(Xp::Flakes);
|
|
}
|
|
|
|
void TearDown() override {}
|
|
};
|
|
|
|
TEST_F(AccessKeysTest, singleOrgGitHub)
|
|
{
|
|
fetchers::Settings fetchSettings = fetchers::Settings{};
|
|
fetchSettings.accessTokens.get().insert({"github.com/a", "token"});
|
|
auto i = Input::fromURL(fetchSettings, "github:a/b");
|
|
|
|
auto token = i.scheme->getAccessToken(fetchSettings, "github.com", "github.com/a/b");
|
|
ASSERT_EQ(token, "token");
|
|
}
|
|
|
|
TEST_F(AccessKeysTest, nonMatches)
|
|
{
|
|
fetchers::Settings fetchSettings = fetchers::Settings{};
|
|
fetchSettings.accessTokens.get().insert({"github.com", "token"});
|
|
auto i = Input::fromURL(fetchSettings, "gitlab:github.com/evil");
|
|
|
|
auto token = i.scheme->getAccessToken(fetchSettings, "gitlab.com", "gitlab.com/github.com/evil");
|
|
ASSERT_EQ(token, std::nullopt);
|
|
}
|
|
|
|
TEST_F(AccessKeysTest, noPartialMatches)
|
|
{
|
|
fetchers::Settings fetchSettings = fetchers::Settings{};
|
|
fetchSettings.accessTokens.get().insert({"github.com/partial", "token"});
|
|
auto i = Input::fromURL(fetchSettings, "github:partial-match/repo");
|
|
|
|
auto token = i.scheme->getAccessToken(fetchSettings, "github.com", "github.com/partial-match");
|
|
ASSERT_EQ(token, std::nullopt);
|
|
}
|
|
|
|
TEST_F(AccessKeysTest, repoGitHub)
|
|
{
|
|
fetchers::Settings fetchSettings = fetchers::Settings{};
|
|
fetchSettings.accessTokens.get().insert({"github.com", "token"});
|
|
fetchSettings.accessTokens.get().insert({"github.com/a/b", "another_token"});
|
|
fetchSettings.accessTokens.get().insert({"github.com/a/c", "yet_another_token"});
|
|
auto i = Input::fromURL(fetchSettings, "github:a/a");
|
|
|
|
auto token = i.scheme->getAccessToken(fetchSettings, "github.com", "github.com/a/a");
|
|
ASSERT_EQ(token, "token");
|
|
|
|
token = i.scheme->getAccessToken(fetchSettings, "github.com", "github.com/a/b");
|
|
ASSERT_EQ(token, "another_token");
|
|
|
|
token = i.scheme->getAccessToken(fetchSettings, "github.com", "github.com/a/c");
|
|
ASSERT_EQ(token, "yet_another_token");
|
|
}
|
|
|
|
TEST_F(AccessKeysTest, multipleGitLab)
|
|
{
|
|
fetchers::Settings fetchSettings = fetchers::Settings{};
|
|
fetchSettings.accessTokens.get().insert({"gitlab.com", "token"});
|
|
fetchSettings.accessTokens.get().insert({"gitlab.com/a/b", "another_token"});
|
|
auto i = Input::fromURL(fetchSettings, "gitlab:a/b");
|
|
|
|
auto token = i.scheme->getAccessToken(fetchSettings, "gitlab.com", "gitlab.com/a/b");
|
|
ASSERT_EQ(token, "another_token");
|
|
|
|
token = i.scheme->getAccessToken(fetchSettings, "gitlab.com", "gitlab.com/a/c");
|
|
ASSERT_EQ(token, "token");
|
|
}
|
|
|
|
TEST_F(AccessKeysTest, multipleSourceHut)
|
|
{
|
|
fetchers::Settings fetchSettings = fetchers::Settings{};
|
|
fetchSettings.accessTokens.get().insert({"git.sr.ht", "token"});
|
|
fetchSettings.accessTokens.get().insert({"git.sr.ht/~a/b", "another_token"});
|
|
auto i = Input::fromURL(fetchSettings, "sourcehut:a/b");
|
|
|
|
auto token = i.scheme->getAccessToken(fetchSettings, "git.sr.ht", "git.sr.ht/~a/b");
|
|
ASSERT_EQ(token, "another_token");
|
|
|
|
token = i.scheme->getAccessToken(fetchSettings, "git.sr.ht", "git.sr.ht/~a/c");
|
|
ASSERT_EQ(token, "token");
|
|
}
|
|
|
|
} // namespace nix::fetchers
|