implement GET /validate-cover

1 files changed, 13 insertions, 30 deletions

diff --git a/interior_test.go b/interior_test.go
index ba3d948..59707b8 100644
--- a/interior_test.go
+++ b/interior_test.go
@@ -1,9 +1,7 @@
 package lulu
 
 import (
-	"context"
 	"testing"
-	"time"
 
 	"github.com/stretchr/testify/require"
 )
@@ -87,36 +85,21 @@ func TestValidateInteriorBasic(t *testing.T) {
 
 func TestGetInteriorValidation(t *testing.T) {
 	c := newClient(t)
-
-	// Start validation job
 	id, err := c.ValidateInteriorBasic(interiorUrl)
 	require.NoError(t, err)
+	poll(t, func() bool {
+		rec, err := c.GetInteriorValidation(id)
+		require.NoError(t, err)
+		if rec.Status.IsFinal() {
+			require.Equal(t, InteriorStatusValidated, rec.Status)
 
-	// Poll until done
-	timeout := 15 * time.Second
-	period := time.Second
-	ctx, cancel := context.WithTimeout(context.Background(), timeout)
-	defer cancel()
-	timer := time.NewTimer(period)
-	for {
-		select {
-		case <-timer.C:
-			rec, err := c.GetInteriorValidation(id)
-			require.NoError(t, err)
-			if rec.Status.IsFinal() {
-				require.Equal(t, InteriorStatusValidated, rec.Status)
-
-				require.Equal(t, id, rec.Id)
-				require.Equal(t, interiorUrl, rec.SrcUrl)
-				require.Equal(t, uint(210), rec.NPages)
-				require.Empty(t, rec.Errors)
-				require.NotEmpty(t, rec.ValidPkgIds)
-				return
-			}
-			timer.Reset(period)
-		case <-ctx.Done():
-			t.Errorf("status still not finalized after %v", timeout)
-			return
+			require.Equal(t, id, rec.Id)
+			require.Equal(t, interiorUrl, rec.SrcUrl)
+			require.Equal(t, uint(210), rec.NPages)
+			require.Empty(t, rec.Errors)
+			require.NotEmpty(t, rec.ValidPkgIds)
+			return true
 		}
-	}
+		return false
+	})
 }