libc/test/src/sched/sched_cpuequal_test.cpp

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
///
/// \file
/// Unittests for __sched_cpuequal.
///
//===----------------------------------------------------------------------===//

#include "hdr/types/cpu_set_t.h"
#include "src/sched/sched_cpuequal.h"
#include "src/sched/sched_setcpuset.h"
#include "src/sched/sched_setcpuzero.h"
#include "test/UnitTest/Test.h"

using LlvmLibcSchedCpuEqualTest = LIBC_NAMESPACE::testing::Test;

TEST_F(LlvmLibcSchedCpuEqualTest, CpuEqual) {
  cpu_set_t mask1, mask2;
  const size_t alloc_size = sizeof(cpu_set_t);

  LIBC_NAMESPACE::__sched_setcpuzero(alloc_size, &mask1);
  LIBC_NAMESPACE::__sched_setcpuzero(alloc_size, &mask2);
  ASSERT_EQ(LIBC_NAMESPACE::__sched_cpuequal(alloc_size, &mask1, &mask2), 1);

  LIBC_NAMESPACE::__sched_setcpuset(1, alloc_size, &mask1);
  ASSERT_EQ(LIBC_NAMESPACE::__sched_cpuequal(alloc_size, &mask1, &mask2), 0);
}