Minor UT fixes

microsoft · Jul 18, 2020 · 8408fcd · 8408fcd
1 parent 74bbcc0
commit 8408fcd
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/Tests/GitHubGists.tests.ps1 b/Tests/GitHubGists.tests.ps1
@@ -309,7 +309,10 @@ try
         }
 
         AfterAll {
-            Remove-Item -Path $tempPath -Recurse -ErrorAction SilentlyContinue -Force
+            if (Get-Variable -Name tempPath -ErrorAction SilentlyContinue)
+            {
+                Remove-Item -Path $tempPath -Recurse -ErrorAction SilentlyContinue -Force
+            }
         }
 
         Context 'Download gist content' {

diff --git a/Tests/GitHubRepositories.tests.ps1 b/Tests/GitHubRepositories.tests.ps1
@@ -1363,6 +1363,11 @@ try
     Describe 'GitHubRepositories\Disable-GitHubRepositoryVulnerabilityAlert' {
         BeforeAll {
             $repo = New-GitHubRepository -RepositoryName ([Guid]::NewGuid().Guid)
+
+            # The CI build has been unreliable with this test.
+            # Adding a short sleep to ensure successive queries reflect updated state.
+            Start-Sleep -Seconds $script:defaultSleepSecondsForReliability
+
             Enable-GitHubRepositoryVulnerabilityAlert -Uri  $repo.svn_url
         }
 
@@ -1402,6 +1407,11 @@ try
     Describe 'GitHubRepositories\Disable-GitHubRepositorySecurityFix' {
         BeforeAll {
             $repo = New-GitHubRepository -RepositoryName ([Guid]::NewGuid().Guid)
+
+            # The CI build has been unreliable with this test.
+            # Adding a short sleep to ensure successive queries reflect updated state.
+            Start-Sleep -Seconds $script:defaultSleepSecondsForReliability
+
             Enable-GitHubRepositorySecurityFix -Uri  $repo.svn_url
         }