Safer user agent test

JuliaCloud · Aug 13, 2021 · 1c6c1bf · 1c6c1bf
1 parent 9c2da08
commit 1c6c1bf
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/test/AWS.jl b/test/AWS.jl
@@ -29,12 +29,16 @@ end
 end
 
 @testset "set user agent" begin
+    old_user_agent = AWS.user_agent[]
     new_user_agent = "new user agent"
 
-    @test AWS.user_agent[] == "AWS.jl/1.0.0"
-    set_user_agent(new_user_agent)
-    @test AWS.user_agent[] == new_user_agent
-    set_user_agent("AWS.jl/1.0.0")
+    try
+        @test AWS.user_agent[] == "AWS.jl/1.0.0"
+        set_user_agent(new_user_agent)
+        @test AWS.user_agent[] == new_user_agent
+    finally
+        set_user_agent(old_user_agent)
+    end
 end
 
 @testset "sign" begin