ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-09-02 00:11:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

Minor formatting issues

Browse Source
This commit is contained in:
9seconds
2021-03-11 21:50:47 +03:00
parent 5c4536c591
commit c0ab254acf
13 changed files with 16 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mtglib/network/circuit_breaker.go
-2
View File
@@ -125,7 +125,6 @@ func (c *circuitBreakerDialer) switchState(state uint32) {
}
c.failuresCount = 0
atomic.StoreUint32(&c.halfOpenAttempts, 0)
atomic.StoreUint32(&c.state, state)
}
@@ -158,7 +157,6 @@ func (c *circuitBreakerDialer) tryHalfOpen() {
func (c *circuitBreakerDialer) stopTimer(timerRef **time.Timer) {
timer := *timerRef
if timer == nil {
return
}
mtglib/network/circuit_breaker_internal_test.go
-2
View File
@@ -51,7 +51,6 @@ func (suite *CircuitBreakerTestSuite) TestMultipleRunsOk() {
Return(suite.connMock, nil)
wg := &sync.WaitGroup{}
wg.Add(5)
go func() {
@@ -126,7 +125,6 @@ func (suite *CircuitBreakerTestSuite) TestHalfOpen() {
time.Sleep(500 * time.Millisecond)
conn, err := suite.d.DialContext(suite.ctx, "tcp", "127.0.0.2")
suite.NoError(err)
suite.Equal("10.0.0.10:80", conn.RemoteAddr().String())
mtglib/network/default_test.go
-6
View File
@@ -20,7 +20,6 @@ func (suite *DefaultDialerTestSuite) SetupSuite() {
suite.HTTPServerTestSuite.SetupSuite()
d, err := network.NewDefaultDialer(0, 0)
suite.NoError(err)
suite.d = d
@@ -28,13 +27,11 @@ func (suite *DefaultDialerTestSuite) SetupSuite() {
func (suite *DefaultDialerTestSuite) TestNegativeTimeout() {
_, err := network.NewDefaultDialer(-1, 0)
suite.Error(err)
}
func (suite *DefaultDialerTestSuite) TestNegativeBufferSize() {
_, err := network.NewDefaultDialer(0, -1)
suite.Error(err)
}
@@ -42,7 +39,6 @@ func (suite *DefaultDialerTestSuite) TestUnsupportedProtocol() {
_, err := suite.d.DialContext(context.Background(),
"udp",
suite.HTTPServerAddress())
suite.Error(err)
}
@@ -50,7 +46,6 @@ func (suite *DefaultDialerTestSuite) TestCannotDial() {
_, err := suite.d.DialContext(context.Background(),
"tcp",
suite.HTTPServerAddress()+suite.HTTPServerAddress())
suite.Error(err)
}
@@ -58,7 +53,6 @@ func (suite *DefaultDialerTestSuite) TestConnectOk() {
conn, err := suite.d.DialContext(context.Background(),
"tcp",
suite.HTTPServerAddress())
suite.NoError(err)
suite.NotNil(conn)
mtglib/network/load_balanced_socks5_test.go
-4
View File
@@ -28,12 +28,10 @@ func (suite *LoadBalancedSocks5TestSuite) SetupSuite() {
func (suite *LoadBalancedSocks5TestSuite) SetupTest() {
baseDialer, _ := network.NewDefaultDialer(0, 0)
lbDialer, err := network.NewLoadBalancedSocks5Dialer(baseDialer, []*url.URL{
suite.MakeSocks5URL("user", "password"),
suite.MakeSocks5URL("user2", "password"),
})
suite.NoError(err)
suite.httpClient = suite.MakeHTTPClient(lbDialer)
@@ -48,13 +46,11 @@ func (suite *LoadBalancedSocks5TestSuite) TestIncorrectURL() {
_, err := network.NewLoadBalancedSocks5Dialer(&DialerMock{}, []*url.URL{
{Scheme: "http"},
})
suite.Error(err)
}
func (suite *LoadBalancedSocks5TestSuite) TestCannotDial() {
baseDialer := &DialerMock{}
baseDialer.On("DialContext", mock.Anything, "tcp", "127.0.0.1:1080").
Times(network.ProxyDialerOpenThreshold).
Return(&net.TCPConn{}, io.EOF)
mtglib/network/network.go
-1
View File
@@ -49,7 +49,6 @@ func (n *network) DialContext(ctx context.Context, protocol, address string) (ne
}
var conn net.Conn
for _, v := range ips {
conn, err = n.dialer.DialContext(ctx, protocol, net.JoinHostPort(v, port))
mtglib/network/proxy_dialer_internal_test.go
+4 -10
View File
@@ -22,7 +22,6 @@ func (suite *ProxyDialerTestSuite) SetupSuite() {
func (suite *ProxyDialerTestSuite) TestSetupDefaults() {
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
suite.EqualValues(ProxyDialerOpenThreshold, d.openThreshold)
suite.EqualValues(ProxyDialerHalfOpenTimeout, d.halfOpenTimeout)
suite.EqualValues(ProxyDialerResetFailuresTimeout, d.resetFailuresTimeout)
@@ -30,14 +29,12 @@ func (suite *ProxyDialerTestSuite) TestSetupDefaults() {
func (suite *ProxyDialerTestSuite) TestSetupValuesAllOk() {
query := url.Values{}
query.Set("open_threshold", "30")
query.Set("reset_failures_timeout", "1s")
query.Set("half_open_timeout", "2s")
suite.u.RawQuery = query.Encode()
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
suite.EqualValues(30, d.openThreshold)
suite.EqualValues(2*time.Second, d.halfOpenTimeout)
suite.EqualValues(time.Second, d.resetFailuresTimeout)
@@ -51,10 +48,9 @@ func (suite *ProxyDialerTestSuite) TestOpenThreshold() {
param := v
suite.T().Run(v, func(t *testing.T) {
query.Set("open_threshold", param)
suite.u.RawQuery = query.Encode()
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
assert.EqualValues(t, ProxyDialerOpenThreshold, d.openThreshold)
})
}
@@ -68,10 +64,9 @@ func (suite *ProxyDialerTestSuite) TestHalfOpenTimeout() {
param := v
suite.T().Run(v, func(t *testing.T) {
query.Set("half_open_timeout", param)
suite.u.RawQuery = query.Encode()
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
assert.EqualValues(t, ProxyDialerHalfOpenTimeout, d.halfOpenTimeout)
})
}
@@ -85,10 +80,9 @@ func (suite *ProxyDialerTestSuite) TestResetFailuresTimeout() {
param := v
suite.T().Run(v, func(t *testing.T) {
query.Set("reset_failures_timeout", param)
suite.u.RawQuery = query.Encode()
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
d := newProxyDialer(&DialerMock{}, suite.u).(*circuitBreakerDialer)
assert.EqualValues(t, ProxyDialerHalfOpenTimeout, d.halfOpenTimeout)
})
}
mtglib/secret.go
-1
View File
@@ -31,7 +31,6 @@ func (s Secret) MarshalText() ([]byte, error) {
func (s *Secret) UnmarshalText(data []byte) error {
text := string(data)
if text == "" {
return ErrSecretEmpty
}
mtglib/secret_test.go
-5
View File
@@ -32,13 +32,11 @@ func (suite *SecretTestSuite) TestParseSecret() {
suite.T().Run(name, func(t *testing.T) {
parsed, err := mtglib.ParseSecret(param)
assert.NoError(t, err)
assert.Equal(t, s.Key, parsed.Key)
assert.Equal(t, s.Host, parsed.Host)
newSecret := mtglib.Secret{}
assert.NoError(t, newSecret.UnmarshalText([]byte(param)))
assert.Equal(t, s.Key, newSecret.Key)
assert.Equal(t, s.Host, newSecret.Host)
@@ -67,7 +65,6 @@ func (suite *SecretTestSuite) TestMarshalData() {
copy(s.Key[:], secretData)
data, err := json.Marshal(&s)
suite.NoError(err)
suite.Equal(string(data), `"7tEcbLvZ7-f-1bwNsiCwlmVnb29nbGUuY29t"`)
}
@@ -87,7 +84,6 @@ func (suite *SecretTestSuite) TestIncorrectSecret() {
suite.T().Run(param, func(t *testing.T) {
_, err := mtglib.ParseSecret(param)
assert.Error(t, err)
})
}
@@ -97,7 +93,6 @@ func (suite *SecretTestSuite) TestInvariant() {
generated := mtglib.GenerateSecret("google.com")
parsed, err := mtglib.ParseSecret(generated.Hex())
suite.NoError(err)
suite.Equal(generated.Key, parsed.Key)
suite.Equal(generated.Host, parsed.Host)