#!/usr/bin/env bats
#
# Copyright (c) 2018 Intel Corporation
#
# SPDX-License-Identifier: Apache-2.0
#

load "${BATS_TEST_DIRNAME}/../../common.bash"
load "${BATS_TEST_DIRNAME}/tests_common.sh"

setup() {
	pod_name="memory-test"
	get_pod_config_dir
}

setup_yaml() {
	sed \
		-e "s/\${memory_size}/${memory_limit_size}/" \
		-e "s/\${memory_allocated}/${allocated_size}/" \
		"${pod_config_dir}/pod-memory-limit.yaml"
}


@test "Exceeding memory constraints" {
	memory_limit_size="50Mi"
	allocated_size="250M"

	# Create test .yaml
	test_yaml="${pod_config_dir}/test_exceed_memory.yaml"
	setup_yaml > "${test_yaml}"

	# Add policy to yaml file
	auto_generate_policy "${pod_config_dir}" "${test_yaml}"

	# Create the pod exceeding memory constraints
	run kubectl create -f "${test_yaml}"
	[ "$status" -ne 0 ]

	rm -f "${test_yaml}"
}

@test "Running within memory constraints" {
	[ "$(uname -m)" == "aarch64" ] && skip "See: https://github.com/kata-containers/kata-containers/issues/10926"

	memory_limit_size="600Mi"
	allocated_size="150M"

	# Create test .yaml
	test_yaml="${pod_config_dir}/test_within_memory.yaml"
	setup_yaml > "${test_yaml}"

	# Add policy to yaml file
	auto_generate_policy "${pod_config_dir}" "${test_yaml}"

	# Create the pod within memory constraints
	kubectl create -f "${test_yaml}"

	# Check pod creation
	kubectl wait --for=condition=Ready --timeout=$timeout pod "$pod_name"

	rm -f "${test_yaml}"
	kubectl delete pod "$pod_name"
}

teardown() {
	# Debugging information
	kubectl describe "pod/$pod_name" || true
}