82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package opnsense
|
|
|
|
import (
|
|
"reflect"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/go-kit/log"
|
|
)
|
|
|
|
func TestParsePercentage(t *testing.T) {
|
|
logger := log.NewNopLogger()
|
|
testRegex := regexp.MustCompile(`\d\.\d %`)
|
|
|
|
tests := []struct {
|
|
name string
|
|
value string
|
|
regex *regexp.Regexp
|
|
replacePattern string
|
|
valueTypeName string
|
|
gatewayName string
|
|
expected float64
|
|
}{
|
|
{
|
|
name: "Valid percentage with space",
|
|
value: "50.5 %",
|
|
regex: testRegex,
|
|
replacePattern: " %",
|
|
valueTypeName: "loss",
|
|
gatewayName: "Gateway1",
|
|
expected: 50.5,
|
|
},
|
|
{
|
|
name: "Valid percentage with space",
|
|
value: "5.5 %",
|
|
regex: testRegex,
|
|
replacePattern: " %",
|
|
valueTypeName: "loss",
|
|
gatewayName: "Gateway1",
|
|
expected: 5.5,
|
|
},
|
|
{
|
|
name: "Invalid percentage format",
|
|
value: "invalid %",
|
|
regex: testRegex,
|
|
replacePattern: " %",
|
|
valueTypeName: "loss",
|
|
gatewayName: "Gateway1",
|
|
expected: -1.0,
|
|
},
|
|
{
|
|
name: "Invalid regex match (no space)",
|
|
value: "50.5%",
|
|
regex: testRegex,
|
|
replacePattern: " %",
|
|
valueTypeName: "loss",
|
|
gatewayName: "Gateway1",
|
|
expected: -1,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
result := parseStringToFloatWithReplace(tc.value, tc.regex, tc.replacePattern, tc.valueTypeName, logger)
|
|
if result != tc.expected {
|
|
t.Errorf("parsePercentage(%s, %v, %s, %s, logger, %s) = %v; want %v",
|
|
tc.value, tc.regex, tc.replacePattern, tc.valueTypeName, tc.gatewayName, result, tc.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSliceIntToMapStringInt(t *testing.T) {
|
|
input := []string{"1", "2", "3"}
|
|
expected := map[string]int{"1": 1, "2": 2, "3": 3}
|
|
|
|
result, _ := sliceIntToMapStringInt(input, "test")
|
|
|
|
if !reflect.DeepEqual(result, expected) {
|
|
t.Errorf("Expected %v, got %v", expected, result)
|
|
}
|
|
}
|