/*
Copyright 2024.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package authorizationmodelrequest

import (
	fgainternal "fga-operator/internal/openfga"
	"fmt"
	"github.com/golang/mock/gomock"
	"k8s.io/client-go/tools/record"
	"k8s.io/utils/clock"
	"path/filepath"
	"runtime"
	"testing"
	"time"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"

	"k8s.io/client-go/kubernetes/scheme"
	"k8s.io/client-go/rest"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/envtest"
	logf "sigs.k8s.io/controller-runtime/pkg/log"
	"sigs.k8s.io/controller-runtime/pkg/log/zap"

	extensionsv1 "fga-operator/api/v1"
	//+kubebuilder:scaffold:imports
)

// These tests use Ginkgo (BDD-style Go testing framework). Refer to
// http://onsi.github.io/ginkgo/ to learn more about Ginkgo.

var (
	cfg                      *rest.Config
	k8sClient                client.Client
	testEnv                  *envtest.Environment
	controllerReconciler     *AuthorizationModelRequestReconciler
	goMockController         *gomock.Controller
	permissionServiceFactory fgainternal.PermissionServiceFactory
)

const (
	resourceName  = "test-resource"
	namespaceName = "default"
)

func TestControllers(t *testing.T) {
	RegisterFailHandler(Fail)

	RunSpecs(t, "Controller Suite")
}

var _ = BeforeSuite(func() {
	logf.SetLogger(zap.New(zap.WriteTo(GinkgoWriter), zap.UseDevMode(true)))

	By("bootstrapping test environment")
	testEnv = &envtest.Environment{
		CRDDirectoryPaths:     []string{filepath.Join("..", "..", "..", "config", "crd", "bases")},
		ErrorIfCRDPathMissing: true,

		// The BinaryAssetsDirectory is only required if you want to run the tests directly
		// without call the makefile target test. If not informed it will look for the
		// default path defined in controller-runtime which is /usr/local/kubebuilder/.
		// Note that you must have the required binaries setup under the bin directory to perform
		// the tests directly. When we run make test it will be setup and used automatically.
		BinaryAssetsDirectory: filepath.Join("..", "..", "..", "bin", "k8s",
			fmt.Sprintf("1.29.0-%s-%s", runtime.GOOS, runtime.GOARCH)),
	}

	var err error
	// cfg is defined in this file globally.
	cfg, err = testEnv.Start()
	Expect(err).NotTo(HaveOccurred())
	Expect(cfg).NotTo(BeNil())

	err = extensionsv1.AddToScheme(scheme.Scheme)
	Expect(err).NotTo(HaveOccurred())

	//+kubebuilder:scaffold:scheme

	k8sClient, err = client.New(cfg, client.Options{Scheme: scheme.Scheme})
	Expect(err).NotTo(HaveOccurred())
	Expect(k8sClient).NotTo(BeNil())

	goMockController = gomock.NewController(GinkgoT())
	permissionServiceFactory = setupMockFactory()

	controllerReconciler = &AuthorizationModelRequestReconciler{
		Client:                   k8sClient,
		Scheme:                   k8sClient.Scheme(),
		Recorder:                 record.NewFakeRecorder(20),
		Clock:                    clock.RealClock{},
		PermissionServiceFactory: permissionServiceFactory,
	}
})

var _ = AfterSuite(func() {
	defer goMockController.Finish()
	By("tearing down the test environment")
	err := testEnv.Stop()
	Expect(err).NotTo(HaveOccurred())
})

func setupMockFactory() fgainternal.PermissionServiceFactory {
	mockFactory := fgainternal.NewMockPermissionServiceFactory(goMockController)
	mockService := fgainternal.NewMockPermissionService(goMockController)
	store := fgainternal.Store{
		Id:        "foo",
		Name:      resourceName,
		CreatedAt: time.Now(),
	}
	authModelId := "123"

	mockFactory.EXPECT().GetService(gomock.Any()).Return(mockService, nil).AnyTimes()
	mockService.EXPECT().CheckExistingStoresByName(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()
	mockService.EXPECT().CreateStore(gomock.Any(), gomock.Any(), gomock.Any()).Return(&store, nil).AnyTimes()
	mockService.EXPECT().SetStoreId(gomock.Any()).AnyTimes()
	mockService.EXPECT().
		CreateAuthorizationModel(gomock.Any(), gomock.Any(), gomock.Any()).
		Return(authModelId, nil).
		AnyTimes()
	return mockFactory
}
