func TestTwoShards(t *testing.T) {
t.Parallel()
maxCount := 5
if testing.Short() {
maxCount = 1
}
// used to prevent collisions
counter := 0
f := func(w traffic.Workload) bool {
defer func() { counter++ }()
cluster := newCluster(t, fmt.Sprintf("TestTwoShards-%d", counter), 2, etcache.NewTestCache())
defer cluster.Close()
// Run the workload
storage.RunWorkload(t, cluster.router.URL, w)
// Make sure we see the changes we should
facts := w.Facts()
storage.RunWorkload(t, cluster.router.URL, facts)
//increment the counter
return true
}
if err := quick.Check(f, &quick.Config{MaxCount: maxCount}); err != nil {
t.Error(err)
}
}
func TestWordCount(t *testing.T) {
t.Parallel()
maxCount := 2
if testing.Short() {
maxCount = 1
}
// First setup the WordCount pipeline
pipeline := `
image ubuntu
input data
run mkdir -p /out/counts
run cat /in/data/* | tr -cs "A-Za-z'" "\n" | sort | uniq -c | sort -n -r | while read count; do echo ${count% *} >/out/counts/${count#* }; done
shuffle counts
run find /out/counts | while read count; do cat $count | awk '{ sum+=$1} END {print sum}' >/tmp/count; mv /tmp/count $count; done
`
// used to prevent collisions
counter := 0
f := func(w traffic.Workload) bool {
defer func() { counter++ }()
cluster := newCluster(t, fmt.Sprintf("TestWordCount-%d", counter), 4, etcache.NewTestCache())
defer cluster.Close()
// Run the workload
storage.RunWorkload(t, cluster.router.URL, w)
// Install the pipeline
response, err := http.Post(cluster.router.URL+"/pipeline/wc", "application/text", strings.NewReader(pipeline))
defer response.Body.Close()
assert.NoError(t, err)
// Make a commit
response, err = http.Post(cluster.router.URL+"/commit?commit=commit1", "", nil)
defer response.Body.Close()
assert.NoError(t, err)
// TODO(jd) make this check for correctness, not just that the request
// completes. It's a bit hard because the input is random. Probably the
// right idea is to modify the traffic package so that it keeps track of
// this.
response, err = http.Get(cluster.router.URL + "/pipeline/wc/file/counts/*?commit=commit1")
defer response.Body.Close()
assert.NoError(t, err)
require.Equal(t, http.StatusOK, response.StatusCode)
return true
}
if err := quick.Check(f, &quick.Config{MaxCount: maxCount}); err != nil {
t.Error(err)
}
}
