diff --git a/internal/generate.go b/internal/generate.go index 9e087fd..a0a9c26 100644 --- a/internal/generate.go +++ b/internal/generate.go @@ -1,3 +1,3 @@ package internal -//go:generate mockgen -destination=mocks/mocks_gen.go -package=mocks -typed . RecordReader,Record +//go:generate go tool mockgen -destination=mocks/mocks_gen.go -package=mocks -typed . RecordReader,Record,ReportWriter diff --git a/internal/mocks/mocks_gen.go b/internal/mocks/mocks_gen.go index 04325d7..64430ee 100644 --- a/internal/mocks/mocks_gen.go +++ b/internal/mocks/mocks_gen.go @@ -1,9 +1,9 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: git.naterciomoniz.net/applications/broker2anexoj/internal (interfaces: RecordReader,Record) +// Source: git.naterciomoniz.net/applications/broker2anexoj/internal (interfaces: RecordReader,Record,ReportWriter) // // Generated by this command: // -// mockgen -destination=mocks/mocks_gen.go -package=mocks -typed . RecordReader,Record +// mockgen -destination=mocks/mocks_gen.go -package=mocks -typed . RecordReader,Record,ReportWriter // // Package mocks is a generated GoMock package. @@ -105,6 +105,44 @@ func (m *MockRecord) EXPECT() *MockRecordMockRecorder { return m.recorder } +// Fees mocks base method. +func (m *MockRecord) Fees() *big.Float { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Fees") + ret0, _ := ret[0].(*big.Float) + return ret0 +} + +// Fees indicates an expected call of Fees. +func (mr *MockRecordMockRecorder) Fees() *MockRecordFeesCall { + mr.mock.ctrl.T.Helper() + call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Fees", reflect.TypeOf((*MockRecord)(nil).Fees)) + return &MockRecordFeesCall{Call: call} +} + +// MockRecordFeesCall wrap *gomock.Call +type MockRecordFeesCall struct { + *gomock.Call +} + +// Return rewrite *gomock.Call.Return +func (c *MockRecordFeesCall) Return(arg0 *big.Float) *MockRecordFeesCall { + c.Call = c.Call.Return(arg0) + return c +} + +// Do rewrite *gomock.Call.Do +func (c *MockRecordFeesCall) Do(f func() *big.Float) *MockRecordFeesCall { + c.Call = c.Call.Do(f) + return c +} + +// DoAndReturn rewrite *gomock.Call.DoAndReturn +func (c *MockRecordFeesCall) DoAndReturn(f func() *big.Float) *MockRecordFeesCall { + c.Call = c.Call.DoAndReturn(f) + return c +} + // Price mocks base method. func (m *MockRecord) Price() *big.Float { m.ctrl.T.Helper() @@ -257,6 +295,44 @@ func (c *MockRecordSymbolCall) DoAndReturn(f func() string) *MockRecordSymbolCal return c } +// Taxes mocks base method. +func (m *MockRecord) Taxes() *big.Float { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Taxes") + ret0, _ := ret[0].(*big.Float) + return ret0 +} + +// Taxes indicates an expected call of Taxes. +func (mr *MockRecordMockRecorder) Taxes() *MockRecordTaxesCall { + mr.mock.ctrl.T.Helper() + call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Taxes", reflect.TypeOf((*MockRecord)(nil).Taxes)) + return &MockRecordTaxesCall{Call: call} +} + +// MockRecordTaxesCall wrap *gomock.Call +type MockRecordTaxesCall struct { + *gomock.Call +} + +// Return rewrite *gomock.Call.Return +func (c *MockRecordTaxesCall) Return(arg0 *big.Float) *MockRecordTaxesCall { + c.Call = c.Call.Return(arg0) + return c +} + +// Do rewrite *gomock.Call.Do +func (c *MockRecordTaxesCall) Do(f func() *big.Float) *MockRecordTaxesCall { + c.Call = c.Call.Do(f) + return c +} + +// DoAndReturn rewrite *gomock.Call.DoAndReturn +func (c *MockRecordTaxesCall) DoAndReturn(f func() *big.Float) *MockRecordTaxesCall { + c.Call = c.Call.DoAndReturn(f) + return c +} + // Timestamp mocks base method. func (m *MockRecord) Timestamp() time.Time { m.ctrl.T.Helper() @@ -294,3 +370,65 @@ func (c *MockRecordTimestampCall) DoAndReturn(f func() time.Time) *MockRecordTim c.Call = c.Call.DoAndReturn(f) return c } + +// MockReportWriter is a mock of ReportWriter interface. +type MockReportWriter struct { + ctrl *gomock.Controller + recorder *MockReportWriterMockRecorder + isgomock struct{} +} + +// MockReportWriterMockRecorder is the mock recorder for MockReportWriter. +type MockReportWriterMockRecorder struct { + mock *MockReportWriter +} + +// NewMockReportWriter creates a new mock instance. +func NewMockReportWriter(ctrl *gomock.Controller) *MockReportWriter { + mock := &MockReportWriter{ctrl: ctrl} + mock.recorder = &MockReportWriterMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockReportWriter) EXPECT() *MockReportWriterMockRecorder { + return m.recorder +} + +// Write mocks base method. +func (m *MockReportWriter) Write(arg0 internal.ReportItem) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Write", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// Write indicates an expected call of Write. +func (mr *MockReportWriterMockRecorder) Write(arg0 any) *MockReportWriterWriteCall { + mr.mock.ctrl.T.Helper() + call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Write", reflect.TypeOf((*MockReportWriter)(nil).Write), arg0) + return &MockReportWriterWriteCall{Call: call} +} + +// MockReportWriterWriteCall wrap *gomock.Call +type MockReportWriterWriteCall struct { + *gomock.Call +} + +// Return rewrite *gomock.Call.Return +func (c *MockReportWriterWriteCall) Return(arg0 error) *MockReportWriterWriteCall { + c.Call = c.Call.Return(arg0) + return c +} + +// Do rewrite *gomock.Call.Do +func (c *MockReportWriterWriteCall) Do(f func(internal.ReportItem) error) *MockReportWriterWriteCall { + c.Call = c.Call.Do(f) + return c +} + +// DoAndReturn rewrite *gomock.Call.DoAndReturn +func (c *MockReportWriterWriteCall) DoAndReturn(f func(internal.ReportItem) error) *MockReportWriterWriteCall { + c.Call = c.Call.DoAndReturn(f) + return c +} diff --git a/internal/record.go b/internal/record.go index 68b2301..9fa5d68 100644 --- a/internal/record.go +++ b/internal/record.go @@ -12,6 +12,8 @@ type Record interface { Price() *big.Float Quantity() *big.Float Timestamp() time.Time + Fees() *big.Float + Taxes() *big.Float } type RecordQueue struct { diff --git a/internal/report.go b/internal/report.go new file mode 100644 index 0000000..6cc582d --- /dev/null +++ b/internal/report.go @@ -0,0 +1,15 @@ +package internal + +import ( + "errors" +) + +type RecordReader interface { + // ReadRecord should return Records until an error is found. + ReadRecord() (Record, error) +} + +type ReportWriter interface { + // ReportWriter writes report items + Write(ReportItem) error +} diff --git a/internal/reporter.go b/internal/reporter.go index 1004b17..9d18964 100644 --- a/internal/reporter.go +++ b/internal/reporter.go @@ -9,11 +9,6 @@ import ( "sync" ) -type RecordReader interface { - // ReadRecord should return Records until an error is found. - ReadRecord() (Record, error) -} - // Reporter consumes each record to produce ReportItem. type Reporter struct { reader RecordReader diff --git a/internal/trading212/record.go b/internal/trading212/record.go index 4fa598d..403f631 100644 --- a/internal/trading212/record.go +++ b/internal/trading212/record.go @@ -39,6 +39,14 @@ func (r Record) Timestamp() time.Time { return r.timestamp } +func (r Record) Fees() *big.Float { + return new(big.Float) // FIX: +} + +func (r Record) Taxes() *big.Float { + return new(big.Float) // FIX: +} + type RecordReader struct { reader *csv.Reader }