@@ -6,6 +6,7 @@ package azservicebus
66import (
77 "context"
88 "io"
9+
910 "net"
1011 "net/http"
1112 "strings"
@@ -159,7 +160,9 @@ func TestNewClientNewSenderNotFound(t *testing.T) {
159160 defer cancel ()
160161
161162 err = sender .SendMessage (ctx , & Message {Body : []byte ("hello" )}, nil )
162- assertRPCNotFound (t , err )
163+ var sbErr * Error
164+ require .ErrorAs (t , err , & sbErr )
165+ require .Equal (t , CodeNotFound , sbErr .Code )
163166}
164167
165168func TestNewClientNewReceiverNotFound (t * testing.T ) {
@@ -175,7 +178,9 @@ func TestNewClientNewReceiverNotFound(t *testing.T) {
175178
176179 messages , err := receiver .ReceiveMessages (ctx , 1 , nil )
177180 require .Nil (t , messages )
178- assertRPCNotFound (t , err )
181+ var sbErr * Error
182+ require .ErrorAs (t , err , & sbErr )
183+ require .Equal (t , CodeNotFound , sbErr .Code )
179184
180185 receiver , err = client .NewReceiverForSubscription ("non-existent-topic" , "non-existent-subscription" , nil )
181186 require .NoError (t , err )
@@ -185,7 +190,9 @@ func TestNewClientNewReceiverNotFound(t *testing.T) {
185190
186191 messages , err = receiver .PeekMessages (ctx , 1 , nil )
187192 require .Nil (t , messages )
188- assertRPCNotFound (t , err )
193+ sbErr = nil
194+ require .ErrorAs (t , err , & sbErr )
195+ require .Equal (t , CodeNotFound , sbErr .Code )
189196}
190197
191198func TestClientNewSessionReceiverNotFound (t * testing.T ) {
@@ -198,14 +205,18 @@ func TestClientNewSessionReceiverNotFound(t *testing.T) {
198205
199206 receiver , err := client .AcceptSessionForQueue (ctx , "non-existent-queue" , "session-id" , nil )
200207 require .Nil (t , receiver )
201- assertRPCNotFound (t , err )
208+ var sbErr * Error
209+ require .ErrorAs (t , err , & sbErr )
210+ require .Equal (t , CodeNotFound , sbErr .Code )
202211
203212 ctx , cancel = context .WithTimeout (context .Background (), fastNotFoundDuration )
204213 defer cancel ()
205214
206215 receiver , err = client .AcceptNextSessionForQueue (ctx , "non-existent-queue" , nil )
207216 require .Nil (t , receiver )
208- assertRPCNotFound (t , err )
217+ sbErr = nil
218+ require .ErrorAs (t , err , & sbErr )
219+ require .Equal (t , CodeNotFound , sbErr .Code )
209220}
210221
211222func TestClientCloseVsClosePermanently (t * testing.T ) {
@@ -555,18 +566,6 @@ func TestNewClientUnitTests(t *testing.T) {
555566 })
556567}
557568
558- func assertRPCNotFound (t * testing.T , err error ) {
559- require .NotNil (t , err )
560-
561- var rpcError interface {
562- RPCCode () int
563- error
564- }
565-
566- require .ErrorAs (t , err , & rpcError )
567- require .Equal (t , http .StatusNotFound , rpcError .RPCCode ())
568- }
569-
570569func forceManagementSettlement (messages []* ReceivedMessage ) {
571570 for _ , m := range messages {
572571 m .settleOnMgmtLink = true
0 commit comments