billing_export_test.go 719 B

12345678910111213141516171819202122232425262728
  1. package service
  2. import (
  3. "testing"
  4. "time"
  5. "github.com/stretchr/testify/require"
  6. )
  7. func TestParseBillingYearMonth(t *testing.T) {
  8. start, end, err := parseBillingYearMonth(2026, 9)
  9. require.NoError(t, err)
  10. require.Equal(t, time.Date(2026, 9, 1, 0, 0, 0, 0, time.UTC), start)
  11. require.Equal(t, time.Date(2026, 10, 1, 0, 0, 0, 0, time.UTC), end)
  12. _, _, err = parseBillingYearMonth(2026, 13)
  13. require.Error(t, err)
  14. _, _, err = parseBillingYearMonth(2019, 1)
  15. require.Error(t, err)
  16. }
  17. func TestBillingExportNilDB(t *testing.T) {
  18. svc := NewBillingExportService(nil)
  19. _, err := svc.Statement(nil, 1, 2026, 9)
  20. require.Error(t, err)
  21. _, _, err = svc.ExportCSV(nil, 1, time.Now(), time.Now())
  22. require.Error(t, err)
  23. }