Update UDP outbound tests

ansible · Nov 28, 2024 · fdc7e9e · fdc7e9e
1 parent ad4fa8f
commit fdc7e9e
Showing 1 changed file with 1 addition and 16 deletions.
diff --git a/pkg/services/udp_proxy_test.go b/pkg/services/udp_proxy_test.go
@@ -101,7 +101,6 @@ func TestUDPProxyServiceOutbound(t *testing.T) {
 	var mockNetceptor *mock_services.MockNetcForUDPProxy
 	var mockNetter *mock_net_interface.MockNetterUDP
 	var mockPacketCon *mock_netceptor.MockPacketConner
-	logger := logger.NewReceptorLogger("test")
 	type testCase struct {
 		name      string
 		service   string
@@ -131,21 +130,7 @@ func TestUDPProxyServiceOutbound(t *testing.T) {
 			calls: func() {
 				mockNetter.EXPECT().ResolveUDPAddr(gomock.Any(), gomock.Any()).Return(nil, nil)
 				mockNetceptor.EXPECT().ListenPacketAndAdvertise(gomock.Any(), gomock.Any()).Return(mockPacketCon, nil)
-				mockPacketCon.EXPECT().ReadFrom(gomock.Any()).Return(0, nil, errors.New("Read From error"))
-			},
-		},
-		{
-			name: "Fail Dial UDP",
-			calls: func() {
-				mockNetter.EXPECT().ResolveUDPAddr(gomock.Any(), gomock.Any()).Return(nil, nil)
-				mockNetceptor.EXPECT().ListenPacketAndAdvertise(gomock.Any(), gomock.Any()).Return(mockPacketCon, nil)
-				gomock.InOrder(
-					mockPacketCon.EXPECT().ReadFrom(gomock.Any()).Return(0, netceptor.Addr{}, nil),
-					mockPacketCon.EXPECT().ReadFrom(gomock.Any()).Return(0, nil, errors.New("Read From error")).AnyTimes(),
-				)
-				mockNetceptor.EXPECT().GetLogger().Return(logger).AnyTimes()
-				mockNetter.EXPECT().DialUDP(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New("Dial UDP error"))
-
+				mockPacketCon.EXPECT().ReadFrom(gomock.Any()).Return(0, nil, errors.New("Read From error")).AnyTimes()
 			},
 		},
 	}