diff --git a/tests/Tgstation.Server.Tests/Instance/RepositoryTest.cs b/tests/Tgstation.Server.Tests/Instance/RepositoryTest.cs index 02b3251622..1bdb2bb7fb 100644 --- a/tests/Tgstation.Server.Tests/Instance/RepositoryTest.cs +++ b/tests/Tgstation.Server.Tests/Instance/RepositoryTest.cs @@ -22,28 +22,13 @@ namespace Tgstation.Server.Tests.Instance public async Task RunPreWatchdog(CancellationToken cancellationToken) { - const string GitHubRef = "TGS4_GITHUB_REF"; - var branchSourceEnvVars = new List - { - "TGS4_TEST_BRANCH", - "APPVEYOR_REPO_BRANCH", - "TRAVIS_BRANCH", - GitHubRef - }; - + const string TestRefEnvVar = "TGS4_GITHUB_REF"; + var envVar = Environment.GetEnvironmentVariable(TestRefEnvVar); string workingBranch = null; - foreach (var envVarName in branchSourceEnvVars) + if (!String.IsNullOrWhiteSpace(envVar)) { - var envVar = Environment.GetEnvironmentVariable(envVarName); - if (!String.IsNullOrWhiteSpace(envVar)) - { - if(envVarName == GitHubRef) - envVar = envVar.Substring("refs/heads/".Length); - - workingBranch = envVar; - Console.WriteLine($"TEST: Set working branch to '{workingBranch}' from env var '{envVarName}'"); - break; - } + workingBranch = envVar; + Console.WriteLine($"TEST: Set working branch to '{workingBranch}' from env var '{TestRefEnvVar}'"); } if (workingBranch == null)