// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package socket

import (
	"net"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestListenerTable(t *testing.T) {
	l := NewListenerTable()

	proto := uint8(4)
	lAddr := net.ParseIP("192.0.2.1")
	httpPort := 80
	rAddr := net.ParseIP("198.18.0.1")
	ephemeralPort := 48199
	ipv6Addr := net.ParseIP("2001:db8:fe80::217:f2ff:fe07:ed62")

	// Any socket with remote port of 0 is listening.
	assert.Equal(t, Listening, l.Direction(proto, lAddr, httpPort, net.IPv4zero, 0))

	// Listener on 192.0.2.1:80
	l.Put(proto, lAddr, httpPort)

	assert.Equal(t, Incoming, l.Direction(proto, lAddr, httpPort, rAddr, ephemeralPort))
	assert.Equal(t, Outgoing, l.Direction(0, lAddr, httpPort, rAddr, ephemeralPort))
	assert.Equal(t, Outgoing, l.Direction(proto, lAddr, ephemeralPort, rAddr, ephemeralPort))

	// Listener on 0.0.0.0:80
	l.Reset()
	l.Put(proto, net.IPv4zero, httpPort)

	assert.Equal(t, Incoming, l.Direction(proto, lAddr, httpPort, rAddr, ephemeralPort))
	assert.Equal(t, Outgoing, l.Direction(proto, ipv6Addr, httpPort, rAddr, ephemeralPort))

	// Listener on :::80
	l.Reset()
	l.Put(proto, net.IPv6zero, httpPort)

	assert.Equal(t, Incoming, l.Direction(proto, ipv6Addr, httpPort, rAddr, ephemeralPort))
	assert.Equal(t, Outgoing, l.Direction(proto, lAddr, httpPort, rAddr, ephemeralPort))
}