package prop

import (
	"testing"
	"time"

	"github.com/pion/mediadevices/pkg/frame"
)

func TestCompareMatch(t *testing.T) {
	testDataSet := map[string]struct {
		a     MediaConstraints
		b     Media
		match bool
	}{
		"DeviceIDExactUnmatch": {
			MediaConstraints{
				DeviceID: StringExact("abc"),
			},
			Media{
				DeviceID: "cde",
			},
			false,
		},
		"DeviceIDExactMatch": {
			MediaConstraints{
				DeviceID: StringExact("abc"),
			},
			Media{
				DeviceID: "abc",
			},
			true,
		},
		"IntIdealUnmatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				Width: Int(30),
			}},
			Media{Video: Video{
				Width: 50,
			}},
			true,
		},
		"IntIdealMatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				Width: Int(30),
			}},
			Media{Video: Video{
				Width: 30,
			}},
			true,
		},
		"IntExactUnmatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				Width: IntExact(30),
			}},
			Media{Video: Video{
				Width: 50,
			}},
			false,
		},
		"IntExactMatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				Width: IntExact(30),
			}},
			Media{Video: Video{
				Width: 30,
			}},
			true,
		},
		"IntRangeUnmatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				Width: IntRanged{Min: 30, Max: 40},
			}},
			Media{Video: Video{
				Width: 50,
			}},
			false,
		},
		"IntRangeMatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				Width: IntRanged{Min: 30, Max: 40},
			}},
			Media{Video: Video{
				Width: 35,
			}},
			true,
		},
		"FrameFormatOneOfUnmatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				FrameFormat: FrameFormatOneOf{frame.FormatYUYV, frame.FormatUYVY},
			}},
			Media{Video: Video{
				FrameFormat: frame.FormatYUYV,
			}},
			true,
		},
		"FrameFormatOneOfMatch": {
			MediaConstraints{VideoConstraints: VideoConstraints{
				FrameFormat: FrameFormatOneOf{frame.FormatYUYV, frame.FormatUYVY},
			}},
			Media{Video: Video{
				FrameFormat: frame.FormatMJPEG,
			}},
			false,
		},
		"DurationExactUnmatch": {
			MediaConstraints{AudioConstraints: AudioConstraints{
				Latency: DurationExact(time.Second),
			}},
			Media{Audio: Audio{
				Latency: time.Second + time.Millisecond,
			}},
			false,
		},
		"DurationExactMatch": {
			MediaConstraints{AudioConstraints: AudioConstraints{
				Latency: DurationExact(time.Second),
			}},
			Media{Audio: Audio{
				Latency: time.Second,
			}},
			true,
		},
		"DurationRangedUnmatch": {
			MediaConstraints{AudioConstraints: AudioConstraints{
				Latency: DurationRanged{Max: time.Second},
			}},
			Media{Audio: Audio{
				Latency: time.Second + time.Millisecond,
			}},
			false,
		},
		"DurationRangedMatch": {
			MediaConstraints{AudioConstraints: AudioConstraints{
				Latency: DurationRanged{Max: time.Second},
			}},
			Media{Audio: Audio{
				Latency: time.Millisecond,
			}},
			true,
		},
	}

	for name, testData := range testDataSet {
		testData := testData
		t.Run(name, func(t *testing.T) {
			_, match := testData.a.FitnessDistance(testData.b)
			if match != testData.match {
				t.Errorf("matching flag differs, expected: %v, got: %v", testData.match, match)
			}
		})
	}
}

func TestMergeWithZero(t *testing.T) {
	a := Media{
		Video: Video{
			Width: 30,
		},
	}

	b := MediaConstraints{
		VideoConstraints: VideoConstraints{
			Height: Int(100),
		},
	}

	a.Merge(b)

	if a.Width == 0 {
		t.Error("expected a.Width to be 30, but got 0")
	}

	if a.Height == 0 {
		t.Error("expected a.Height to be 100, but got 0")
	}
}

func TestMergeWithSameField(t *testing.T) {
	a := Media{
		Video: Video{
			Width: 30,
		},
	}

	b := MediaConstraints{
		VideoConstraints: VideoConstraints{
			Width: Int(100),
		},
	}

	a.Merge(b)

	if a.Width != 100 {
		t.Error("expected a.Width to be 100, but got 0")
	}
}

func TestMergeNested(t *testing.T) {
	type constraints struct {
		Media
	}

	a := constraints{
		Media{
			Video: Video{
				Width: 30,
			},
		},
	}

	b := MediaConstraints{
		VideoConstraints: VideoConstraints{
			Width: Int(100),
		},
	}

	a.Merge(b)

	if a.Width != 100 {
		t.Error("expected a.Width to be 100, but got 0")
	}
}