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 3a4aba93e5
Release 0.2.0
- new language: go
- new shops: cybertek.fr, mediamarkt.ch
- deprecated shops: alternate.be, minershop.eu
- improved database transaction management
- better web parsing library (ferret, requires headless chrome browser)
- include or exclude products by applying regex on their names
- check for PID file to avoid running the bot twice
- hastags are now configurable

Signed-off-by: Julien Riou <julien@riou.xyz>
2021-02-27 08:10:43 +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")
}
})
}
}