Files
gocryptotrader/engine/connection_manager_test.go
Adrian Gallagher 19b8957f3f codebase: Replace !errors.Is(err, target) with testify (#1931)
* tests: Replace !errors.Is(err, target) with testify equivalents

* codebase: Manual !errors.Is(err, target) replacements

* typo: Replace errMisMatchedEvent with errMismatchedEvent

* tests: Enhance error messages for better output

* tests: Refactor error assertions in various test cases to use require and improve clarity

* misc linter: Fix assert should wording

* tests: Simplify assertions in TestCreateSignals for clarity and conciseness

* tests: Enhance assertion message in TestCreateSignals
2025-06-10 16:29:57 +10:00

108 lines
2.1 KiB
Go

package engine
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/thrasher-corp/gocryptotrader/config"
)
func TestSetupConnectionManager(t *testing.T) {
t.Parallel()
_, err := setupConnectionManager(nil)
assert.ErrorIs(t, err, errNilConfig)
m, err := setupConnectionManager(&config.ConnectionMonitorConfig{})
assert.NoError(t, err)
if m == nil {
t.Error("expected manager")
}
}
func TestConnectionMonitorIsRunning(t *testing.T) {
t.Parallel()
m, err := setupConnectionManager(&config.ConnectionMonitorConfig{})
assert.NoError(t, err)
err = m.Start()
assert.NoError(t, err)
if !m.IsRunning() {
t.Error("expected true")
}
m.started = 0
if m.IsRunning() {
t.Error("expected false")
}
m = nil
if m.IsRunning() {
t.Error("expected false")
}
}
func TestConnectionMonitorStart(t *testing.T) {
t.Parallel()
m, err := setupConnectionManager(&config.ConnectionMonitorConfig{})
assert.NoError(t, err)
err = m.Start()
assert.NoError(t, err)
err = m.Start()
assert.ErrorIs(t, err, ErrSubSystemAlreadyStarted)
m = nil
err = m.Start()
assert.ErrorIs(t, err, ErrNilSubsystem)
}
func TestConnectionMonitorStop(t *testing.T) {
t.Parallel()
err := (&connectionManager{started: 1}).Stop()
assert.ErrorIs(t, err, errConnectionCheckerIsNil)
m, err := setupConnectionManager(&config.ConnectionMonitorConfig{})
assert.NoError(t, err)
err = m.Start()
assert.NoError(t, err)
err = m.Stop()
assert.NoError(t, err)
err = m.Stop()
assert.ErrorIs(t, err, ErrSubSystemNotStarted)
m = nil
err = m.Stop()
assert.ErrorIs(t, err, ErrNilSubsystem)
}
func TestConnectionMonitorIsOnline(t *testing.T) {
t.Parallel()
m, err := setupConnectionManager(&config.ConnectionMonitorConfig{})
assert.NoError(t, err)
err = m.Start()
assert.NoError(t, err)
// If someone runs this offline, who are we to fail them?
m.IsOnline()
err = m.Stop()
if err != nil {
t.Fatal(err)
}
if m.IsOnline() {
t.Error("expected false")
}
m.conn = nil
if m.IsOnline() {
t.Error("expected false")
}
m = nil
if m.IsOnline() {
t.Error("expected false")
}
}