mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
19b8957f3f
* 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
106 lines
2.7 KiB
Go
106 lines
2.7 KiB
Go
package fill
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// TestSetup tests the setup function of the Fills struct
|
|
func TestSetup(t *testing.T) {
|
|
fill := &Fills{}
|
|
channel := make(chan any)
|
|
fill.Setup(true, channel)
|
|
|
|
if fill.dataHandler == nil {
|
|
t.Error("expected dataHandler to be set")
|
|
}
|
|
|
|
if !fill.fillsFeedEnabled {
|
|
t.Error("expected fillsFeedEnabled to be true")
|
|
}
|
|
}
|
|
|
|
// TestUpdateDisabledFeed tests the Update function when fillsFeedEnabled is false
|
|
func TestUpdateDisabledFeed(t *testing.T) {
|
|
channel := make(chan any, 1)
|
|
fill := Fills{dataHandler: channel, fillsFeedEnabled: false}
|
|
|
|
// Send a test data to the Update function
|
|
testData := Data{Timestamp: time.Now(), Price: 15.2, Amount: 3.2}
|
|
assert.ErrorIs(t, fill.Update(testData), ErrFeedDisabled)
|
|
|
|
select {
|
|
case <-channel:
|
|
t.Errorf("Expected no data on channel, got data")
|
|
default:
|
|
// nothing to do
|
|
}
|
|
}
|
|
|
|
// TestUpdate tests the Update function of the Fills struct.
|
|
func TestUpdate(t *testing.T) {
|
|
channel := make(chan any, 1)
|
|
fill := &Fills{dataHandler: channel, fillsFeedEnabled: true}
|
|
receivedData := Data{Timestamp: time.Now(), Price: 15.2, Amount: 3.2}
|
|
if err := fill.Update(receivedData); err != nil {
|
|
t.Errorf("Update returned error %v", err)
|
|
}
|
|
|
|
select {
|
|
case data := <-channel:
|
|
dataSlice, ok := data.([]Data)
|
|
if !ok {
|
|
t.Errorf("expected []Data, got %T", data)
|
|
}
|
|
|
|
if len(dataSlice) != 1 || dataSlice[0] != receivedData {
|
|
t.Errorf("expected data to be sent through channel")
|
|
}
|
|
default:
|
|
t.Errorf("No data sent to channel")
|
|
}
|
|
}
|
|
|
|
// TestUpdateNoData tests the Update function with no Data objects
|
|
func TestUpdateNoData(t *testing.T) {
|
|
channel := make(chan any, 1)
|
|
fill := &Fills{dataHandler: channel, fillsFeedEnabled: true}
|
|
if err := fill.Update(); err != nil {
|
|
t.Errorf("Update returned error %v", err)
|
|
}
|
|
|
|
select {
|
|
case <-channel:
|
|
t.Errorf("Expected no data on channel, got data")
|
|
default:
|
|
// pass, nothing to do
|
|
}
|
|
}
|
|
|
|
// TestUpdateMultipleData tests the Update function with multiple Data objects
|
|
func TestUpdateMultipleData(t *testing.T) {
|
|
channel := make(chan any, 2)
|
|
fill := &Fills{dataHandler: channel, fillsFeedEnabled: true}
|
|
receivedData := Data{Timestamp: time.Now(), Price: 15.2, Amount: 3.2}
|
|
receivedData2 := Data{Timestamp: time.Now(), Price: 18.2, Amount: 9.0}
|
|
if err := fill.Update(receivedData, receivedData2); err != nil {
|
|
t.Errorf("Update returned error %v", err)
|
|
}
|
|
|
|
select {
|
|
case data := <-channel:
|
|
dataSlice, ok := data.([]Data)
|
|
if !ok {
|
|
t.Errorf("expected []Data, got %T", data)
|
|
}
|
|
|
|
if len(dataSlice) != 2 || dataSlice[0] != receivedData || dataSlice[1] != receivedData2 {
|
|
t.Errorf("expected data to be sent through channel")
|
|
}
|
|
default:
|
|
t.Errorf("No data sent to channel")
|
|
}
|
|
}
|