package object

import (
	"testing"
	"time"

	assertapi "github.com/stretchr/testify/assert"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	"edge-infra.dev/pkg/k8s/object/fobject"
)

func TestGetStatusObservedGeneration(t *testing.T) {
	assert := assertapi.New(t)

	obj := &fobject.Fake{}
	_, err := GetStatusObservedGeneration(obj)
	if assert.Error(err) {
		assert.ErrorIs(err, ErrObservedGenerationNotFound)
	}

	// Get set status observedGeneration.
	obj.Status.ObservedGeneration = 7
	og, err := GetStatusObservedGeneration(obj)
	assert.NoError(err)
	assert.Equal(int64(7), og)
}

func TestGetRequeueInterval(t *testing.T) {
	assert := assertapi.New(t)

	obj := &fobject.Fake{}
	pd, err := GetRequeueInterval(obj)
	assert.NoError(err)
	assert.Equal(time.Duration(0), pd)

	// Get set requeue interval value.
	obj.Spec.Interval = metav1.Duration{Duration: 3 * time.Second}
	pd, err = GetRequeueInterval(obj)
	assert.NoError(err)
	assert.Equal(3*time.Second, pd)

	// Get non-existent requeue interval value.
	obj2 := &corev1.Secret{}
	_, err = GetRequeueInterval(obj2)
	if assert.Error(err) {
		assert.ErrorIs(err, ErrRequeueIntervalNotFound)
	}
}