From a6c5121d54d27cb56bb23bb761162c7d265678bc Mon Sep 17 00:00:00 2001 From: bel Date: Thu, 29 Jul 2021 23:42:16 -0600 Subject: [PATCH] check ledger output a little bit --- upload.go | 6 +++++- upload_test.go | 6 +++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/upload.go b/upload.go index 8998b15..dd19372 100755 --- a/upload.go +++ b/upload.go @@ -8,6 +8,7 @@ import ( "net/http" "net/url" "os" + "regexp" "strconv" "strings" "time" @@ -60,7 +61,10 @@ func uploadLedger(config Config, transaction *Transaction) error { if amount > 0 { remote = "Deposit:" } - remote += strings.ReplaceAll(transaction.Vendor, " ", "") + regexp := regexp.MustCompile(`[0-9a-zA-Z]`) + for _, substr := range regexp.FindAllString(transaction.Vendor, -1) { + remote += substr + } fmt.Fprintf(f, "%-50s%-s\n", formatGMailDate(transaction.Date), transaction.Vendor) fmt.Fprintf(f, "%-50s%-50s$%.2f\n", "", "AssetAccount:"+transaction.Bank.String(), amount) fmt.Fprintf(f, "%-50s%-s\n", "", remote) diff --git a/upload_test.go b/upload_test.go index 7e6f47d..97bd0c9 100644 --- a/upload_test.go +++ b/upload_test.go @@ -1,6 +1,7 @@ package main import ( + "bytes" "io/ioutil" "local/storage" "path" @@ -18,7 +19,7 @@ func TestUploadLedger(t *testing.T) { ID: uuid.New().String(), Bank: Chase, Amount: "1.10", - Vendor: "vendor", + Vendor: "my's totally realistic!!! VENDOR 11", Date: "today", }, }, @@ -39,6 +40,9 @@ func TestUploadLedger(t *testing.T) { if err != nil { t.Fatal(err) } + if !bytes.Contains(b, []byte("mystotallyrealisticVENDOR11")) { + t.Fatal(string(b)) + } t.Logf("\n%s", b) }) }