// Copyright © by Jeff Foley 2017-2023. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
// SPDX-License-Identifier: Apache-2.0

package requests

import (
	"net"
	"testing"
	"time"

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

func TestEmpty(t *testing.T) {
	cache := NewASNCache()

	if entry := cache.ASNSearch(0); entry != nil {
		t.Errorf("ASNSearch returned a non-nil value when empty: %v", entry)
	}
	if entry := cache.AddrSearch("72.237.4.113"); entry != nil {
		t.Errorf("AddrSearch returned a non-nil value when empty: %v", entry)
	}
}

func TestUpdate(t *testing.T) {
	cache := NewASNCache()

	if entry := cache.AddrSearch("72.237.4.113"); entry != nil {
		t.Errorf("AddrSearch returned a non-nil value when empty: %v", entry)
	}

	cache.Update(&ASNRequest{
		Address: "72.237.4.113",
		ASN:     26808,
		Prefix:  "72.237.4.0/24",
	})

	if entry := cache.AddrSearch("72.237.4.113"); entry == nil {
		t.Errorf("AddrSearch returned nil after updated with the correct data")
	}

	cache.Update(&ASNRequest{
		Address:        "72.237.4.113",
		ASN:            26808,
		Prefix:         "72.237.4.0/24",
		CC:             "US",
		Registry:       "ARIN",
		AllocationDate: time.Now(),
		Description:    "UTICA-COLLEGE",
		Netblocks:      []string{"72.237.4.0/24", "8.24.68.0/23"},
	})

	if entry := cache.AddrSearch("72.237.4.113"); entry == nil || entry.CC != "US" || entry.Description != "UTICA-COLLEGE" {
		t.Errorf("Update failed to enhance the ASN entry with the more detailed data")
	}
	if entry := cache.AddrSearch("8.24.68.1"); entry == nil || entry.ASN != 26808 {
		t.Errorf("Update failed to add the new netblock to the ASN")
	}
}

func TestASNSearch(t *testing.T) {
	cache := NewASNCache()

	if entry := cache.ASNSearch(26808); entry != nil {
		t.Errorf("ASNSearch returned a non-nil value when empty: %v", entry)
	}

	cache.Update(&ASNRequest{
		Address: "72.237.4.113",
		ASN:     26808,
		Prefix:  "72.237.4.0/24",
	})

	if entry := cache.ASNSearch(26808); entry == nil {
		t.Errorf("ASNSearch returned nil after updated with the correct data")
	}
}

func TestAddrSearch(t *testing.T) {
	cache := NewASNCache()

	if entry := cache.AddrSearch("127.0.0.1"); entry == nil || entry.ASN != 0 {
		t.Errorf("AddrSearch returned nil when searching for a reserved network address block")
	}
	if entry := cache.AddrSearch("72.237.4.113"); entry != nil {
		t.Errorf("AddrSearch returned a non-nil value when empty: %v", entry)
	}

	cache.Update(&ASNRequest{
		Address:        "72.237.4.113",
		ASN:            26808,
		Prefix:         "72.237.4.0/24",
		CC:             "US",
		Registry:       "ARIN",
		AllocationDate: time.Now(),
		Description:    "UTICA-COLLEGE",
		Netblocks:      []string{"72.237.4.0/24", "8.24.68.0/23"},
	})

	if entry := cache.AddrSearch("72.237.4.120"); entry == nil {
		t.Errorf("AddrSearch returned nil after updated with the correct data")
	}

	addr := "8.24.68.1"
	ip := net.ParseIP(addr)
	entry := cache.AddrSearch(addr)
	if entry == nil {
		t.Fatalf("AddrSearch returned nil for an address known to the cache")
	}
	if _, ipnet, err := net.ParseCIDR(entry.Prefix); err != nil || !ipnet.Contains(ip) {
		t.Errorf("AddrSearch returned the wrong Prefix value for the provided IP address: %s", entry.Prefix)
	}
}

func TestIsReservedAddress(t *testing.T) {
	t.Parallel()
	tests := []struct {
		name       string
		addr       string
		isReserved bool
	}{
		{
			name:       "Test Invalid IP",
			addr:       "300.300.300.300",
			isReserved: false,
		},
		{
			name:       "Test Reserved Address",
			addr:       "192.168.0.0",
			isReserved: true,
		},
		{
			name:       "Test Unreserved Address",
			addr:       "202.145.4.15",
			isReserved: false,
		},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			isReserved, _ := isReservedAddress(test.addr)
			require.Equal(t, isReserved, test.isReserved)
		})
	}

}