Archived
1
0
Fork 0
This repository has been archived on 2024-12-18. You can view files and clone it, but cannot push or open issues or pull requests.
restockbot/twitter_test.go
Julien Riou 4342f65211
Use array of key/value for Twitter hashtags (#1)
With Go, maps are not ordered the way they are declared. Keys can be read at any
order. When a pattern is too large ("rtx 3060"), when placed first, it can match
a name of another product ("rtx 3060 ti"). When placed second, the good hashtag
is chosen. This commit uses an array of maps because arrays are ordered.

Signed-off-by: Julien Riou <julien@riou.xyz>
2021-02-27 18:34:13 +01:00

57 lines
1.5 KiB
Go

package main
import (
"fmt"
"testing"
)
func TestBuildHashtags(t *testing.T) {
tests := []struct {
input string
expected string
}{
{"MSI GeForce RTX 3060 GAMING X", "#nvidia #rtx3060"},
{"MSI GeForce RTX 3060 Ti GAMING X", "#nvidia #rtx3060ti"},
{"MSI GeForce RTX 3070 GAMING TRIO", "#nvidia #rtx3070"},
{"MSI GeForce RTX 3080 SUPRIM X", "#nvidia #rtx3080"},
{"MSI GeForce RTX 3090 GAMING X TRIO 24G", "#nvidia #rtx3090"},
{"MSI Radeon RX 5700 XT GAMING X", "#amd #rx5700xt"},
{"MSI Radeon RX 6800", "#amd #rx6800"},
{"MSI Radeon RX 6800 XT", "#amd #rx6800xt"},
{"MSI Radeon RX 6900 XT GAMING X TRIO 16G", "#amd #rx6900xt"},
{"unknown product", ""},
{"", ""},
}
notifier := TwitterNotifier{
client: nil,
user: nil,
db: nil,
hashtagsMap: []map[string]string{
{"rtx 3060 ti": "#nvidia #rtx3060ti"},
{"rtx 3060ti": "#nvidia #rtx3060ti"},
{"rtx 3060": "#nvidia #rtx3060"},
{"rtx 3070": "#nvidia #rtx3070"},
{"rtx 3080": "#nvidia #rtx3080"},
{"rtx 3090": "#nvidia #rtx3090"},
{"rx 6800 xt": "#amd #rx6800xt"},
{"rx 6800xt": "#amd #rx6800xt"},
{"rx 6900xt": "#amd #rx6900xt"},
{"rx 6900 xt": "#amd #rx6900xt"},
{"rx 6800": "#amd #rx6800"},
{"rx 5700 xt": "#amd #rx5700xt"},
{"rx 5700xt": "#amd #rx5700xt"},
},
}
for i, tc := range tests {
t.Run(fmt.Sprintf("TestBuildHashtags#%d", i), func(t *testing.T) {
got := notifier.buildHashtags(tc.input)
if got != tc.expected {
t.Errorf("got %s, want %s", got, tc.expected)
} else {
t.Logf("success")
}
})
}
}