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

58 lines
1.5 KiB
Go
Raw Normal View History

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")
}
})
}
}