retry failing tests

Author: efcasado

samples/client/petstore/elixir/test/pet_test.exs

@@ -24,17 +24,21 @@ defmodule PetTest do
     {:ok, %Tesla.Env{} = response} = PetApi.add_pet(connection, pet)
     assert response.status == 200
 
-    {:ok, pet} = PetApi.get_pet_by_id(connection, petId)
-    assert pet.id == petId
-    assert pet.name == "elixir client test"
-    assert pet.photoUrls == ["http://test_elixir_unit_test.com"]
-    assert pet.category == %Category{id: petId, name: "test elixir category"}
-    assert pet.tags == [%Tag{:id => petId, :name => "test elixir tag"}]
+    retry_assert(fn ->
+      {:ok, pet} = PetApi.get_pet_by_id(connection, petId)
+      assert pet.id == petId
+      assert pet.name == "elixir client test"
+      assert pet.photoUrls == ["http://test_elixir_unit_test.com"]
+      assert pet.category == %Category{id: petId, name: "test elixir category"}
+      assert pet.tags == [%Tag{:id => petId, :name => "test elixir tag"}]
+    end)
 
     {:ok, response} = PetApi.delete_pet(connection, petId)
     assert response.status == 200
-    {:ok, response} = PetApi.get_pet_by_id(connection, petId)
-    assert response.status == 404
+    retry_assert(fn ->
+      {:ok, response} = PetApi.get_pet_by_id(connection, petId)
+      assert response.status == 404
+    end)
   end
 
   test "update a pet", %{connection: connection} do
@@ -50,10 +54,24 @@ defmodule PetTest do
     {:ok, response} = PetApi.update_pet(connection, pet)
     assert response.status == 200
 
-    {:ok, pet} = PetApi.get_pet_by_id(connection, petId)
-    assert pet.id == petId
-    assert pet.name == "elixir client updatePet"
-    assert pet.status == "pending"
+    retry_assert(fn ->
+      {:ok, pet} = PetApi.get_pet_by_id(connection, petId)
+      assert pet.id == petId
+      assert pet.name == "elixir client updatePet"
+      assert pet.status == "pending"
+    end, 5, 100)
+  end
+
+  def retry_assert(fun, attempts \\ 3, delay \\ 100)
+  def retry_assert(_fun, 0, _delay), do: flunk("assertion failed after retries")
+  def retry_assert(fun, attempts, delay) do
+    try do
+      fun.()
+    rescue
+      _e ->
+        Process.sleep(delay)
+        retry_assert(fun, attempts - 1, delay)
+    end
   end
 
   test "find pet by status", %{connection: connection} do