// SPDX-FileCopyrightText: 2022 Free Mobile
// SPDX-License-Identifier: AGPL-3.0-only

package clickhouse

import (
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/mitchellh/mapstructure"

	"akvorado/common/helpers"
)

func TestNetworkNamesUnmarshalHook(t *testing.T) {
	cases := []struct {
		Description string
		Input       map[string]interface{}
		Output      map[string]NetworkAttributes
		Error       bool
	}{
		{
			Description: "nil",
			Input:       nil,
		}, {
			Description: "empty",
			Input:       gin.H{},
		}, {
			Description: "IPv4",
			Input:       gin.H{"203.0.113.0/24": gin.H{"name": "customer"}},
			Output: map[string]NetworkAttributes{
				"203.0.113.0/24": {Name: "customer"},
			},
		}, {
			Description: "IPv6",
			Input:       gin.H{"2001:db8:1::/64": gin.H{"name": "customer"}},
			Output: map[string]NetworkAttributes{
				"2001:db8:1::/64": {Name: "customer"},
			},
		}, {
			Description: "IPv4 subnet (compatibility)",
			Input:       gin.H{"203.0.113.0/24": "customer"},
			Output: map[string]NetworkAttributes{
				"203.0.113.0/24": {Name: "customer"},
			},
		}, {
			Description: "IPv6 subnet (compatibility)",
			Input:       gin.H{"2001:db8:1::/64": "customer"},
			Output: map[string]NetworkAttributes{
				"2001:db8:1::/64": {Name: "customer"},
			},
		}, {
			Description: "all attributes",
			Input: gin.H{"203.0.113.0/24": gin.H{
				"name":   "customer1",
				"role":   "customer",
				"site":   "paris",
				"region": "france",
				"tenant": "mobile",
			}},
			Output: map[string]NetworkAttributes{"203.0.113.0/24": {
				Name:   "customer1",
				Role:   "customer",
				Site:   "paris",
				Region: "france",
				Tenant: "mobile",
			}},
		}, {
			Description: "Invalid subnet (1)",
			Input:       gin.H{"192.0.2.1/38": "customer"},
			Error:       true,
		}, {
			Description: "Invalid subnet (2)",
			Input:       gin.H{"192.0.2.1/255.0.255.0": "customer"},
			Error:       true,
		},
	}
	for _, tc := range cases {
		t.Run(tc.Description, func(t *testing.T) {
			var got helpers.SubnetMap[NetworkAttributes]
			decoder, err := mapstructure.NewDecoder(helpers.GetMapStructureDecoderConfig(&got))
			if err != nil {
				t.Fatalf("NewDecoder() error:\n%+v", err)
			}
			err = decoder.Decode(tc.Input)
			if err != nil && !tc.Error {
				t.Fatalf("Decode() error:\n%+v", err)
			} else if err == nil && tc.Error {
				t.Fatal("Decode() did not return an error")
			} else if diff := helpers.Diff(got.ToMap(), tc.Output); diff != "" {
				t.Fatalf("Decode() (-got, +want):\n%s", diff)
			}
		})
	}
}

func TestDefaultConfiguration(t *testing.T) {
	config := DefaultConfiguration()
	config.Kafka.Topic = "flow"
	if err := helpers.Validate.Struct(config); err != nil {
		t.Fatalf("validate.Struct() error:\n%+v", err)
	}
}