refactor: rename i/e/n/mockablex => i/netxmocks (#397)
Needed to more easily do https://github.com/ooni/probe/issues/1505
This commit is contained in:
parent
8a0beee808
commit
16aa8e5538
|
@ -4,12 +4,12 @@ import (
|
||||||
"errors"
|
"errors"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestConnWorksOnSuccess(t *testing.T) {
|
func TestConnWorksOnSuccess(t *testing.T) {
|
||||||
counter := New()
|
counter := New()
|
||||||
underlying := &mockablex.Conn{
|
underlying := &netxmocks.Conn{
|
||||||
MockRead: func(b []byte) (int, error) {
|
MockRead: func(b []byte) (int, error) {
|
||||||
return 10, nil
|
return 10, nil
|
||||||
},
|
},
|
||||||
|
@ -39,7 +39,7 @@ func TestConnWorksOnFailure(t *testing.T) {
|
||||||
readError := errors.New("read error")
|
readError := errors.New("read error")
|
||||||
writeError := errors.New("write error")
|
writeError := errors.New("write error")
|
||||||
counter := New()
|
counter := New()
|
||||||
underlying := &mockablex.Conn{
|
underlying := &netxmocks.Conn{
|
||||||
MockRead: func(b []byte) (int, error) {
|
MockRead: func(b []byte) (int, error) {
|
||||||
return 0, readError
|
return 0, readError
|
||||||
},
|
},
|
||||||
|
|
|
@ -10,7 +10,7 @@ import (
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/handlers"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/handlers"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestEmitterFailure(t *testing.T) {
|
func TestEmitterFailure(t *testing.T) {
|
||||||
|
@ -20,7 +20,7 @@ func TestEmitterFailure(t *testing.T) {
|
||||||
Beginning: time.Now(),
|
Beginning: time.Now(),
|
||||||
Handler: saver,
|
Handler: saver,
|
||||||
})
|
})
|
||||||
d := EmitterDialer{Dialer: mockablex.Dialer{
|
d := EmitterDialer{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
@ -69,9 +69,9 @@ func TestEmitterSuccess(t *testing.T) {
|
||||||
Beginning: time.Now(),
|
Beginning: time.Now(),
|
||||||
Handler: saver,
|
Handler: saver,
|
||||||
})
|
})
|
||||||
d := EmitterDialer{Dialer: mockablex.Dialer{
|
d := EmitterDialer{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return &mockablex.Conn{
|
return &netxmocks.Conn{
|
||||||
MockRead: func(b []byte) (int, error) {
|
MockRead: func(b []byte) (int, error) {
|
||||||
return 0, io.EOF
|
return 0, io.EOF
|
||||||
},
|
},
|
||||||
|
|
|
@ -9,8 +9,8 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/bytecounter"
|
"github.com/ooni/probe-cli/v3/internal/bytecounter"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/iox"
|
"github.com/ooni/probe-cli/v3/internal/iox"
|
||||||
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func dorequest(ctx context.Context, url string) error {
|
func dorequest(ctx context.Context, url string) error {
|
||||||
|
@ -76,7 +76,7 @@ func TestByteCounterNoHandlers(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestByteCounterConnectFailure(t *testing.T) {
|
func TestByteCounterConnectFailure(t *testing.T) {
|
||||||
dialer := &byteCounterDialer{Dialer: mockablex.Dialer{
|
dialer := &byteCounterDialer{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
|
|
@ -8,12 +8,12 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestErrorWrapperFailure(t *testing.T) {
|
func TestErrorWrapperFailure(t *testing.T) {
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
d := &errorWrapperDialer{Dialer: mockablex.Dialer{
|
d := &errorWrapperDialer{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
@ -43,9 +43,9 @@ func errorWrapperCheckErr(t *testing.T, err error, op string) {
|
||||||
|
|
||||||
func TestErrorWrapperSuccess(t *testing.T) {
|
func TestErrorWrapperSuccess(t *testing.T) {
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
d := &errorWrapperDialer{Dialer: mockablex.Dialer{
|
d := &errorWrapperDialer{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return &mockablex.Conn{
|
return &netxmocks.Conn{
|
||||||
MockRead: func(b []byte) (int, error) {
|
MockRead: func(b []byte) (int, error) {
|
||||||
return 0, io.EOF
|
return 0, io.EOF
|
||||||
},
|
},
|
||||||
|
|
|
@ -8,13 +8,13 @@ import (
|
||||||
"net/url"
|
"net/url"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestProxyDialerDialContextNoProxyURL(t *testing.T) {
|
func TestProxyDialerDialContextNoProxyURL(t *testing.T) {
|
||||||
expected := errors.New("mocked error")
|
expected := errors.New("mocked error")
|
||||||
d := &proxyDialer{
|
d := &proxyDialer{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, expected
|
return nil, expected
|
||||||
},
|
},
|
||||||
|
@ -45,7 +45,7 @@ func TestProxyDialerDialContextInvalidScheme(t *testing.T) {
|
||||||
func TestProxyDialerDialContextWithEOF(t *testing.T) {
|
func TestProxyDialerDialContextWithEOF(t *testing.T) {
|
||||||
const expect = "10.0.0.1:9050"
|
const expect = "10.0.0.1:9050"
|
||||||
d := &proxyDialer{
|
d := &proxyDialer{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
if address != expect {
|
if address != expect {
|
||||||
return nil, errors.New("unexpected address")
|
return nil, errors.New("unexpected address")
|
||||||
|
|
|
@ -9,15 +9,15 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/trace"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/trace"
|
||||||
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestSaverDialerFailure(t *testing.T) {
|
func TestSaverDialerFailure(t *testing.T) {
|
||||||
expected := errors.New("mocked error")
|
expected := errors.New("mocked error")
|
||||||
saver := &trace.Saver{}
|
saver := &trace.Saver{}
|
||||||
dlr := &saverDialer{
|
dlr := &saverDialer{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, expected
|
return nil, expected
|
||||||
},
|
},
|
||||||
|
@ -59,7 +59,7 @@ func TestSaverConnDialerFailure(t *testing.T) {
|
||||||
expected := errors.New("mocked error")
|
expected := errors.New("mocked error")
|
||||||
saver := &trace.Saver{}
|
saver := &trace.Saver{}
|
||||||
dlr := &saverConnDialer{
|
dlr := &saverConnDialer{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, expected
|
return nil, expected
|
||||||
},
|
},
|
||||||
|
@ -79,9 +79,9 @@ func TestSaverConnDialerSuccess(t *testing.T) {
|
||||||
saver := &trace.Saver{}
|
saver := &trace.Saver{}
|
||||||
dlr := &saverConnDialer{
|
dlr := &saverConnDialer{
|
||||||
Dialer: &saverDialer{
|
Dialer: &saverDialer{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return &mockablex.Conn{
|
return &netxmocks.Conn{
|
||||||
MockRead: func(b []byte) (int, error) {
|
MockRead: func(b []byte) (int, error) {
|
||||||
return 0, io.EOF
|
return 0, io.EOF
|
||||||
},
|
},
|
||||||
|
|
|
@ -1,2 +0,0 @@
|
||||||
// Package mockable contains mocks for netx types.
|
|
||||||
package mockablex
|
|
|
@ -10,7 +10,7 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/apex/log"
|
"github.com/apex/log"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestDialerResolverNoPort(t *testing.T) {
|
func TestDialerResolverNoPort(t *testing.T) {
|
||||||
|
@ -69,7 +69,7 @@ func (r MockableResolver) Address() string {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDialerResolverDialForSingleIPFails(t *testing.T) {
|
func TestDialerResolverDialForSingleIPFails(t *testing.T) {
|
||||||
dialer := &DialerResolver{Dialer: mockablex.Dialer{
|
dialer := &DialerResolver{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
@ -85,7 +85,7 @@ func TestDialerResolverDialForSingleIPFails(t *testing.T) {
|
||||||
|
|
||||||
func TestDialerResolverDialForManyIPFails(t *testing.T) {
|
func TestDialerResolverDialForManyIPFails(t *testing.T) {
|
||||||
dialer := &DialerResolver{
|
dialer := &DialerResolver{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
@ -102,9 +102,9 @@ func TestDialerResolverDialForManyIPFails(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDialerResolverDialForManyIPSuccess(t *testing.T) {
|
func TestDialerResolverDialForManyIPSuccess(t *testing.T) {
|
||||||
dialer := &DialerResolver{Dialer: mockablex.Dialer{
|
dialer := &DialerResolver{Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return &mockablex.Conn{
|
return &netxmocks.Conn{
|
||||||
MockClose: func() error {
|
MockClose: func() error {
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
|
@ -125,7 +125,7 @@ func TestDialerResolverDialForManyIPSuccess(t *testing.T) {
|
||||||
|
|
||||||
func TestDialerLoggerFailure(t *testing.T) {
|
func TestDialerLoggerFailure(t *testing.T) {
|
||||||
d := &DialerLogger{
|
d := &DialerLogger{
|
||||||
Dialer: mockablex.Dialer{
|
Dialer: netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package mockablex
|
package netxmocks
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net"
|
"net"
|
|
@ -1,4 +1,4 @@
|
||||||
package mockablex
|
package netxmocks
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
|
@ -1,4 +1,4 @@
|
||||||
package mockablex
|
package netxmocks
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
|
@ -1,4 +1,4 @@
|
||||||
package mockablex
|
package netxmocks
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
2
internal/netxmocks/doc.go
Normal file
2
internal/netxmocks/doc.go
Normal file
|
@ -0,0 +1,2 @@
|
||||||
|
// Package netxmocks contains mocks for netx types.
|
||||||
|
package netxmocks
|
|
@ -8,7 +8,7 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/atomicx"
|
"github.com/ooni/probe-cli/v3/internal/atomicx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestOBFS4DialerWorks(t *testing.T) {
|
func TestOBFS4DialerWorks(t *testing.T) {
|
||||||
|
@ -48,7 +48,7 @@ func TestOBFS4DialerFailsWithInvalidCert(t *testing.T) {
|
||||||
func TestOBFS4DialerFailsWithConnectionErrorAndNoContextExpiration(t *testing.T) {
|
func TestOBFS4DialerFailsWithConnectionErrorAndNoContextExpiration(t *testing.T) {
|
||||||
expected := errors.New("mocked error")
|
expected := errors.New("mocked error")
|
||||||
o4d := DefaultTestingOBFS4Bridge()
|
o4d := DefaultTestingOBFS4Bridge()
|
||||||
o4d.UnderlyingDialer = &mockablex.Dialer{
|
o4d.UnderlyingDialer = &netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, expected
|
return nil, expected
|
||||||
},
|
},
|
||||||
|
@ -67,7 +67,7 @@ func TestOBFS4DialerFailsWithConnectionErrorAndContextExpiration(t *testing.T) {
|
||||||
defer cancel()
|
defer cancel()
|
||||||
expected := errors.New("mocked error")
|
expected := errors.New("mocked error")
|
||||||
o4d := DefaultTestingOBFS4Bridge()
|
o4d := DefaultTestingOBFS4Bridge()
|
||||||
o4d.UnderlyingDialer = &mockablex.Dialer{
|
o4d.UnderlyingDialer = &netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
// We cancel the context before returning the error, which makes
|
// We cancel the context before returning the error, which makes
|
||||||
// the context cancellation happen before us returning.
|
// the context cancellation happen before us returning.
|
||||||
|
@ -101,7 +101,7 @@ func TestOBFS4DialerWorksWithContextExpiration(t *testing.T) {
|
||||||
defer cancel()
|
defer cancel()
|
||||||
called := &atomicx.Int64{}
|
called := &atomicx.Int64{}
|
||||||
o4d := DefaultTestingOBFS4Bridge()
|
o4d := DefaultTestingOBFS4Bridge()
|
||||||
o4d.UnderlyingDialer = &mockablex.Dialer{
|
o4d.UnderlyingDialer = &netxmocks.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
// We cancel the context before returning the error, which makes
|
// We cancel the context before returning the error, which makes
|
||||||
// the context cancellation happen before us returning.
|
// the context cancellation happen before us returning.
|
||||||
|
|
|
@ -13,8 +13,8 @@ import (
|
||||||
|
|
||||||
"github.com/apex/log"
|
"github.com/apex/log"
|
||||||
"github.com/ooni/probe-cli/v3/internal/atomicx"
|
"github.com/ooni/probe-cli/v3/internal/atomicx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/iox"
|
"github.com/ooni/probe-cli/v3/internal/iox"
|
||||||
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestListenerLoggerWorks(t *testing.T) {
|
func TestListenerLoggerWorks(t *testing.T) {
|
||||||
|
@ -96,8 +96,8 @@ func TestListenerCastListenerWorksFineOnError(t *testing.T) {
|
||||||
|
|
||||||
// mockableSocksConn is a mockable ptxSocksConn.
|
// mockableSocksConn is a mockable ptxSocksConn.
|
||||||
type mockableSocksConn struct {
|
type mockableSocksConn struct {
|
||||||
// mockablex.Conn allows to mock all net.Conn functionality.
|
// netxmocks.Conn allows to mock all net.Conn functionality.
|
||||||
*mockablex.Conn
|
*netxmocks.Conn
|
||||||
|
|
||||||
// MockGrant allows to mock the Grant function.
|
// MockGrant allows to mock the Grant function.
|
||||||
MockGrant func(addr *net.TCPAddr) error
|
MockGrant func(addr *net.TCPAddr) error
|
||||||
|
@ -160,7 +160,7 @@ func TestListenerHandleSocksConnWithDialContextFailure(t *testing.T) {
|
||||||
}
|
}
|
||||||
lst := &Listener{PTDialer: d}
|
lst := &Listener{PTDialer: d}
|
||||||
c := &mockableSocksConn{
|
c := &mockableSocksConn{
|
||||||
Conn: &mockablex.Conn{
|
Conn: &netxmocks.Conn{
|
||||||
MockClose: func() error {
|
MockClose: func() error {
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
|
|
|
@ -7,7 +7,7 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/atomicx"
|
"github.com/ooni/probe-cli/v3/internal/atomicx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/netxmocks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestSnowflakeDialerWorks(t *testing.T) {
|
func TestSnowflakeDialerWorks(t *testing.T) {
|
||||||
|
@ -50,7 +50,7 @@ func TestSnowflakeDialerWorksWithMocks(t *testing.T) {
|
||||||
newClientTransport: func(brokerURL, frontDomain string, iceAddresses []string, keepLocalAddresses bool, maxSnowflakes int) (snowflakeTransport, error) {
|
newClientTransport: func(brokerURL, frontDomain string, iceAddresses []string, keepLocalAddresses bool, maxSnowflakes int) (snowflakeTransport, error) {
|
||||||
return &mockableSnowflakeTransport{
|
return &mockableSnowflakeTransport{
|
||||||
MockDial: func() (net.Conn, error) {
|
MockDial: func() (net.Conn, error) {
|
||||||
return &mockablex.Conn{
|
return &netxmocks.Conn{
|
||||||
MockClose: func() error {
|
MockClose: func() error {
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
|
@ -144,7 +144,7 @@ func TestSnowflakeDialerWorksWithWithCancelledContext(t *testing.T) {
|
||||||
return &mockableSnowflakeTransport{
|
return &mockableSnowflakeTransport{
|
||||||
MockDial: func() (net.Conn, error) {
|
MockDial: func() (net.Conn, error) {
|
||||||
cancel() // cause a cancel before we can really have a conn
|
cancel() // cause a cancel before we can really have a conn
|
||||||
return &mockablex.Conn{
|
return &netxmocks.Conn{
|
||||||
MockClose: func() error {
|
MockClose: func() error {
|
||||||
called.Add(1)
|
called.Add(1)
|
||||||
return nil
|
return nil
|
||||||
|
|
Loading…
Reference in New Issue
Block a user