/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Macros for defining a KVM test
 *
 * Copyright (C) 2022, Google LLC.
 */

#ifndef SELFTEST_KVM_TEST_HARNESS_H
#define SELFTEST_KVM_TEST_HARNESS_H

#include "kselftest_harness.h"

#define KVM_ONE_VCPU_TEST_SUITE(name)					\
	FIXTURE(name) {							\
		struct kvm_vcpu *vcpu;					\
	};								\
									\
	FIXTURE_SETUP(name) {						\
		(void)vm_create_with_one_vcpu(&self->vcpu, NULL);	\
	}								\
									\
	FIXTURE_TEARDOWN(name) {					\
		kvm_vm_free(self->vcpu->vm);				\
	}

#define KVM_ONE_VCPU_TEST(suite, test, guestcode)			\
static void __suite##_##test(struct kvm_vcpu *vcpu);			\
									\
TEST_F(suite, test)							\
{									\
	vcpu_arch_set_entry_point(self->vcpu, guestcode);		\
	__suite##_##test(self->vcpu);					\
}									\
static void __suite##_##test(struct kvm_vcpu *vcpu)

#endif /* SELFTEST_KVM_TEST_HARNESS_H */