From a4e0f4c193b02091775eecdf71d678769816f00b Mon Sep 17 00:00:00 2001
From: Lan Tian <xuyh0120@outlook.com>
Date: Sun, 9 Jan 2022 00:10:06 -0600
Subject: [PATCH] frontend: skip network related tests when unavailable

Fix #46
---
 frontend/bgpmap_test.go  |  4 ++++
 frontend/network_test.go | 30 ++++++++++++++++++++++++++++++
 frontend/whois_test.go   |  2 ++
 3 files changed, 36 insertions(+)
 create mode 100644 frontend/network_test.go

diff --git a/frontend/bgpmap_test.go b/frontend/bgpmap_test.go
index 2806c60..e52de06 100644
--- a/frontend/bgpmap_test.go
+++ b/frontend/bgpmap_test.go
@@ -6,6 +6,8 @@ import (
 )
 
 func TestGetASNRepresentationDNS(t *testing.T) {
+	checkNetwork(t)
+
 	setting.dnsInterface = "asn.cymru.com"
 	setting.whoisServer = ""
 	result := getASNRepresentation("6939")
@@ -15,6 +17,8 @@ func TestGetASNRepresentationDNS(t *testing.T) {
 }
 
 func TestGetASNRepresentationWhois(t *testing.T) {
+	checkNetwork(t)
+
 	setting.dnsInterface = ""
 	setting.whoisServer = "whois.arin.net"
 	result := getASNRepresentation("6939")
diff --git a/frontend/network_test.go b/frontend/network_test.go
new file mode 100644
index 0000000..847c173
--- /dev/null
+++ b/frontend/network_test.go
@@ -0,0 +1,30 @@
+package main
+
+import (
+	"net"
+	"testing"
+	"time"
+)
+
+const (
+	NETWORK_UNKNOWN = 0
+	NETWORK_DOWN = 1
+	NETWORK_UP = 2
+)
+
+var networkState int = NETWORK_UNKNOWN
+func checkNetwork(t *testing.T) {
+	if networkState == NETWORK_UNKNOWN {
+		conn, err := net.DialTimeout("tcp", "8.8.8.8:53", 1*time.Second)
+		if err != nil {
+			networkState = NETWORK_DOWN
+		} else {
+			networkState = NETWORK_UP
+			conn.Close()
+		}
+	}
+
+	if networkState == NETWORK_DOWN {
+		t.Skipf("Test skipped for network error")
+	}
+}
diff --git a/frontend/whois_test.go b/frontend/whois_test.go
index e44b202..3a85a6c 100644
--- a/frontend/whois_test.go
+++ b/frontend/whois_test.go
@@ -6,6 +6,8 @@ import (
 )
 
 func TestWhois(t *testing.T) {
+	checkNetwork(t)
+
 	setting.whoisServer = "whois.arin.net"
 	result := whois("AS6939")
 	if !strings.Contains(result, "HURRICANE") {