64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"net"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestUWaterlooIPv4(t *testing.T) {
|
||
|
expected := "uwaterloo"
|
||
|
actual := IdentifyNetwork(net.ParseIP("129.97.134.71"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUWaterlooIPv6(t *testing.T) {
|
||
|
expected := "uwaterloo"
|
||
|
actual := IdentifyNetwork(net.ParseIP("2620:101:f000:4901:c5c::f:1055"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestOtherIPv4(t *testing.T) {
|
||
|
expected := "other"
|
||
|
actual := IdentifyNetwork(net.ParseIP("8.8.8.8"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestOtherIPv6(t *testing.T) {
|
||
|
expected := "other"
|
||
|
actual := IdentifyNetwork(net.ParseIP("2001:4860:4860::8888"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestIPv4Protocol(t *testing.T) {
|
||
|
expected := "ipv4"
|
||
|
actual := IdentifyIPProtocol(net.ParseIP("129.97.134.71"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestIPv6Protocol(t *testing.T) {
|
||
|
expected := "ipv6"
|
||
|
actual := IdentifyIPProtocol(net.ParseIP("2620:101:f000:4901:c5c::f:1055"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUnknownProtocol(t *testing.T) {
|
||
|
expected := "other"
|
||
|
actual := IdentifyIPProtocol(net.ParseIP("computer"))
|
||
|
if actual != expected {
|
||
|
t.Fatalf("Expected %s, got %s", expected, actual)
|
||
|
}
|
||
|
}
|
||
|
|