Files
allwpilib/hal/src/test/native/cpp/mockdata/PWMDataTest.cpp
2026-03-13 15:53:24 -07:00

85 lines
2.7 KiB
C++

// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.
#include "wpi/hal/simulation/PWMData.h"
#include <string>
#include <gtest/gtest.h>
#include "wpi/hal/Errors.h"
#include "wpi/hal/HALBase.h"
#include "wpi/hal/PWM.h"
#include "wpi/hal/handles/HandlesInternal.hpp"
namespace wpi::hal {
std::string gTestPwmCallbackName;
HAL_Value gTestPwmCallbackValue;
void TestPwmInitializationCallback(const char* name, void* param,
const struct HAL_Value* value) {
gTestPwmCallbackName = name;
gTestPwmCallbackValue = *value;
}
TEST(PWMSimTest, PwmInitialization) {
const int INDEX_TO_TEST = 7;
int callbackParam = 0;
int callbackId = HALSIM_RegisterPWMInitializedCallback(
INDEX_TO_TEST, &TestPwmInitializationCallback, &callbackParam, false);
ASSERT_TRUE(0 != callbackId);
int32_t status = 0;
int32_t channel = 0;
HAL_DigitalHandle pwmHandle;
// Use out of range index
channel = 8000;
gTestPwmCallbackName = "Unset";
pwmHandle = HAL_InitializePWMPort(channel, nullptr, &status);
EXPECT_EQ(HAL_kInvalidHandle, pwmHandle);
EXPECT_EQ(HAL_USE_LAST_ERROR, status);
HAL_GetLastError(&status);
EXPECT_EQ(RESOURCE_OUT_OF_RANGE, status);
EXPECT_STREQ("Unset", gTestPwmCallbackName.c_str());
// Successful setup
status = 0;
channel = INDEX_TO_TEST;
gTestPwmCallbackName = "Unset";
pwmHandle = HAL_InitializePWMPort(channel, nullptr, &status);
EXPECT_TRUE(HAL_kInvalidHandle != pwmHandle);
EXPECT_EQ(0, status);
EXPECT_STREQ("Initialized", gTestPwmCallbackName.c_str());
// Double initialize... should fail
status = 0;
channel = INDEX_TO_TEST;
gTestPwmCallbackName = "Unset";
pwmHandle = HAL_InitializePWMPort(channel, nullptr, &status);
EXPECT_EQ(HAL_kInvalidHandle, pwmHandle);
EXPECT_EQ(HAL_USE_LAST_ERROR, status);
HAL_GetLastError(&status);
EXPECT_EQ(RESOURCE_IS_ALLOCATED, status);
EXPECT_STREQ("Unset", gTestPwmCallbackName.c_str());
// Reset, should allow you to re-register
wpi::hal::HandleBase::ResetGlobalHandles();
HALSIM_ResetPWMData(INDEX_TO_TEST);
callbackId = HALSIM_RegisterPWMInitializedCallback(
INDEX_TO_TEST, &TestPwmInitializationCallback, &callbackParam, false);
status = 0;
channel = INDEX_TO_TEST;
gTestPwmCallbackName = "Unset";
pwmHandle = HAL_InitializePWMPort(channel, nullptr, &status);
EXPECT_TRUE(HAL_kInvalidHandle != pwmHandle);
EXPECT_EQ(0, status);
EXPECT_STREQ("Initialized", gTestPwmCallbackName.c_str());
HALSIM_CancelPWMInitializedCallback(INDEX_TO_TEST, callbackId);
}
} // namespace wpi::hal