// Copyright (c) 2019 SAP SE or an SAP affiliate company. All rights reserved. This file is licensed under the Apache Software License, v. 2 except as noted otherwise in the LICENSE file
//
// 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 operation_test

import (
	"context"

	gardencorev1alpha1 "github.com/gardener/gardener/pkg/apis/core/v1alpha1"
	gardencorev1beta1 "github.com/gardener/gardener/pkg/apis/core/v1beta1"
	mockclient "github.com/gardener/gardener/pkg/mock/controller-runtime/client"
	mock "github.com/gardener/gardener/pkg/mock/gardener/client/kubernetes"
	. "github.com/gardener/gardener/pkg/operation"
	kutil "github.com/gardener/gardener/pkg/utils/kubernetes"
	"github.com/golang/mock/gomock"
	"sigs.k8s.io/controller-runtime/pkg/client"

	operationseed "github.com/gardener/gardener/pkg/operation/seed"
	operationshoot "github.com/gardener/gardener/pkg/operation/shoot"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/ginkgo/extensions/table"
	. "github.com/onsi/gomega"
	"github.com/onsi/gomega/types"
	apierrors "k8s.io/apimachinery/pkg/api/errors"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

var _ = Describe("operation", func() {
	DescribeTable("#ComputeIngressHost", func(prefix, shootName, projectName, domain string, matcher types.GomegaMatcher) {
		var (
			seed = &gardencorev1beta1.Seed{
				Spec: gardencorev1beta1.SeedSpec{
					DNS: gardencorev1beta1.SeedDNS{
						IngressDomain: domain,
					},
				},
			}
			shoot = &gardencorev1beta1.Shoot{
				ObjectMeta: metav1.ObjectMeta{
					Name: shootName,
				},
			}
			o = &Operation{
				Seed: &operationseed.Seed{
					Info: seed,
				},
				Shoot: &operationshoot.Shoot{
					Info: shoot,
				},
			}
		)

		shoot.Status = gardencorev1beta1.ShootStatus{
			TechnicalID: operationshoot.ComputeTechnicalID(projectName, shoot),
		}

		Expect(o.ComputeIngressHost(prefix)).To(matcher)
	},
		Entry("ingress calculation",
			"t",
			"fooShoot",
			"barProject",
			"ingress.seed.example.com",
			Equal("t-barProject--fooShoot.ingress.seed.example.com"),
		),
	)

	Describe("#EnsureShootStateExists", func() {
		var (
			shoot                  *gardencorev1beta1.Shoot
			ctrl                   *gomock.Controller
			gardenClient           *mock.MockInterface
			k8sGardenRuntimeClient *mockclient.MockClient
			o                      *Operation
		)

		BeforeEach(func() {
			shoot = &gardencorev1beta1.Shoot{
				ObjectMeta: metav1.ObjectMeta{
					Name:      "foo",
					Namespace: "bar",
				},
			}

			ctrl = gomock.NewController(GinkgoT())
			gardenClient = mock.NewMockInterface(ctrl)
			k8sGardenRuntimeClient = mockclient.NewMockClient(ctrl)

			o = &Operation{
				K8sGardenClient: gardenClient,
				Shoot: &operationshoot.Shoot{
					Info: shoot,
				},
			}
		})

		It("should retrieve ShouldState and add it to the Operation struct", func() {
			gomock.InOrder(
				gardenClient.EXPECT().Client().Return(k8sGardenRuntimeClient),
				k8sGardenRuntimeClient.EXPECT().Get(context.TODO(), kutil.Key("bar", "foo"), gomock.AssignableToTypeOf(&gardencorev1alpha1.ShootState{})).Return(nil),
				k8sGardenRuntimeClient.EXPECT().Update(context.TODO(), gomock.AssignableToTypeOf(&gardencorev1alpha1.ShootState{})).Return(nil),
			)

			Expect(o.EnsureShootStateExists(context.TODO())).To(Succeed())

			Expect(o.ShootState).ToNot(BeNil())
		})

		It("should create ShootState with correct ownerReferences and add it to the Operation struct", func() {
			gomock.InOrder(
				gardenClient.EXPECT().Client().Return(k8sGardenRuntimeClient),
				k8sGardenRuntimeClient.EXPECT().Get(context.TODO(), kutil.Key("bar", "foo"), gomock.AssignableToTypeOf(&gardencorev1alpha1.ShootState{})).DoAndReturn(
					func(_ context.Context, _ client.ObjectKey, csr *gardencorev1alpha1.ShootState) error {
						return apierrors.NewNotFound(gardencorev1alpha1.Resource("shootstate"), "foo")
					}),
				k8sGardenRuntimeClient.EXPECT().Create(context.TODO(), gomock.AssignableToTypeOf(&gardencorev1alpha1.ShootState{})).Return(nil),
			)

			Expect(o.EnsureShootStateExists(context.TODO())).To(Succeed())

			Expect(o.ShootState).ToNot(BeNil())
			Expect(len(o.ShootState.OwnerReferences)).To(Equal(1))
			Expect(o.ShootState.OwnerReferences[0].Name).To(Equal("foo"))
			Expect(o.ShootState.OwnerReferences[0].Kind).To(Equal("Shoot"))
			Expect(o.ShootState.OwnerReferences[0].BlockOwnerDeletion).ToNot(BeNil())
			Expect(*o.ShootState.OwnerReferences[0].BlockOwnerDeletion).To(BeFalse())
			Expect(o.ShootState.OwnerReferences[0].Controller).ToNot(BeNil())
			Expect(*o.ShootState.OwnerReferences[0].Controller).To(BeTrue())
		})
	})
})