restockbot/twitter_test.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"}, // with space (3060 Ti)
		{"MSI RTX 3060Ti VENTUS 2X OC", "#nvidia #rtx3060ti"},      // without space (3060Ti)
		{"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"},                      // with space (5700 XT)
		{"ASUS Radeon RX 5700XT ROG-STRIX-RX5700XT-O8G-GAMING", "#amd #rx5700xt"}, // without space (5700XT)
		{"MSI Radeon RX 6800", "#amd #rx6800"},
		{"MSI Radeon RX 6800 XT", "#amd #rx6800xt"},                   // with space (6800 XT)
		{"POWERCOLOR RX 6800XT Red Dragon", "#amd #rx6800xt"},         // without space (6800XT)
		{"MSI Radeon RX 6900 XT GAMING X TRIO 16G", "#amd #rx6900xt"}, // with space (6900 XT)
		{"POWERCOLOR RED DEVIL RX 6900XT 16GB", "#amd #rx6900xt"},     // without space (6900XT)
		{"unknown product", ""},
		{"", ""},
	}

	notifier := TwitterNotifier{
		client: nil,
		user:   nil,
		db:     nil,
		hashtagsMap: []map[string]string{
			{"rtx 3060( )?ti": "#nvidia #rtx3060ti"},
			{"rtx 3060": "#nvidia #rtx3060"},
			{"rtx 3070": "#nvidia #rtx3070"},
			{"rtx 3080": "#nvidia #rtx3080"},
			{"rtx 3090": "#nvidia #rtx3090"},
			{"rx 6800( )?xt": "#amd #rx6800xt"},
			{"rx 6900( )?xt": "#amd #rx6900xt"},
			{"rx 6800": "#amd #rx6800"},
			{"rx 5700( )?xt": "#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("for %s, got %s, want %s", tc.input, got, tc.expected)
			} else {
				t.Logf("for %s, want %s, got %s", tc.input, got, tc.expected)
			}
		})
	}
}

func TestFormatPrice(t *testing.T) {
	tests := []struct {
		value    float64
		currency string
		expected string
	}{
		{999.99, "EUR", "999.99€"},
		{999.99, "USD", "$999.99"},
		{999.99, "CHF", "CHF999.99"},
		{999.99, "", "999.99"},
	}

	for i, tc := range tests {
		t.Run(fmt.Sprintf("TestFormatPrice#%d", i), func(t *testing.T) {
			got := formatPrice(tc.value, tc.currency)
			if got != tc.expected {
				t.Errorf("for value %0.2f and currency %s, got %s, want %s", tc.value, tc.currency, got, tc.expected)
			} else {
				t.Logf("for value %0.2f and currency %s, got %s, want %s", tc.value, tc.currency, got, tc.expected)
			}
		})
	}
}

/*
func formatAvailableTweet(shopName string, productName string, productPrice float64, productCurrency string, productURL string, hashtags string) string {
	// format message
	formattedPrice := formatPrice(productPrice, productCurrency)
	message := fmt.Sprintf("%s: %s for %s is available at %s %s", shopName, productName, formattedPrice, productURL, hashtags)

	// truncate tweet if too big
	if utf8.RuneCountInString(message) > tweetMaxSize {
		// maximum tweet size - other characters - additional "…" to say product name has been truncated
		productNameSize := tweetMaxSize - utf8.RuneCountInString(fmt.Sprintf("%s:  for %s is available at %s %s", shopName, formattedPrice, productURL, hashtags)) - 1
		format := fmt.Sprintf("%%s: %%.%ds… for %%s is available at %%s %%s", productNameSize)
		message = fmt.Sprintf(format, shopName, productName, formattedPrice, productURL, hashtags)
	}

	return message
}
*/

func TestFormatAvailableTweet(t *testing.T) {
	tests := []struct {
		shopName        string
		productName     string
		productPrice    float64
		productCurrency string
		productURL      string
		hashtags        string
		expected        string
	}{
		{"shop.com", "my awesome product", 999.99, "USD", "https://shop.com/awesome", "#awesome #product", "shop.com: my awesome product for $999.99 is available at https://shop.com/awesome #awesome #product"},
		{"shop.com", "my awesome product with very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very long name", 999.99, "USD", "https://shop.com/awesome", "#awesome #product", "shop.com: my awesome product with very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very… for $999.99 is available at https://shop.com/awesome #awesome #product"},
	}
	for i, tc := range tests {
		t.Run(fmt.Sprintf("TestFormatAvailableTweet#%d", i), func(t *testing.T) {
			got := formatAvailableTweet(tc.shopName, tc.productName, tc.productPrice, tc.productCurrency, tc.productURL, tc.hashtags)
			if got != tc.expected {
				t.Errorf("for %s, got '%s', want '%s'", tc.productName, got, tc.expected)
			} else {
				t.Logf("for %s, got '%s', want '%s'", tc.productName, got, tc.expected)
			}
		})
	}
}