assertionHelpers.go

// Copyright 2018-2019 Authors of Cilium
//
// 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 RuntimeTest

import (
	. "github.com/cilium/cilium/test/ginkgo-ext"
	"github.com/cilium/cilium/test/helpers"

	. "github.com/onsi/gomega"
)

// ExpectPolicyEnforcementUpdated sets PolicyEnforcement mode on the provided vm
// running Cilium to policyEnforcementType, and waits for endpoints on vm to
// be in 'ready' state. It asserts whether the configuration update was successful,
// and if the endpoints are ready.
func ExpectPolicyEnforcementUpdated(vm *helpers.SSHMeta, policyEnforcementType string) {
	By("Setting PolicyEnforcement=%s", policyEnforcementType)
	areEndpointsReady := vm.SetPolicyEnforcementAndWait(policyEnforcementType)
	ExpectWithOffset(1, areEndpointsReady).Should(BeTrue(), "unable to set PolicyEnforcement=%s", policyEnforcementType)
}

// ExpectEndpointSummary asserts whether the amount of endpoints managed by
// Cilium running on vm with PolicyEnforcement equal to policyEnforcementType
// is equal to numWithPolicyEnforcementType.
func ExpectEndpointSummary(vm *helpers.SSHMeta, policyEnforcementType string, numWithPolicyEnforcementType int) {
	endpoints, err := vm.PolicyEndpointsSummary()
	ExpectWithOffset(1, err).Should(BeNil(), "error getting endpoint summary")
	ExpectWithOffset(1, endpoints[policyEnforcementType]).To(Equal(numWithPolicyEnforcementType), "number of endpoints with %s=%s does not match", helpers.PolicyEnforcement, policyEnforcementType)
}

// ExpectCiliumReady asserts that cilium status is ready
func ExpectCiliumReady(vm *helpers.SSHMeta) {
	err := vm.WaitUntilReady(helpers.CiliumStartTimeout)
	ExpectWithOffset(1, err).To(BeNil(), "Cilium-agent cannot be started")

	vm.NetworkCreate(helpers.CiliumDockerNetwork, "")
	res := vm.NetworkGet(helpers.CiliumDockerNetwork)
	ExpectWithOffset(1, res).To(helpers.CMDSuccess(), "Cilium docker network is not created")

	res = vm.SetPolicyEnforcement(helpers.PolicyEnforcementDefault)
	ExpectWithOffset(1, res).To(helpers.CMDSuccess(), "Cannot set policy enforcement default")
}

// ExpectCiliumNotRunning checks that cilium has stopped with a grace period.
func ExpectCiliumNotRunning(vm *helpers.SSHMeta) {
	body := func() bool {
		res := vm.Exec("systemctl is-active cilium")
		return !res.WasSuccessful()
	}
	err := helpers.WithTimeout(body, "Cilium is still running after timeout", &helpers.TimeoutConfig{Timeout: helpers.CiliumStartTimeout})
	ExpectWithOffset(1, err).To(BeNil(), "Cilium is still running after timeout")
}

// ExpectDockerContainersMatchCiliumEndpoints asserts that docker containers in
// Cilium network match with the endpoint list
func ExpectDockerContainersMatchCiliumEndpoints(vm *helpers.SSHMeta) {
	ExpectWithOffset(1, vm.ValidateEndpointsAreCorrect(helpers.CiliumDockerNetwork)).To(BeNil(),
		"Docker containers mistmach with Cilium Endpoints")
}