cmd/swarm: Update Mutable Resource (now Swarm feed) smoke test

This commit is contained in:
lash 2018-10-11 09:19:39 +02:00
parent f951e23fb5
commit de3199b908
2 changed files with 28 additions and 5 deletions

View file

@ -21,7 +21,6 @@ import (
"sort" "sort"
"github.com/ethereum/go-ethereum/log" "github.com/ethereum/go-ethereum/log"
colorable "github.com/mattn/go-colorable"
cli "gopkg.in/urfave/cli.v1" cli "gopkg.in/urfave/cli.v1"
) )
@ -34,11 +33,10 @@ var (
filesize int filesize int
from int from int
to int to int
verbosity int
) )
func main() { func main() {
log.PrintOrigins(true)
log.Root().SetHandler(log.LvlFilterHandler(log.LvlTrace, log.StreamHandler(colorable.NewColorableStderr(), log.TerminalFormat(true))))
app := cli.NewApp() app := cli.NewApp()
app.Name = "smoke-test" app.Name = "smoke-test"
@ -80,6 +78,12 @@ func main() {
Usage: "file size for generated random file in KB", Usage: "file size for generated random file in KB",
Destination: &filesize, Destination: &filesize,
}, },
cli.IntFlag{
Name: "verbosity",
Value: 1,
Usage: "verbosity",
Destination: &verbosity,
},
} }
app.Commands = []cli.Command{ app.Commands = []cli.Command{
@ -89,6 +93,12 @@ func main() {
Usage: "upload and sync", Usage: "upload and sync",
Action: cliUploadAndSync, Action: cliUploadAndSync,
}, },
{
Name: "feed_sync",
Aliases: []string{"f"},
Usage: "feed update generate, upload and sync",
Action: cliFeedUploadAndSync,
},
} }
sort.Sort(cli.FlagsByName(app.Flags)) sort.Sort(cli.FlagsByName(app.Flags))

View file

@ -19,7 +19,8 @@ package main
import ( import (
"bytes" "bytes"
"crypto/md5" "crypto/md5"
"crypto/rand" crand "crypto/rand"
"errors"
"fmt" "fmt"
"io" "io"
"io/ioutil" "io/ioutil"
@ -166,6 +167,18 @@ func digest(r io.Reader) ([]byte, error) {
return h.Sum(nil), nil return h.Sum(nil), nil
} }
// generates random data in heap buffer
func generateRandomData(datasize int) ([]byte, error) {
b := make([]byte, datasize)
c, err := crand.Read(b)
if err != nil {
return nil, err
} else if c != datasize {
return nil, errors.New("short read")
}
return b, nil
}
// generateRandomFile is creating a temporary file with the requested byte size // generateRandomFile is creating a temporary file with the requested byte size
func generateRandomFile(size int) (f *os.File, teardown func()) { func generateRandomFile(size int) (f *os.File, teardown func()) {
// create a tmp file // create a tmp file
@ -181,7 +194,7 @@ func generateRandomFile(size int) (f *os.File, teardown func()) {
} }
buf := make([]byte, size) buf := make([]byte, size)
_, err = rand.Read(buf) _, err = crand.Read(buf)
if err != nil { if err != nil {
panic(err) panic(err)
} }