ing

discovery/protocol/mdns/mdns_test.go

@@ -6,37 +6,45 @@ import (
 
 	omd "git.loafle.net/overflow/model/discovery"
 	omm "git.loafle.net/overflow/model/meta"
-	"git.loafle.net/overflow_scanner/probe/model"
+	"git.loafle.net/overflow_scanner/probe/discovery/session"
 	"github.com/grandcat/zeroconf"
 )
 
 func TestScan(t *testing.T) {
+	s := session.MockDiscoverySession()
+	s.InitWithDiscoverHost(
+		nil,
+		&omd.Zone{
+			Network:    "192.168.1.0/24",
+			Iface:      "enp3s0",
+			MetaIPType: omm.ToMetaIPType(omm.MetaIPTypeEnumV4),
+			Address:    "192.168.1.101",
+			Mac:        "44:8a:5b:f1:f1:f3",
+		},
+		&omd.DiscoverHost{
+			MetaIPType:     omm.ToMetaIPType(omm.MetaIPTypeEnumV4),
+			FirstScanRange: "192.168.1.1",
+			LastScanRange:  "192.168.1.254",
+		},
+	)
+
 	type args struct {
-		discovered model.Discovered
+		discoverySession session.DiscoverySession
 	}
 	tests := []struct {
 		name string
 		args args
 	}{
-		// TODO: Add test cases.
 		{
 			name: "1",
 			args: args{
-				discovered: model.New(
-					&omd.Zone{
-						Network:    "192.168.1.0/24",
-						Iface:      "enp3s0",
-						MetaIPType: omm.ToMetaIPType(omm.MetaIPTypeEnumV4),
-						Address:    "192.168.1.101",
-						Mac:        "44:8a:5b:f1:f1:f3",
-					},
-				),
+				discoverySession: s,
 			},
 		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			Scan(tt.args.discovered)
+			Scan(tt.args.discoverySession)
 		})
 	}
 }