From 7594a8b4914a2a4413bf0251e37da085f3f88ce2 Mon Sep 17 00:00:00 2001 From: Edward McFarlane Date: Thu, 22 Feb 2024 18:24:49 +0000 Subject: [PATCH] Fix test assert --- cmd/demoserver/main_test.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cmd/demoserver/main_test.go b/cmd/demoserver/main_test.go index ae775c2..c143b0f 100644 --- a/cmd/demoserver/main_test.go +++ b/cmd/demoserver/main_test.go @@ -73,11 +73,11 @@ func TestElizaServer(t *testing.T) { defer wg.Done() for _, sentence := range sendValues { err := stream.Send(&elizav1.ConverseRequest{Sentence: sentence}) - if assert.ErrorContainsf(t, err, `failed for string sentence: "%s"`, sentence) { + if !assert.NoError(t, err, `failed for string sentence: "%s"`, sentence) { break } } - assert.NoError(t, stream.CloseRequest()) + assert.NoError(t, stream.CloseRequest()) }() go func() { defer wg.Done() @@ -92,7 +92,7 @@ func TestElizaServer(t *testing.T) { assert.NotEmpty(t, msg.GetSentence()) receivedValues = append(receivedValues, msg.GetSentence()) } - assert.NoError(t, stream.CloseResponse()) + assert.NoError(t, stream.CloseResponse()) }() wg.Wait() assert.Equal(t, len(receivedValues), len(sendValues))