@@ -83,7 +83,7 @@ func TestAdminTrace(t *testing.T) {
8383 writesCount ++
8484 return nil
8585 }
86- if err := startTraceInfo (ctx , mockWSConn , adminClient ); err != nil {
86+ if err := startTraceInfo (ctx , mockWSConn , adminClient , serviceTraceOpts { AllTraffic : true , ErrOnly : false } ); err != nil {
8787 t .Errorf ("Failed on %s:, error occurred: %s" , function , err .Error ())
8888 }
8989 // check that the TestReceiver got the same number of data from trace.
@@ -95,7 +95,7 @@ func TestAdminTrace(t *testing.T) {
9595 connWriteMessageMock = func (messageType int , data []byte ) error {
9696 return fmt .Errorf ("error on write" )
9797 }
98- if err := startTraceInfo (ctx , mockWSConn , adminClient ); assert .Error (err ) {
98+ if err := startTraceInfo (ctx , mockWSConn , adminClient , serviceTraceOpts {} ); assert .Error (err ) {
9999 assert .Equal ("error on write" , err .Error ())
100100 }
101101
@@ -121,7 +121,7 @@ func TestAdminTrace(t *testing.T) {
121121 connWriteMessageMock = func (messageType int , data []byte ) error {
122122 return nil
123123 }
124- if err := startTraceInfo (ctx , mockWSConn , adminClient ); assert .Error (err ) {
124+ if err := startTraceInfo (ctx , mockWSConn , adminClient , serviceTraceOpts {} ); assert .Error (err ) {
125125 assert .Equal ("error on trace" , err .Error ())
126126 }
127127}
0 commit comments