Function createMockSession

modules/arp_spoof/arp_spoof_test.go:195–261

Create a mock session for testing

()

// Create a mock session for testing
func createMockSession() (*MockSession, *MockPacketQueue, *MockFirewall) {
	// Create interface
	iface := &network.Endpoint{
		IpAddress: "192.168.1.100",
		HwAddress: "aa:bb:cc:dd:ee:ff",
		Hostname:  "eth0",
	}
	iface.SetIP("192.168.1.100")
	iface.SetBits(24)

	// Parse interface addresses
	ifaceIP := net.ParseIP("192.168.1.100")
	ifaceHW, _ := net.ParseMAC("aa:bb:cc:dd:ee:ff")
	iface.IP = ifaceIP
	iface.HW = ifaceHW

	// Create gateway
	gateway := &network.Endpoint{
		IpAddress: "192.168.1.1",
		HwAddress: "11:22:33:44:55:66",
	}
	gatewayIP := net.ParseIP("192.168.1.1")
	gatewayHW, _ := net.ParseMAC("11:22:33:44:55:66")
	gateway.IP = gatewayIP
	gateway.HW = gatewayHW

	// Create mock queue and firewall
	mockQueue := NewMockPacketQueue()
	mockFirewall := NewMockFirewall()

	// Create environment
	env, _ := session.NewEnvironment("")

	// Create LAN
	aliases, _ := data.NewUnsortedKV("", 0)
	lan := network.NewLAN(iface, gateway, aliases, func(e *network.Endpoint) {}, func(e *network.Endpoint) {})

	// Create session
	sess := &session.Session{
		Interface: iface,
		Gateway:   gateway,
		Lan:       lan,
		StartedAt: time.Now(),
		Active:    true,
		Env:       env,
		Queue:     mockQueue.Queue,
		Firewall:  mockFirewall,
		Modules:   make(session.ModuleList, 0),
	}

	// Initialize events
	sess.Events = session.NewEventPool(false, false)

	// Add mock net.recon module
	mockNetRecon := NewMockNetRecon(sess)
	sess.Modules = append(sess.Modules, mockNetRecon)

	// Create mock session wrapper