mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
78 lines
3.2 KiB
C++
78 lines
3.2 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 <string>
|
|
#include <tuple>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "wpi/driverstation/Joystick.hpp"
|
|
#include "wpi/driverstation/internal/DriverStationBackend.hpp"
|
|
#include "wpi/simulation/DriverStationSim.hpp"
|
|
#include "wpi/simulation/SimHooks.hpp"
|
|
|
|
class IsJoystickConnectedParametersTest
|
|
: public ::testing::TestWithParam<std::tuple<int, int, int, bool>> {};
|
|
|
|
TEST_P(IsJoystickConnectedParametersTest, IsJoystickConnected) {
|
|
wpi::sim::DriverStationSim::SetJoystickAxesMaximumIndex(
|
|
1, std::get<0>(GetParam()));
|
|
wpi::sim::DriverStationSim::SetJoystickButtonsMaximumIndex(
|
|
1, std::get<1>(GetParam()));
|
|
wpi::sim::DriverStationSim::SetJoystickPOVsMaximumIndex(
|
|
1, std::get<2>(GetParam()));
|
|
wpi::sim::DriverStationSim::NotifyNewData();
|
|
|
|
ASSERT_EQ(std::get<3>(GetParam()),
|
|
wpi::internal::DriverStationBackend::IsJoystickConnected(1));
|
|
}
|
|
|
|
INSTANTIATE_TEST_SUITE_P(IsConnectedTests, IsJoystickConnectedParametersTest,
|
|
::testing::Values(std::make_tuple(0, 0, 0, false),
|
|
std::make_tuple(1, 0, 0, true),
|
|
std::make_tuple(0, 1, 0, true),
|
|
std::make_tuple(0, 0, 1, true),
|
|
std::make_tuple(1, 1, 1, true),
|
|
std::make_tuple(4, 10, 1, true)));
|
|
class JoystickConnectionWarningTest
|
|
: public ::testing::TestWithParam<
|
|
std::tuple<bool, bool, bool, std::string>> {};
|
|
|
|
TEST_P(JoystickConnectionWarningTest, JoystickConnectionWarnings) {
|
|
// Capture all output to stderr.
|
|
::testing::internal::CaptureStderr();
|
|
|
|
// Set FMS and Silence settings
|
|
wpi::sim::DriverStationSim::SetFmsAttached(std::get<0>(GetParam()));
|
|
wpi::sim::DriverStationSim::NotifyNewData();
|
|
wpi::internal::DriverStationBackend::SilenceJoystickConnectionWarning(
|
|
std::get<1>(GetParam()));
|
|
|
|
// Create joystick and attempt to retrieve button.
|
|
wpi::Joystick joystick(0);
|
|
joystick.GetRawButton(1);
|
|
|
|
wpi::sim::StepTiming(1_s);
|
|
EXPECT_EQ(wpi::internal::DriverStationBackend::
|
|
IsJoystickConnectionWarningSilenced(),
|
|
std::get<2>(GetParam()));
|
|
EXPECT_EQ(::testing::internal::GetCapturedStderr().substr(
|
|
0, std::get<3>(GetParam()).size()),
|
|
std::get<3>(GetParam()));
|
|
}
|
|
|
|
INSTANTIATE_TEST_SUITE_P(
|
|
DriverStationTests, JoystickConnectionWarningTest,
|
|
::testing::Values(
|
|
std::make_tuple(false, true, true, ""),
|
|
std::make_tuple(false, false, false,
|
|
"Warning: Joystick on port 0 not available, check if "
|
|
"all controllers are plugged in\n"),
|
|
std::make_tuple(true, true, false,
|
|
"Warning: Joystick on port 0 not available, check if "
|
|
"all controllers are plugged in\n"),
|
|
std::make_tuple(true, false, false,
|
|
"Warning: Joystick on port 0 not available, check if "
|
|
"all controllers are plugged in\n")));
|