arguments_test.go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0

package oauth2

import (
	"testing"

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

type exactTestCase struct {
	args   Arguments
	exact  string
	expect bool
}

var exactTests = []exactTestCase{
	{
		args:   Arguments{"foo"},
		exact:  "foo",
		expect: true,
	},
	{
		args:   Arguments{"foo", "bar"},
		exact:  "foo",
		expect: false,
	},
	{
		args:   Arguments{"foo", "bar"},
		exact:  "bar",
		expect: false,
	},
	{
		args:   Arguments{"foo", "bar"},
		exact:  "baz",
		expect: false,
	},
	{
		args:   Arguments{},
		exact:  "baz",
		expect: false,
	},
}

func TestArgumentsExactOne(t *testing.T) {
	testCases := append(exactTests, []exactTestCase{ //nolint:gocritic
		{
			args:   Arguments{"foo", "bar"},
			exact:  "foo bar",
			expect: false,
		},
	}...)

	for k, c := range testCases {
		assert.Equal(t, c.expect, c.args.ExactOne(c.exact), "%d", k)
	}
}

func TestArgumentsHas(t *testing.T) {
	for k, c := range []struct {
		args   Arguments
		has    []string
		expect bool
	}{
		{
			args:   Arguments{"foo", "bar"},
			has:    []string{"foo", "bar"},
			expect: true,
		},
		{
			args:   Arguments{"foo", "bar"},
			has:    []string{"bar", "foo"},
			expect: true,
		},
		{
			args:   Arguments{"bar", "foo"},
			has:    []string{"foo"},
			expect: true,
		},
		{
			args:   Arguments{"foo", "bar"},
			has:    []string{"bar", "foo", "baz"},
			expect: false,
		},
		{
			args:   Arguments{"foo", "bar"},
			has:    []string{"foo"},
			expect: true,
		},
		{
			args:   Arguments{"foo", "bar"},
			has:    []string{"bar"},
			expect: true,
		},
		{
			args:   Arguments{"foo", "bar"},
			has:    []string{"baz"},
			expect: false,
		},
		{
			args:   Arguments{},
			has:    []string{"baz"},
			expect: false,
		},
	} {
		assert.Equal(t, c.expect, c.args.Has(c.has...), "%d", k)
	}
}

type matchesTestCase struct {
	args   Arguments
	is     []string
	expect bool
}

var matchesTests = []matchesTestCase{
	{
		args:   Arguments{},
		is:     []string{},
		expect: true,
	},
	{
		args:   Arguments{"foo", "bar"},
		is:     []string{"foo", "bar"},
		expect: true,
	},
	{
		args:   Arguments{"Foo", "Bar"},
		is:     []string{"Foo", "Bar"},
		expect: true,
	},
	{
		args:   Arguments{"foo", "foo"},
		is:     []string{"foo"},
		expect: false,
	},
	{
		args:   Arguments{"foo", "foo"},
		is:     []string{"bar", "foo"},
		expect: false,
	},
	{
		args:   Arguments{"foo", "bar"},
		is:     []string{"bar", "foo", "baz"},
		expect: false,
	},
	{
		args:   Arguments{"foo", "bar"},
		is:     []string{"foo"},
		expect: false,
	},
	{
		args:   Arguments{"foo", "bar"},
		is:     []string{"bar", "bar"},
		expect: false,
	},
	{
		args:   Arguments{"foo", "bar"},
		is:     []string{"baz"},
		expect: false,
	},
	{
		args:   Arguments{},
		is:     []string{"baz"},
		expect: false,
	},
}

func TestArgumentsMatchesExact(t *testing.T) {
	testCases := append(matchesTests, []matchesTestCase{ //nolint:gocritic
		// should fail if items are out of order
		{
			args:   Arguments{"foo", "bar"},
			is:     []string{"bar", "foo"},
			expect: false,
		},
		// should fail due to case-sensitivity.
		{
			args:   Arguments{"fOo", "bar"},
			is:     []string{"foo", "BaR"},
			expect: false,
		},
		// duplicate items should return allowed.
		{
			args:   Arguments{"foo", "foo"},
			is:     []string{"foo", "foo"},
			expect: true,
		},
	}...)
	for k, c := range testCases {
		assert.Equal(t, c.expect, c.args.MatchesExact(c.is...), "%d", k)
	}
}

func TestArgumentsMatches(t *testing.T) {
	testCases := append(matchesTests, []matchesTestCase{ //nolint:gocritic
		// should match if items are out of order.
		{
			args:   Arguments{"foo", "bar"},
			is:     []string{"bar", "foo"},
			expect: true,
		},
		// should not allow case-insensitive matching.
		{
			args:   Arguments{"fOo", "bar"},
			is:     []string{"foo", "BaR"},
			expect: false,
		},
		// should return non-matching if duplicate items exist.
		{
			args:   Arguments{"foo", "bar"},
			is:     []string{"FOO", "FOO", "bar"},
			expect: false,
		},
		{
			args:   Arguments{"foo", "foo"},
			is:     []string{"foo", "foo"},
			expect: false,
		},
	}...)
	for k, c := range testCases {
		assert.Equal(t, c.expect, c.args.Matches(c.is...), "%d", k)
	}
}

func TestArgumentsOneOf(t *testing.T) {
	for k, c := range []struct {
		args   Arguments
		oneOf  []string
		expect bool
	}{
		{
			args:   Arguments{"baz", "bar"},
			oneOf:  []string{"foo", "bar"},
			expect: true,
		},
		{
			args:   Arguments{"foo", "baz"},
			oneOf:  []string{"foo", "bar"},
			expect: true,
		},
		{
			args:   Arguments{"baz"},
			oneOf:  []string{"foo", "bar"},
			expect: false,
		},
	} {
		assert.Equal(t, c.expect, c.args.HasOneOf(c.oneOf...), "%d", k)
	}
}