From efaaeb58ebd6571c545569ad18f1396cc701870b Mon Sep 17 00:00:00 2001
From: Simon Zolin <s.zolin@adguard.com>
Date: Wed, 26 Jun 2019 17:53:05 +0300
Subject: [PATCH 1/2] + dhcpd, clients: add more tests

---
 dhcpd/dhcpd_test.go  | 35 +++++++++++++++++++++++++++++++----
 home/clients_test.go | 10 ++++++++++
 2 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/dhcpd/dhcpd_test.go b/dhcpd/dhcpd_test.go
index b5a860fa..e5e16b0d 100644
--- a/dhcpd/dhcpd_test.go
+++ b/dhcpd/dhcpd_test.go
@@ -38,12 +38,19 @@ func TestDHCP(t *testing.T) {
 
 	p = make(dhcp4.Packet, 241)
 
-	// Reserve an IP
+	// Discover and reserve an IP
 	hw = []byte{3, 2, 3, 4, 5, 6}
 	p.SetCHAddr(hw)
-	lease, _ = s.reserveLease(p)
-	check(t, bytes.Equal(lease.HWAddr, hw), "lease.HWAddr")
-	check(t, bytes.Equal(lease.IP, []byte{1, 1, 1, 1}), "lease.IP")
+	p.SetCIAddr([]byte{0, 0, 0, 0})
+	opt = make(dhcp4.Options, 10)
+	p2 = s.handleDiscover(p, opt)
+	opt = p2.ParseOptions()
+	check(t, bytes.Equal(opt[dhcp4.OptionDHCPMessageType], []byte{byte(dhcp4.Offer)}), "dhcp4.Offer")
+	check(t, bytes.Equal(p2.YIAddr(), []byte{1, 1, 1, 1}), "p2.YIAddr")
+	check(t, bytes.Equal(p2.CHAddr(), hw), "p2.CHAddr")
+	check(t, bytes.Equal(opt[dhcp4.OptionIPAddressLeaseTime], dhcp4.OptionsLeaseTime(5*time.Second)), "OptionIPAddressLeaseTime")
+	check(t, bytes.Equal(opt[dhcp4.OptionServerIdentifier], s.ipnet.IP), "OptionServerIdentifier")
+
 	lease = s.findLease(p)
 	check(t, bytes.Equal(lease.HWAddr, hw), "lease.HWAddr")
 	check(t, bytes.Equal(lease.IP, []byte{1, 1, 1, 1}), "lease.IP")
@@ -88,6 +95,8 @@ func TestDHCP(t *testing.T) {
 	check(t, bytes.Equal(opt[dhcp4.OptionIPAddressLeaseTime], dhcp4.OptionsLeaseTime(5*time.Second)), "OptionIPAddressLeaseTime")
 	check(t, bytes.Equal(opt[dhcp4.OptionServerIdentifier], s.ipnet.IP), "OptionServerIdentifier")
 
+	check(t, bytes.Equal(s.FindIPbyMAC(hw), []byte{1, 1, 1, 1}), "FindIPbyMAC")
+
 	// Commit the previously reserved lease #2
 	hw = []byte{2, 2, 3, 4, 5, 6}
 	p.SetCHAddr(hw)
@@ -103,10 +112,28 @@ func TestDHCP(t *testing.T) {
 	lease, _ = s.reserveLease(p)
 	check(t, lease == nil, "lease == nil")
 
+	s.reset()
+	testStaticLeases(t, &s)
+
 	s.reset()
 	misc(t, &s)
 }
 
+func testStaticLeases(t *testing.T, s *Server) {
+	var err error
+	var l Lease
+	l.IP = []byte{1, 1, 1, 1}
+	l.HWAddr = []byte{2, 2, 3, 4, 5, 6}
+	err = s.AddStaticLease(l)
+	check(t, err == nil, "AddStaticLease")
+
+	ll := s.StaticLeases()
+	check(t, len(ll) != 0 && bytes.Equal(ll[0].IP, []byte{1, 1, 1, 1}), "StaticLeases")
+
+	err = s.RemoveStaticLease(l)
+	check(t, err == nil, "RemoveStaticLease")
+}
+
 // Small tests that don't require a static server's state
 func misc(t *testing.T, s *Server) {
 	var p, p2 dhcp4.Packet
diff --git a/home/clients_test.go b/home/clients_test.go
index a8cb44c2..39ef4b6d 100644
--- a/home/clients_test.go
+++ b/home/clients_test.go
@@ -29,6 +29,16 @@ func TestClients(t *testing.T) {
 		t.Fatalf("clientAdd #2")
 	}
 
+	c, b = clientFind("1.1.1.1")
+	if !b || c.Name != "client1" {
+		t.Fatalf("clientFind #1")
+	}
+
+	c, b = clientFind("2.2.2.2")
+	if !b || c.Name != "client2" {
+		t.Fatalf("clientFind #2")
+	}
+
 	// failed add - name in use
 	c = Client{
 		IP:   "1.2.3.5",

From 25da23497a19118a22b97d64749fa70337544116 Mon Sep 17 00:00:00 2001
From: Simon Zolin <s.zolin@adguard.com>
Date: Wed, 26 Jun 2019 18:13:09 +0300
Subject: [PATCH 2/2] + dnsfilter: more tests

---
 dnsfilter/dnsfilter_test.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/dnsfilter/dnsfilter_test.go b/dnsfilter/dnsfilter_test.go
index 00324471..27d203bb 100644
--- a/dnsfilter/dnsfilter_test.go
+++ b/dnsfilter/dnsfilter_test.go
@@ -10,6 +10,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/bluele/gcache"
 	"github.com/miekg/dns"
 )
 
@@ -572,3 +573,17 @@ func BenchmarkSafeSearchParallel(b *testing.B) {
 		}
 	})
 }
+
+func TestDnsfilterDialCache(t *testing.T) {
+	d := Dnsfilter{}
+	dialCache = gcache.New(1).LRU().Expiration(30 * time.Minute).Build()
+
+	d.shouldBeInDialCache("hostname")
+	if searchInDialCache("hostname") != "" {
+		t.Errorf("searchInDialCache")
+	}
+	addToDialCache("hostname", "1.1.1.1")
+	if searchInDialCache("hostname") != "1.1.1.1" {
+		t.Errorf("searchInDialCache")
+	}
+}