diff options
Diffstat (limited to 'client/go/internal/cli/cmd/feed_test.go')
-rw-r--r-- | client/go/internal/cli/cmd/feed_test.go | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/client/go/internal/cli/cmd/feed_test.go b/client/go/internal/cli/cmd/feed_test.go index 1bf1ef6ab9b..521d2b2abd0 100644 --- a/client/go/internal/cli/cmd/feed_test.go +++ b/client/go/internal/cli/cmd/feed_test.go @@ -1,6 +1,7 @@ package cmd import ( + "bytes" "os" "path/filepath" "testing" @@ -42,7 +43,7 @@ func TestFeed(t *testing.T) { require.Nil(t, cli.Run("feed", jsonFile)) assert.Equal(t, "", stderr.String()) - assert.Equal(t, `{ + want := `{ "feeder.seconds": 1.000, "feeder.ok.count": 1, "feeder.ok.rate": 1.000, @@ -63,5 +64,15 @@ func TestFeed(t *testing.T) { "200": 1 } } -`, stdout.String()) +` + assert.Equal(t, want, stdout.String()) + + stdout.Reset() + cli.Stdin = bytes.NewBuffer([]byte(`{ + "put": "id:ns:type::doc1", + "fields": {"foo": "123"} +}`)) + httpClient.NextResponseString(200, `{"message":"OK"}`) + require.Nil(t, cli.Run("feed", "-")) + assert.Equal(t, want, stdout.String()) } |