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/parser_url_test.go
Julien Riou e67ab63ca8
Prepare for new parsers
- Rename "Parser" to "URLParser"
- Make "Parse" function generic
- Rename "crawlShop" function to "handleProducts"
- Reduce "handleProducts" footprint a little bit

Signed-off-by: Julien Riou <julien@riou.xyz>
2021-03-23 09:00:10 +01:00

86 lines
2.8 KiB
Go

package main
import (
"fmt"
"testing"
)
func TestFilterInclusive(t *testing.T) {
tests := []struct {
regex string // inclusive regex
name string // product name
included bool // should be included or not
}{
{"(?i)(rtx|rx)(.*)(3060|3070|3080|3090|5700|6800|6900)( )?(xt|ti)?", "MSI GeForce RTX 3060 GAMING X", true}, // 3060 in the include regex
{"(?i)(rtx|rx)(.*)(3060|3070|3080|3090|5700|6800|6900)( )?(xt|ti)?", "ASUS AMD Radeon RX 5600 XT TUF Gaming X3", false}, // 5600 not in the include regex
{"", "MSI GeForce RTX 3060 GAMING X", true}, // do nothing when the include regex is empty
}
for i, tc := range tests {
t.Run(fmt.Sprintf("TestFilterInclusive#%d", i), func(t *testing.T) {
p, err := NewURLParser("", "", tc.regex, "")
if err != nil {
t.Errorf("failed to initialize parser: %s", err)
} else {
products := []*Product{{Name: tc.name}}
filtered := p.filterInclusive(products)
included := false
for _, product := range filtered {
if product.Name == tc.name && !included {
included = true
}
}
if tc.included != included {
t.Errorf("regex '%s' for product '%s': got included=%t, want included=%t", tc.regex, tc.name, included, tc.included)
} else {
if included {
t.Logf("regex '%s' includes product '%s'", tc.regex, tc.name)
} else {
t.Logf("regex '%s' excludes product '%s'", tc.regex, tc.name)
}
}
}
})
}
}
func TestFilterExclusive(t *testing.T) {
tests := []struct {
regex string // exclusive regex
name string // product name
included bool // should be included or not
}{
{"(?i)(rtx|rx)(.*)(3060|3070|3080|3090|5700|6800|6900)( )?(xt|ti)?", "MSI GeForce RTX 3060 GAMING X", false}, // 3060 in the exclude regex
{"(?i)(rtx|rx)(.*)(3060|3070|3080|3090|5700|6800|6900)( )?(xt|ti)?", "ASUS AMD Radeon RX 5600 XT TUF Gaming X3", true}, // 5600 not in the exclude regex
{"", "MSI GeForce RTX 3060 GAMING X", true}, // do nothing when the exclude regex is empty
}
for i, tc := range tests {
t.Run(fmt.Sprintf("TestFilterExclusive#%d", i), func(t *testing.T) {
p, err := NewURLParser("", "", "", tc.regex)
if err != nil {
t.Errorf("failed to initialize parser: %s", err)
} else {
products := []*Product{{Name: tc.name}}
filtered := p.filterExclusive(products)
included := false
for _, product := range filtered {
if product.Name == tc.name && !included {
included = true
}
}
if tc.included != included {
t.Errorf("regex '%s' for product '%s': got included=%t, want included=%t", tc.regex, tc.name, included, tc.included)
} else {
if included {
t.Logf("regex '%s' includes product '%s'", tc.regex, tc.name)
} else {
t.Logf("regex '%s' excludes product '%s'", tc.regex, tc.name)
}
}
}
})
}
}