applications/any2anexoj
1
0
Fork 0
Code Pull Requests Actions Activity

handle context cancelation

Browse Source
This commit is contained in:
Natercio Moniz
2025-11-13 16:05:16 +00:00
parent 54fced39aa
commit f3d0f5d71a
7 changed files with 50 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/report_test.go
View File

@@ -1,6 +1,7 @@
package internal_test
import (
"context"
"io"
"math/big"
"testing"
@@ -20,7 +21,7 @@ func TestReporter_Run(t *testing.T) {
mockRecord(ctrl, 20.0, 10.0, internal.SideBuy, now),
mockRecord(ctrl, 25.0, 10.0, internal.SideSell, now.Add(1)),
}
reader.EXPECT().ReadRecord().DoAndReturn(func() (internal.Record, error) {
reader.EXPECT().ReadRecord(gomock.Any()).DoAndReturn(func(ctx context.Context) (internal.Record, error) {
if len(records) > 0 {
r := records[0]
records = records[1:]
@@ -31,7 +32,7 @@ func TestReporter_Run(t *testing.T) {
}).Times(3)
writer := mocks.NewMockReportWriter(ctrl)
writer.EXPECT().Write(gomock.Eq(internal.ReportItem{
writer.EXPECT().Write(gomock.Any(), gomock.Eq(internal.ReportItem{
BuyValue: new(big.Float).SetFloat64(200.0),
BuyTimestamp: now,
SellValue: new(big.Float).SetFloat64(250.0),
@@ -40,7 +41,7 @@ func TestReporter_Run(t *testing.T) {
Taxes: new(big.Float),
})).Times(1)
gotErr := internal.BuildReport(reader, writer)
gotErr := internal.BuildReport(t.Context(), reader, writer)
if gotErr != nil {
t.Fatalf("got unexpected err: %v", gotErr)
}