fix(stuinput): repair tests

ooni · May 31, 2023 · 854189c · 854189c
1 parent df9677a
commit 854189c
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/internal/stuninput/stuninput_test.go b/internal/stuninput/stuninput_test.go
@@ -1,17 +1,19 @@
 package stuninput
 
-import "testing"
+import (
+	"strings"
+	"testing"
+)
 
 func TestAsSnowflakeInput(t *testing.T) {
 	outputs := AsSnowflakeInput()
 	if len(outputs) != len(inputs) {
 		t.Fatal("unexpected number of entries")
 	}
-	for idx := 0; idx < len(inputs); idx++ {
-		output := outputs[idx]
-		input := "stun:" + inputs[idx]
-		if input != output {
-			t.Fatal("mismatch")
+	for _, output := range outputs {
+		output = strings.TrimPrefix(output, "stun:")
+		if !inputs[output] {
+			t.Fatal("not found in inputs", output)
 		}
 	}
 }
@@ -21,11 +23,10 @@ func TestAsStunReachabilityInput(t *testing.T) {
 	if len(outputs) != len(inputs) {
 		t.Fatal("unexpected number of entries")
 	}
-	for idx := 0; idx < len(inputs); idx++ {
-		output := outputs[idx]
-		input := "stun://" + inputs[idx]
-		if input != output {
-			t.Fatal("mismatch")
+	for _, output := range outputs {
+		output = strings.TrimPrefix(output, "stun://")
+		if !inputs[output] {
+			t.Fatal("not found in inputs", output)
 		}
 	}
 }