// Copyright (c) 2017 Tigera, Inc. All rights reserved.

// 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 updateprocessors_test

import (
	"net"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	apiv3 "github.com/projectcalico/api/pkg/apis/projectcalico/v3"
	"github.com/projectcalico/api/pkg/lib/numorstring"

	"github.com/projectcalico/calico/lib/std/uniquelabels"
	libapiv3 "github.com/projectcalico/calico/libcalico-go/lib/apis/v3"
	"github.com/projectcalico/calico/libcalico-go/lib/backend/k8s/conversion"
	"github.com/projectcalico/calico/libcalico-go/lib/backend/model"
	"github.com/projectcalico/calico/libcalico-go/lib/backend/syncersv1/updateprocessors"
	cnet "github.com/projectcalico/calico/libcalico-go/lib/net"
)

var _ = Describe("Test the WorkloadEndpoint update processor", func() {
	hn1 := "host1"
	hn2 := "host2"
	oid1 := "orchestrator1"
	oid2 := "orchestrator2"
	wid1 := "workload1"
	wid2 := "workload2"
	eid1 := "endpoint1"
	eid2 := "endpoint2"
	ns1 := "namespace1"
	ns2 := "namespace2"
	name1 := "host1-orchestrator1-workload1-endpoint1"
	name2 := "host2-orchestrator2-workload2-endpoint2"
	iface1 := "iface1"
	iface2 := "iface2"

	v3WorkloadEndpointKey1 := model.ResourceKey{
		Kind:      libapiv3.KindWorkloadEndpoint,
		Name:      name1,
		Namespace: ns1,
	}
	v3WorkloadEndpointKey2 := model.ResourceKey{
		Kind:      libapiv3.KindWorkloadEndpoint,
		Name:      name2,
		Namespace: ns2,
	}
	v1WorkloadEndpointKey1 := model.WorkloadEndpointKey{
		Hostname:       hn1,
		OrchestratorID: oid1,
		WorkloadID:     ns1 + "/" + wid1,
		EndpointID:     eid1,
	}
	v1WorkloadEndpointKey2 := model.WorkloadEndpointKey{
		Hostname:       hn2,
		OrchestratorID: oid2,
		WorkloadID:     ns2 + "/" + wid2,
		EndpointID:     eid2,
	}

	It("should handle conversion of valid WorkloadEndpoints", func() {
		netmac2, err := net.ParseMAC("01:23:45:67:89:ab")
		Expect(err).NotTo(HaveOccurred())
		mac2 := cnet.MAC{HardwareAddr: netmac2}

		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		By("converting a WorkloadEndpoint with minimum configuration")
		res := libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns1
		res.Labels = map[string]string{
			"projectcalico.org/namespace":    ns1,
			"projectcalico.org/orchestrator": oid1,
		}
		res.Spec.Node = hn1
		res.Spec.Orchestrator = oid1
		res.Spec.Workload = wid1
		res.Spec.Endpoint = eid1
		res.Spec.InterfaceName = iface1
		res.Spec.IPNetworks = []string{"10.100.10.1"}

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())
		Expect(kvps).To(HaveLen(1))
		_, ipn, err := cnet.ParseCIDROrIP("10.100.10.1")
		Expect(err).NotTo(HaveOccurred())
		expectedIPv4Net := *(ipn.Network())
		Expect(kvps[0]).To(Equal(&model.KVPair{
			Key: v1WorkloadEndpointKey1,
			Value: &model.WorkloadEndpoint{
				State: "active",
				Name:  iface1,
				Ports: []model.EndpointPort{},
				Labels: uniquelabels.Make(map[string]string{
					"projectcalico.org/namespace":    ns1,
					"projectcalico.org/orchestrator": oid1,
				}),
				IPv4Nets: []cnet.IPNet{expectedIPv4Net},
			},
			Revision: "abcde",
		}))

		By("adding another WorkloadEndpoint with a full configuration")
		res = libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns2
		res.Labels = map[string]string{
			"testLabel":                      "label",
			"projectcalico.org/namespace":    ns2,
			"projectcalico.org/orchestrator": oid2,
		}
		res.Spec.Node = hn2
		res.Spec.Orchestrator = oid2
		res.Spec.Workload = wid2
		res.Spec.Endpoint = eid2
		res.Spec.InterfaceName = iface2
		res.Spec.ContainerID = "container2"
		res.Spec.MAC = "01:23:45:67:89:ab"
		res.Spec.Profiles = []string{"testProfile"}
		res.Spec.IPNetworks = []string{"10.100.10.1"}
		_, ipn, err = cnet.ParseCIDROrIP("10.100.10.1")
		Expect(err).NotTo(HaveOccurred())
		expectedIPv4Net = *(ipn.Network())
		res.Spec.IPNATs = []libapiv3.IPNAT{
			{
				InternalIP: "10.100.1.1",
				ExternalIP: "10.1.10.1",
			},
		}
		expectedIPv4NAT := *updateprocessors.ConvertV2ToV1IPNAT(res.Spec.IPNATs[0])
		res.Spec.IPv4Gateway = "10.10.10.1"
		expectedIPv4Gateway, _, err := cnet.ParseCIDROrIP("10.10.10.1")
		res.Spec.IPv6Gateway = "2001:0db8:85a3:0000:0000:8a2e:0370:7334"
		expectedIPv6Gateway, _, err := cnet.ParseCIDROrIP("2001:0db8:85a3:0000:0000:8a2e:0370:7334")
		res.Spec.Ports = []libapiv3.WorkloadEndpointPort{
			{
				Name:     "portname",
				Protocol: numorstring.ProtocolFromInt(uint8(30)),
				Port:     uint16(8080),
			},
		}
		res.Spec.AllowSpoofedSourcePrefixes = []string{"8.8.8.8/32"}

		kvps, err = up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey2,
			Value:    res,
			Revision: "1234",
		})
		Expect(err).NotTo(HaveOccurred())
		Expect(kvps).To(Equal([]*model.KVPair{
			{
				Key: v1WorkloadEndpointKey2,
				Value: &model.WorkloadEndpoint{
					State:      "active",
					Name:       iface2,
					Mac:        &mac2,
					ProfileIDs: []string{"testProfile"},
					IPv4Nets:   []cnet.IPNet{expectedIPv4Net},
					IPv4NAT:    []model.IPNAT{expectedIPv4NAT},
					Labels: uniquelabels.Make(map[string]string{
						"testLabel":                      "label",
						"projectcalico.org/namespace":    ns2,
						"projectcalico.org/orchestrator": oid2,
					}),
					IPv4Gateway: expectedIPv4Gateway,
					IPv6Gateway: expectedIPv6Gateway,
					Ports: []model.EndpointPort{
						{
							Name:     "portname",
							Protocol: numorstring.ProtocolFromInt(uint8(30)),
							Port:     uint16(8080),
						},
					},
					AllowSpoofedSourcePrefixes: []cnet.IPNet{cnet.MustParseCIDR("8.8.8.8/32")},
				},
				Revision: "1234",
			},
		}))

		By("deleting the first workload endpoint")
		kvps, err = up.Process(&model.KVPair{
			Key:   v3WorkloadEndpointKey1,
			Value: nil,
		})
		Expect(err).NotTo(HaveOccurred())
		Expect(kvps).To(Equal([]*model.KVPair{
			{
				Key:   v1WorkloadEndpointKey1,
				Value: nil,
			},
		}))
	})

	It("should fail to convert an invalid resource", func() {
		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		By("trying to convert with the wrong key type.")
		res := libapiv3.NewWorkloadEndpoint()
		res.Name = name1
		res.Namespace = ns1

		_, err := up.Process(&model.KVPair{
			Key: model.GlobalBGPPeerKey{
				PeerIP: cnet.MustParseIP("1.2.3.4"),
			},
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).To(HaveOccurred())

		By("trying to convert with the wrong value type and get a valid key with a nil value.")
		wres := apiv3.NewHostEndpoint()

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    wres,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())
		Expect(kvps).To(Equal([]*model.KVPair{
			{
				Key:   v1WorkloadEndpointKey1,
				Value: nil,
			},
		}))

		By("trying to convert without enough information to create a v1 key.")
		eres := libapiv3.NewWorkloadEndpoint()
		v3WorkloadEndpointKey1 := model.ResourceKey{
			Kind: libapiv3.KindWorkloadEndpoint,
			Name: name1,
		}

		_, err = up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    eres,
			Revision: "abcde",
		})
		Expect(err).To(HaveOccurred())
	})

	It("should filter out a WEP with no IPNetworks", func() {
		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		By("converting a WorkloadEndpoint with no IPNetworks")
		res := libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns1
		res.Labels = map[string]string{
			"projectcalico.org/namespace":    ns1,
			"projectcalico.org/orchestrator": oid1,
		}
		res.Spec.Node = hn1
		res.Spec.Orchestrator = oid1
		res.Spec.Workload = wid1
		res.Spec.Endpoint = eid1
		res.Spec.InterfaceName = iface1

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())
		Expect(kvps).To(HaveLen(1))
		Expect(kvps[0]).To(Equal(&model.KVPair{
			Key:   v1WorkloadEndpointKey1,
			Value: nil,
		}))
	})

	It("should filter out a WEP with bad AllowSpoofedSourcePrefixes", func() {
		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		By("converting a WorkloadEndpoint with bad AllowSpoofedSourcePrefixes")
		res := libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns1
		res.Labels = map[string]string{
			"projectcalico.org/namespace":    ns1,
			"projectcalico.org/orchestrator": oid1,
		}
		res.Spec.Node = hn1
		res.Spec.Orchestrator = oid1
		res.Spec.Workload = wid1
		res.Spec.Endpoint = eid1
		res.Spec.InterfaceName = iface1
		res.Spec.IPNetworks = []string{"10.100.10.1"}

		// Include an invalid prefix - an empty string, in this case.
		res.Spec.AllowSpoofedSourcePrefixes = []string{""}

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())

		// The update processor treats invalid values as a delete, so
		// we should expect a single update with a nil value.
		Expect(kvps).To(HaveLen(1))
		Expect(kvps[0]).To(Equal(&model.KVPair{
			Key:   v1WorkloadEndpointKey1,
			Value: nil,
		}))
	})

	It("should filter out a WEP with namespace or serviceAccount labels", func() {
		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		nsLabel := conversion.NamespaceLabelPrefix + "ns1"
		saLabel := conversion.ServiceAccountLabelPrefix + "sa1"
		res := libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns1
		res.Labels = map[string]string{
			"projectcalico.org/namespace":    ns1,
			"projectcalico.org/orchestrator": oid1,
			nsLabel:                          "ns1",
			saLabel:                          "sa1",
			"k1":                             "v1",
		}
		res.Spec.Node = hn1
		res.Spec.Orchestrator = oid1
		res.Spec.Workload = wid1
		res.Spec.Endpoint = eid1
		res.Spec.InterfaceName = iface1
		res.Spec.IPNetworks = []string{"10.100.10.1"}

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())
		_, ipn, err := cnet.ParseCIDROrIP("10.100.10.1")
		expectedIPv4Net := *(ipn.Network())
		Expect(kvps).To(HaveLen(1))
		Expect(kvps[0]).To(Equal(&model.KVPair{
			Key: v1WorkloadEndpointKey1,
			Value: &model.WorkloadEndpoint{
				State: "active",
				Name:  iface1,
				Ports: []model.EndpointPort{},
				Labels: uniquelabels.Make(map[string]string{
					"projectcalico.org/namespace":    ns1,
					"projectcalico.org/orchestrator": oid1,
					"k1":                             "v1",
				}),
				IPv4Nets: []cnet.IPNet{expectedIPv4Net},
			},
			Revision: "abcde",
		}))
	})

	It("should add a label representing the serviceaccount name", func() {
		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		res := libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns1
		res.Labels = map[string]string{
			"projectcalico.org/namespace":    ns1,
			"projectcalico.org/orchestrator": oid1,
			"k1":                             "v1",
		}
		res.Spec.Node = hn1
		res.Spec.Orchestrator = oid1
		res.Spec.Workload = wid1
		res.Spec.Endpoint = eid1
		res.Spec.InterfaceName = iface1
		res.Spec.IPNetworks = []string{"10.100.10.1"}
		res.Spec.ServiceAccountName = "test-serviceaccount-name"

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())
		_, ipn, err := cnet.ParseCIDROrIP("10.100.10.1")
		expectedIPv4Net := *(ipn.Network())
		Expect(kvps).To(HaveLen(1))
		Expect(kvps[0]).To(Equal(&model.KVPair{
			Key: v1WorkloadEndpointKey1,
			Value: &model.WorkloadEndpoint{
				State: "active",
				Name:  iface1,
				Ports: []model.EndpointPort{},
				Labels: uniquelabels.Make(map[string]string{
					"projectcalico.org/namespace":      ns1,
					"projectcalico.org/orchestrator":   oid1,
					"k1":                               "v1",
					"projectcalico.org/serviceaccount": "test-serviceaccount-name",
				}),
				IPv4Nets: []cnet.IPNet{expectedIPv4Net},
			},
			Revision: "abcde",
		}))
	})

	It("should convert a WEP with QoSControls configured", func() {
		up := updateprocessors.NewWorkloadEndpointUpdateProcessor()

		res := libapiv3.NewWorkloadEndpoint()
		res.Namespace = ns1
		res.Labels = map[string]string{
			"projectcalico.org/namespace":    ns1,
			"projectcalico.org/orchestrator": oid1,
		}
		res.Spec.Node = hn1
		res.Spec.Orchestrator = oid1
		res.Spec.Workload = wid1
		res.Spec.Endpoint = eid1
		res.Spec.InterfaceName = iface1
		res.Spec.IPNetworks = []string{"10.100.10.1"}
		res.Spec.QoSControls = &libapiv3.QoSControls{
			IngressBandwidth:      1000000,
			EgressBandwidth:       2000000,
			IngressBurst:          3000000,
			EgressBurst:           4000000,
			IngressPeakrate:       5000000,
			EgressPeakrate:        6000000,
			IngressMinburst:       7000000,
			EgressMinburst:        8000000,
			IngressPacketRate:     9000000,
			EgressPacketRate:      10000000,
			IngressPacketBurst:    11000000,
			EgressPacketBurst:     12000000,
			IngressMaxConnections: 13000000,
			EgressMaxConnections:  14000000,
		}

		kvps, err := up.Process(&model.KVPair{
			Key:      v3WorkloadEndpointKey1,
			Value:    res,
			Revision: "abcde",
		})
		Expect(err).NotTo(HaveOccurred())
		Expect(kvps).To(HaveLen(1))
		_, ipn, err := cnet.ParseCIDROrIP("10.100.10.1")
		Expect(err).NotTo(HaveOccurred())
		expectedIPv4Net := *(ipn.Network())
		Expect(kvps[0]).To(Equal(&model.KVPair{
			Key: v1WorkloadEndpointKey1,
			Value: &model.WorkloadEndpoint{
				State: "active",
				Name:  iface1,
				Ports: []model.EndpointPort{},
				Labels: uniquelabels.Make(map[string]string{
					"projectcalico.org/namespace":    ns1,
					"projectcalico.org/orchestrator": oid1,
				}),
				IPv4Nets: []cnet.IPNet{expectedIPv4Net},
				QoSControls: &model.QoSControls{
					IngressBandwidth:      1000000,
					EgressBandwidth:       2000000,
					IngressBurst:          3000000,
					EgressBurst:           4000000,
					IngressPeakrate:       5000000,
					EgressPeakrate:        6000000,
					IngressMinburst:       7000000,
					EgressMinburst:        8000000,
					IngressPacketRate:     9000000,
					EgressPacketRate:      10000000,
					IngressPacketBurst:    11000000,
					EgressPacketBurst:     12000000,
					IngressMaxConnections: 13000000,
					EgressMaxConnections:  14000000,
				},
			},
			Revision: "abcde",
		}))
	})
})
