package main

import (
	"bytes"
	"io/ioutil"
	"net/mail"
	"testing"
)

func TestScrapeChase2021(t *testing.T) {
	b, err := ioutil.ReadFile("./testdata/chase.2021.txt")
	if err != nil {
		t.Fatal(err)
	}
	message := &mail.Message{
		Header: map[string][]string{
			"Subject": []string{"Your $38.84 transaction with TARGET T-1754"},
		},
		Body: bytes.NewReader(b),
	}

	chase := &chaseScraper{}

	gots, err := chase.scrape2021(message)
	if err != nil {
		t.Fatal(err)
	}
	if len(gots) != 1 {
		t.Fatal(gots)
	}
	got := gots[0]

	if got.Account != "8824" {
		t.Fatalf("bad account: %v: %+v", got.Account, got)
	}
	if got.Amount != "38.84" {
		t.Fatalf("bad amount: %v: %+v", got.Amount, got)
	}
	if got.Vendor != "TARGET T-1754" {
		t.Fatalf("bad vendor: %v: %+v", got.Vendor, got)
	}
	t.Logf("%+v", got)
}

func TestScrapeChase2020(t *testing.T) {
	b, err := ioutil.ReadFile("./testdata/chase.2020.txt")
	if err != nil {
		t.Fatal(err)
	}
	message := &mail.Message{
		Body: bytes.NewReader(b),
	}

	chase := &chaseScraper{}

	gots, err := chase.scrape2020(message)
	if err != nil {
		t.Fatal(err)
	}
	if len(gots) != 1 {
		t.Fatal(gots)
	}
	got := gots[0]

	if got.Account != "8824" {
		t.Fatalf("bad account: %v: %+v", got.Account, got)
	}
	if got.Amount != "16.08" {
		t.Fatalf("bad amount: %v: %+v", got.Amount, got)
	}
	if got.Vendor != "PAYPAL *BLIZZARDENT" {
		t.Fatalf("bad vendor: %q: %+v", got.Vendor, got)
	}
	t.Logf("%+v", got)
}