equalKeys_test.go

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

package oauth2_test

import (
	"testing"

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

	"authelia.com/provider/oauth2/internal/reflection"
)

func TestAssertObjectsAreEqualByKeys(t *testing.T) {
	type foo struct {
		Name string
		Body int
	}
	a := &foo{"foo", 1}
	b := &foo{"bar", 1}
	c := &foo{"baz", 3}

	AssertObjectKeysEqual(t, a, a, "Name", "Body")
	AssertObjectKeysNotEqual(t, a, b, "Name")
	AssertObjectKeysNotEqual(t, a, c, "Name", "Body")
}

func AssertObjectKeysEqual(t *testing.T, a, b any, keys ...string) {
	assert.True(t, len(keys) > 0, "No key provided.")
	for _, k := range keys {
		c, err := reflection.GetField(a, k)
		assert.NoError(t, err)
		d, err := reflection.GetField(b, k)
		assert.NoError(t, err)
		assert.Equal(t, c, d, "field: %s", k)
	}
}

func AssertObjectKeysNotEqual(t *testing.T, a, b any, keys ...string) {
	assert.True(t, len(keys) > 0, "No key provided.")
	for _, k := range keys {
		c, err := reflection.GetField(a, k)
		assert.NoError(t, err)
		d, err := reflection.GetField(b, k)
		assert.NoError(t, err)
		assert.NotEqual(t, c, d, "%s", k)
	}
}

func RequireObjectKeysEqual(t *testing.T, a, b any, keys ...string) {
	assert.True(t, len(keys) > 0, "No key provided.")
	for _, k := range keys {
		c, err := reflection.GetField(a, k)
		assert.NoError(t, err)
		d, err := reflection.GetField(b, k)
		assert.NoError(t, err)
		require.Equal(t, c, d, "%s", k)
	}
}

func RequireObjectKeysNotEqual(t *testing.T, a, b any, keys ...string) {
	assert.True(t, len(keys) > 0, "No key provided.")
	for _, k := range keys {
		c, err := reflection.GetField(a, k)
		assert.NoError(t, err)
		d, err := reflection.GetField(b, k)
		assert.NoError(t, err)
		require.NotEqual(t, c, d, "%s", k)
	}
}