From db58bd33545afc2d1a3f0b9630e65d6c0f8c20c9 Mon Sep 17 00:00:00 2001
From: Lan Tian <xuyh0120@outlook.com>
Date: Sat, 6 May 2023 00:26:40 -0700
Subject: [PATCH] frontend: add a test case for lookup DNS -> WHOIS fallback

---
 frontend/asn_cache_test.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/frontend/asn_cache_test.go b/frontend/asn_cache_test.go
index 2ba914d..6178482 100644
--- a/frontend/asn_cache_test.go
+++ b/frontend/asn_cache_test.go
@@ -19,6 +19,18 @@ func TestGetASNRepresentationDNS(t *testing.T) {
 	}
 }
 
+func TestGetASNRepresentationDNSFallback(t *testing.T) {
+	checkNetwork(t)
+
+	setting.dnsInterface = "invalid.example.com"
+	setting.whoisServer = "whois.arin.net"
+	cache := make(ASNCache)
+	result := cache.Lookup("6939")
+	if !strings.Contains(result, "HURRICANE") {
+		t.Errorf("Lookup AS6939 failed, got %s", result)
+	}
+}
+
 func TestGetASNRepresentationWhois(t *testing.T) {
 	checkNetwork(t)