85 lines
5.0 KiB
Go
85 lines
5.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/nmoniz/any2anexoj/internal"
|
|
"github.com/shopspring/decimal"
|
|
)
|
|
|
|
func TestPrettyPrinter_Render(t *testing.T) {
|
|
// Create test data
|
|
aw := internal.NewAggregatorWriter()
|
|
ctx := context.Background()
|
|
|
|
// Add some sample report items
|
|
err := aw.Write(ctx, internal.ReportItem{
|
|
Symbol: "AAPL",
|
|
Nature: internal.NatureG01,
|
|
BrokerCountry: 826, // United Kingdom
|
|
AssetCountry: 840, // United States
|
|
BuyValue: decimal.NewFromFloat(100.50),
|
|
BuyTimestamp: time.Date(2023, 1, 15, 0, 0, 0, 0, time.UTC),
|
|
SellValue: decimal.NewFromFloat(150.75),
|
|
SellTimestamp: time.Date(2023, 6, 20, 0, 0, 0, 0, time.UTC),
|
|
Fees: decimal.NewFromFloat(2.50),
|
|
Taxes: decimal.NewFromFloat(5.00),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("failed to write first report item: %v", err)
|
|
}
|
|
|
|
err = aw.Write(ctx, internal.ReportItem{
|
|
Symbol: "GOOGL",
|
|
Nature: internal.NatureG20,
|
|
BrokerCountry: 826, // United Kingdom
|
|
AssetCountry: 840, // United States
|
|
BuyValue: decimal.NewFromFloat(200.00),
|
|
BuyTimestamp: time.Date(2023, 3, 10, 0, 0, 0, 0, time.UTC),
|
|
SellValue: decimal.NewFromFloat(225.50),
|
|
SellTimestamp: time.Date(2023, 9, 5, 0, 0, 0, 0, time.UTC),
|
|
Fees: decimal.NewFromFloat(3.00),
|
|
Taxes: decimal.NewFromFloat(7.50),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("failed to write second report item: %v", err)
|
|
}
|
|
|
|
// Create English localizer
|
|
localizer, err := NewLocalizer("en")
|
|
if err != nil {
|
|
t.Fatalf("failed to create localizer: %v", err)
|
|
}
|
|
|
|
// Create pretty printer with buffer
|
|
var buf bytes.Buffer
|
|
pp := NewPrettyPrinter(&buf, localizer)
|
|
|
|
// Render the table
|
|
pp.Render(aw)
|
|
|
|
// Get the output
|
|
got := buf.String()
|
|
|
|
// Expected output
|
|
want := `┌───┬────────────────────────────┬───────────────────────────────────┬───────────────────────────────────┬──────────────────────────────────────────────────────────┐
|
|
│ │ │ REALIZATION │ ACQUISITION │ │
|
|
│ │ SOURCE COUNTRY │ CODE │ YEAR │ MONTH │ DAY │ VALUE │ YEAR │ MONTH │ DAY │ VALUE │ EXPENSES AND CH │ TAX PAID ABROAD │ COUNTER COUNTRY │
|
|
│ │ │ │ │ │ │ │ │ │ │ │ ARGES │ │ │
|
|
├───┼─────────────────────┼──────┼──────┼───────┼─────┼──────────────┼──────┼───────┼─────┼──────────────┼─────────────────┼─────────────────┼──────────────────────┤
|
|
│ 1 │ 840 - United States │ G01 │ 2023 │ 6 │ 20 │ 150.75 € │ 2023 │ 1 │ 15 │ 100.50 € │ 2.50 € │ 5.00 € │ 826 - United Kingdom │
|
|
│ 2 │ 840 - United States │ G20 │ 2023 │ 9 │ 5 │ 225.50 € │ 2023 │ 3 │ 10 │ 200.00 € │ 3.00 € │ 7.50 € │ 826 - United Kingdom │
|
|
├───┼─────────────────────┴──────┴──────┴───────┴─────┼──────────────┼──────┴───────┴─────┼──────────────┼─────────────────┼─────────────────┼──────────────────────┤
|
|
│ │ SUM │ 376.25 € │ │ 300.5 € │ 5.5 € │ 12.5 € │ │
|
|
└───┴─────────────────────────────────────────────────┴──────────────┴────────────────────┴──────────────┴─────────────────┴─────────────────┴──────────────────────┘
|
|
`
|
|
|
|
// Compare output
|
|
if got != want {
|
|
t.Errorf("PrettyPrinter.Render() output doesn't match expected.\n\nGot:\n%s\n\nWant:\n%s", got, want)
|
|
}
|
|
}
|