mirror of
https://github.com/firewalkwithm3/qmk_firmware.git
synced 2024-11-23 03:50:28 +08:00
145 lines
3.8 KiB
C++
145 lines
3.8 KiB
C++
/* Copyright 2021 Balz Guenat
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "gmock/gmock.h"
|
|
#include <vector>
|
|
#include <algorithm>
|
|
#include <stdio.h>
|
|
|
|
extern "C" {
|
|
#include "encoder.h"
|
|
#include "encoder/tests/mock.h"
|
|
}
|
|
|
|
struct update {
|
|
int8_t index;
|
|
bool clockwise;
|
|
};
|
|
|
|
uint8_t updates_array_idx = 0;
|
|
update updates[32];
|
|
|
|
bool encoder_update_kb(uint8_t index, bool clockwise) {
|
|
updates[updates_array_idx % 32] = {index, clockwise};
|
|
updates_array_idx++;
|
|
return true;
|
|
}
|
|
|
|
bool setAndRead(pin_t pin, bool val) {
|
|
setPin(pin, val);
|
|
return encoder_task();
|
|
}
|
|
|
|
class EncoderTest : public ::testing::Test {};
|
|
|
|
TEST_F(EncoderTest, TestInit) {
|
|
updates_array_idx = 0;
|
|
encoder_init();
|
|
EXPECT_EQ(pinIsInputHigh[0], true);
|
|
EXPECT_EQ(pinIsInputHigh[1], true);
|
|
EXPECT_EQ(updates_array_idx, 0);
|
|
}
|
|
|
|
TEST_F(EncoderTest, TestOneClockwise) {
|
|
updates_array_idx = 0;
|
|
encoder_init();
|
|
// send 4 pulses. with resolution 4, that's one step and we should get 1 update.
|
|
setAndRead(0, false);
|
|
setAndRead(1, false);
|
|
setAndRead(0, true);
|
|
setAndRead(1, true);
|
|
|
|
EXPECT_EQ(updates_array_idx, 1);
|
|
EXPECT_EQ(updates[0].index, 0);
|
|
EXPECT_EQ(updates[0].clockwise, true);
|
|
}
|
|
|
|
TEST_F(EncoderTest, TestOneCounterClockwise) {
|
|
updates_array_idx = 0;
|
|
encoder_init();
|
|
setAndRead(1, false);
|
|
setAndRead(0, false);
|
|
setAndRead(1, true);
|
|
setAndRead(0, true);
|
|
|
|
EXPECT_EQ(updates_array_idx, 1);
|
|
EXPECT_EQ(updates[0].index, 0);
|
|
EXPECT_EQ(updates[0].clockwise, false);
|
|
}
|
|
|
|
TEST_F(EncoderTest, TestTwoClockwiseOneCC) {
|
|
updates_array_idx = 0;
|
|
encoder_init();
|
|
setAndRead(0, false);
|
|
setAndRead(1, false);
|
|
setAndRead(0, true);
|
|
setAndRead(1, true);
|
|
setAndRead(0, false);
|
|
setAndRead(1, false);
|
|
setAndRead(0, true);
|
|
setAndRead(1, true);
|
|
setAndRead(1, false);
|
|
setAndRead(0, false);
|
|
setAndRead(1, true);
|
|
setAndRead(0, true);
|
|
|
|
EXPECT_EQ(updates_array_idx, 3);
|
|
EXPECT_EQ(updates[0].index, 0);
|
|
EXPECT_EQ(updates[0].clockwise, true);
|
|
EXPECT_EQ(updates[1].index, 0);
|
|
EXPECT_EQ(updates[1].clockwise, true);
|
|
EXPECT_EQ(updates[2].index, 0);
|
|
EXPECT_EQ(updates[2].clockwise, false);
|
|
}
|
|
|
|
TEST_F(EncoderTest, TestNoEarly) {
|
|
updates_array_idx = 0;
|
|
encoder_init();
|
|
// send 3 pulses. with resolution 4, that's not enough for a step.
|
|
setAndRead(0, false);
|
|
setAndRead(1, false);
|
|
setAndRead(0, true);
|
|
EXPECT_EQ(updates_array_idx, 0);
|
|
// now send last pulse
|
|
setAndRead(1, true);
|
|
EXPECT_EQ(updates_array_idx, 1);
|
|
EXPECT_EQ(updates[0].index, 0);
|
|
EXPECT_EQ(updates[0].clockwise, true);
|
|
}
|
|
|
|
TEST_F(EncoderTest, TestHalfway) {
|
|
updates_array_idx = 0;
|
|
encoder_init();
|
|
// go halfway
|
|
setAndRead(0, false);
|
|
setAndRead(1, false);
|
|
EXPECT_EQ(updates_array_idx, 0);
|
|
// back off
|
|
setAndRead(1, true);
|
|
setAndRead(0, true);
|
|
EXPECT_EQ(updates_array_idx, 0);
|
|
// go all the way
|
|
setAndRead(0, false);
|
|
setAndRead(1, false);
|
|
setAndRead(0, true);
|
|
setAndRead(1, true);
|
|
// should result in 1 update
|
|
EXPECT_EQ(updates_array_idx, 1);
|
|
EXPECT_EQ(updates[0].index, 0);
|
|
EXPECT_EQ(updates[0].clockwise, true);
|
|
}
|