wait for refresh to happen as there may be async jobs running on the pool

1 files changed, 17 insertions, 0 deletions

diff --git a/libvirt/utils.go b/libvirt/utils.go
index deec2900..3dc8e17e 100644
--- a/libvirt/utils.go
+++ b/libvirt/utils.go
@@ -2,6 +2,7 @@ package libvirt
 
 import (
 	"fmt"
+	"time"
 )
 
 var diskLetters []rune = []rune("abcdefghijklmnopqrstuvwxyz")
@@ -18,3 +19,19 @@ func DiskLetterForIndex(i int) string {
 
 	return fmt.Sprintf("%s%c", DiskLetterForIndex(q-1), letter)
 }
+
+// wait for success and timeout after 5 minutes.
+func WaitForSuccess(errorMessage string, f func() error) error {
+	start := time.Now()
+	for {
+		err := f()
+		if err == nil {
+			return nil
+		}
+
+		time.Sleep(1 * time.Second)
+		if time.Since(start) > 5*time.Minute {
+			return fmt.Errorf("%s: %s", errorMessage, err)
+		}
+	}
+}