add sync-delay and single configurations to swarm smoke tests

cmd/swarm/swarm-smoke/main.go

@@ -45,10 +45,12 @@ var (
 	appName          string
 	scheme           string
 	filesize         int
+	syncDelay        int
 	from             int
 	to               int
 	verbosity        int
 	timeout          int
+	single           bool
 )
 
 func main() {
@@ -99,6 +101,12 @@ func main() {
 			Usage:       "file size for generated random file in KB",
 			Destination: &filesize,
 		},
+		cli.IntFlag{
+			Name:        "sync-delay",
+			Value:       5,
+			Usage:       "duration of delay in seconds to wait for content to be synced",
+			Destination: &syncDelay,
+		},
 		cli.IntFlag{
 			Name:        "verbosity",
 			Value:       1,
@@ -111,6 +119,11 @@ func main() {
 			Usage:       "timeout in seconds after which kill the process",
 			Destination: &timeout,
 		},
+		cli.BoolFlag{
+			Name:        "single",
+			Usage:       "whether to fetch content from a single node or from all nodes",
+			Destination: &single,
+		},
 	}
 
 	app.Flags = append(app.Flags, []cli.Flag{

cmd/swarm/swarm-smoke/upload_and_sync.go

@@ -25,6 +25,7 @@ import (
 	"fmt"
 	"io"
 	"io/ioutil"
+	"math/rand"
 	"net/http"
 	"net/http/httptrace"
 	"os"
@@ -112,9 +113,25 @@ func uploadAndSync(c *cli.Context) error {
 
 	log.Info("uploaded successfully", "hash", hash, "digest", fmt.Sprintf("%x", fhash))
 
-	time.Sleep(3 * time.Second)
+	time.Sleep(time.Duration(syncDelay) * time.Second)
 
 	wg := sync.WaitGroup{}
+	if single {
+		randIndex := 1 + rand.Intn(len(endpoints)-1)
+		ruid := uuid.New()[:8]
+		wg.Add(1)
+		go func(endpoint string, ruid string) {
+			for {
+				err := fetch(hash, endpoint, fhash, ruid)
+				if err != nil {
+					continue
+				}
+
+				wg.Done()
+				return
+			}
+		}(endpoints[randIndex], ruid)
+	} else {
 		for _, endpoint := range endpoints {
 			ruid := uuid.New()[:8]
 			wg.Add(1)
@@ -130,6 +147,7 @@ func uploadAndSync(c *cli.Context) error {
 				}
 			}(endpoint, ruid)
 		}
+	}
 	wg.Wait()
 	log.Info("all endpoints synced random file successfully")