Merge pull request ethereum#434 from ethersphere/smoke_test_refactor

cmd/swarm/swarm-smoke: extract cluster type and from/to ports and scheme

cmd/swarm/swarm-smoke/main.go

@@ -17,7 +17,6 @@
 package main
 
 import (
-	"fmt"
 	"os"
 	"sort"
 
@@ -28,17 +27,15 @@ import (
 )
 
 var (
-	endpoints []string
-	endpoint  string
-	filesize  int
+	endpoints        []string
+	includeLocalhost bool
+	cluster          string
+	scheme           string
+	filesize         int
+	from             int
+	to               int
 )
 
-func init() {
-	for port := 8501; port <= 8512; port++ {
-		endpoints = append(endpoints, fmt.Sprintf("http://%v.testing.swarm-gateways.net", port))
-	}
-}
-
 func main() {
 	log.PrintOrigins(true)
 	log.Root().SetHandler(log.LvlFilterHandler(log.LvlTrace, log.StreamHandler(colorable.NewColorableStderr(), log.TerminalFormat(true))))
@@ -49,10 +46,33 @@ func main() {
 
 	app.Flags = []cli.Flag{
 		cli.StringFlag{
-			Name:        "bzz-api",
-			Value:       endpoints[0],
-			Usage:       "upload node endpoint",
-			Destination: &endpoint,
+			Name:        "cluster-endpoint",
+			Value:       "testing",
+			Usage:       "cluster to point to (open, or testing)",
+			Destination: &cluster,
+		},
+		cli.IntFlag{
+			Name:        "cluster-from",
+			Value:       8501,
+			Usage:       "swarm node (from)",
+			Destination: &from,
+		},
+		cli.IntFlag{
+			Name:        "cluster-to",
+			Value:       8512,
+			Usage:       "swarm node (to)",
+			Destination: &to,
+		},
+		cli.StringFlag{
+			Name:        "cluster-scheme",
+			Value:       "http",
+			Usage:       "http or https",
+			Destination: &scheme,
+		},
+		cli.BoolFlag{
+			Name:        "include-localhost",
+			Usage:       "whether to include localhost:8500 as an endpoint",
+			Destination: &includeLocalhost,
 		},
 		cli.IntFlag{
 			Name:        "filesize",

cmd/swarm/swarm-smoke/upload_and_sync.go

@@ -20,10 +20,22 @@ import (
 	cli "gopkg.in/urfave/cli.v1"
 )
 
+func generateEndpoints(scheme string, cluster string, from int, to int) {
+	for port := from; port <= to; port++ {
+		endpoints = append(endpoints, fmt.Sprintf("%s://%v.%s.swarm-gateways.net", scheme, port, cluster))
+	}
+
+	if includeLocalhost {
+		endpoints = append(endpoints, "http://localhost:8500")
+	}
+}
+
 func cliUploadAndSync(c *cli.Context) error {
 	defer func(now time.Time) { log.Info("total time", "time", time.Since(now), "size", filesize) }(time.Now())
 
-	log.Info("uploading and syncing")
+	generateEndpoints(scheme, cluster, from, to)
+
+	log.Info("uploading to " + endpoints[0] + " and syncing")
 
 	f, cleanup := generateRandomFile(filesize * 1000000)
 	defer cleanup()