From f5d007a3e08c94731ca15bb6ca5e67058764ebc9 Mon Sep 17 00:00:00 2001 From: ionutboangiu Date: Wed, 22 Feb 2023 10:11:51 -0500 Subject: [PATCH] Revise caps unit test --- cores/caps_test.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/cores/caps_test.go b/cores/caps_test.go index 38d906707..ea3a3f89a 100644 --- a/cores/caps_test.go +++ b/cores/caps_test.go @@ -180,12 +180,16 @@ func TestNewCapsBiRPCCodec(t *testing.T) { t.Errorf("Expected: %v ,received:%v", expR, r) } - if err = codec.ReadRequestBody("args"); err == nil || err != utils.ErrNotImplemented { + expR.Method = utils.SessionSv1CapsError + + if err = codec.ReadHeader(r, nil); err != nil { t.Fatal(err) + } else if !reflect.DeepEqual(r, expR) { + t.Errorf("Expected: %v ,received:%v", expR, r) } - if err = codec.ReadRequestBody("args"); err != utils.ErrMaxConcurrentRPCExceededNoCaps { - t.Errorf("Expected error: %v ,received: %v ", utils.ErrMaxConcurrentRPCExceededNoCaps, err) + if err = codec.ReadRequestBody("args"); err == nil || err != utils.ErrNotImplemented { + t.Fatal(err) } if err = codec.WriteResponse(&rpc2.Response{