public void DontDoubleStoreExistingCredential()
{
MockTracer tracer = new MockTracer();
MockGitProcess gitProcess = this.GetGitProcess();
GitAuthentication dut = new GitAuthentication(gitProcess, "mock://repoUrl");
dut.TryInitializeAndRequireAuth(tracer, out _);
string authString;
dut.TryGetCredentials(tracer, out authString, out _).ShouldBeTrue();
dut.ApproveCredentials(tracer, authString);
dut.ApproveCredentials(tracer, authString);
dut.ApproveCredentials(tracer, authString);
dut.ApproveCredentials(tracer, authString);
dut.ApproveCredentials(tracer, authString);
gitProcess.CredentialApprovals["mock://repoUrl"].Count.ShouldEqual(1);
gitProcess.CredentialRejections.Count.ShouldEqual(0);
gitProcess.StoredCredentials.Count.ShouldEqual(1);
gitProcess.StoredCredentials.Single().Key.ShouldEqual("mock://repoUrl");
}
public void DontStoreDifferentCredentialFromCachedValue()
{
MockTracer tracer = new MockTracer();
MockGitProcess gitProcess = this.GetGitProcess();
GitAuthentication dut = new GitAuthentication(gitProcess, "mock://repoUrl");
dut.TryInitializeAndRequireAuth(tracer, out _);
// Get and store an initial value that will be cached
string authString;
dut.TryGetCredentials(tracer, out authString, out _).ShouldBeTrue();
dut.ApproveCredentials(tracer, authString);
// Try and store a different value from the one that is cached
dut.ApproveCredentials(tracer, "different value");
gitProcess.CredentialApprovals["mock://repoUrl"].Count.ShouldEqual(1);
gitProcess.CredentialRejections.Count.ShouldEqual(0);
gitProcess.StoredCredentials.Count.ShouldEqual(1);
gitProcess.StoredCredentials.Single().Key.ShouldEqual("mock://repoUrl");
}
public void BackoffIsNotInEffectAfterSuccess()
{
MockTracer tracer = new MockTracer();
MockGitProcess gitProcess = this.GetGitProcess();
GitAuthentication dut = new GitAuthentication(gitProcess, "mock://repoUrl");
dut.TryInitializeAndRequireAuth(tracer, out _);
string authString;
string error;
for (int i = 0; i < 5; ++i)
{
dut.TryGetCredentials(tracer, out authString, out error).ShouldEqual(true, "Failed to get credential on iteration " + i + ": " + error);
dut.RejectCredentials(tracer, authString);
dut.TryGetCredentials(tracer, out authString, out error).ShouldEqual(true, "Failed to retry getting credential on iteration " + i + ": " + error);
dut.ApproveCredentials(tracer, authString);
dut.IsBackingOff.ShouldEqual(false, "Should reset backoff after successfully refreshing credentials");
gitProcess.CredentialRejections["mock://repoUrl"].Count.ShouldEqual(i + 1, $"Should have {i+1} credentials rejection");
gitProcess.CredentialApprovals["mock://repoUrl"].Count.ShouldEqual(i + 1, $"Should have {i+1} credential approvals");
}
}
public void TwoThreadsInterleavingFailuresShouldntStompASuccess()
{
MockTracer tracer = new MockTracer();
MockGitProcess gitProcess = this.GetGitProcess();
GitAuthentication dut = new GitAuthentication(gitProcess, "mock://repoUrl");
dut.TryInitializeAndRequireAuth(tracer, out _);
string thread1Auth;
string thread2Auth;
string error;
// Populate an initial PAT on two threads
dut.TryGetCredentials(tracer, out thread1Auth, out error).ShouldEqual(true);
dut.TryGetCredentials(tracer, out thread2Auth, out error).ShouldEqual(true);
// Simulate a 401 error on one threads
dut.RejectCredentials(tracer, thread1Auth);
gitProcess.CredentialRejections["mock://repoUrl"].Count.ShouldEqual(1);
gitProcess.CredentialRejections["mock://repoUrl"][0].BasicAuthString.ShouldEqual(thread1Auth);
// That thread then retries and succeeds
dut.TryGetCredentials(tracer, out thread1Auth, out error).ShouldEqual(true);
dut.ApproveCredentials(tracer, thread1Auth);
gitProcess.CredentialApprovals["mock://repoUrl"].Count.ShouldEqual(1);
gitProcess.CredentialApprovals["mock://repoUrl"][0].BasicAuthString.ShouldEqual(thread1Auth);
// If the second thread fails with the old PAT, it shouldn't stomp the new PAT
dut.RejectCredentials(tracer, thread2Auth);
gitProcess.CredentialRejections["mock://repoUrl"].Count.ShouldEqual(1);
// The second thread should be able to get a PAT
dut.TryGetCredentials(tracer, out thread2Auth, out error).ShouldEqual(true);
thread2Auth.ShouldEqual(thread1Auth, "The second thread stomp the first threads good auth string");
}