forked from mirrors/qmk_firmware
95d20e6d8b
* Fix Caps Word and Unicode Map
* Tests for Caps Word + Auto Shift and Unicode Map.
* Fix formatting
* Add additional keyboard report expectation macros
This commit defines five test utilities, EXPECT_REPORT, EXPECT_UNICODE,
EXPECT_EMPTY_REPORT, EXPECT_ANY_REPORT and EXPECT_NO_REPORT for use with
TestDriver.
EXPECT_REPORT sets a gmock expectation that a given keyboard report will
be sent. For instance,
EXPECT_REPORT(driver, (KC_LSFT, KC_A));
is shorthand for
EXPECT_CALL(driver,
send_keyboard_mock(KeyboardReport(KC_LSFT, KC_A)));
EXPECT_UNICODE sets a gmock expectation that a given Unicode code point
will be sent using UC_LNX input mode. For instance for U+2013,
EXPECT_UNICODE(driver, 0x2013);
expects the sequence of keys:
"Ctrl+Shift+U, 2, 0, 1, 3, space".
EXPECT_EMPTY_REPORT sets a gmock expectation that a given keyboard
report will be sent. For instance
EXPECT_EMPTY_REPORT(driver);
expects a single report without keypresses or modifiers.
EXPECT_ANY_REPORT sets a gmock expectation that a arbitrary keyboard
report will be sent, without matching its contents. For instance
EXPECT_ANY_REPORT(driver).Times(1);
expects a single arbitrary keyboard report will be sent.
EXPECT_NO_REPORT sets a gmock expectation that no keyboard report will
be sent at all.
* Add tap_key() and tap_keys() to TestFixture.
This commit adds a `tap_key(key)` method to TestFixture that taps a
given KeymapKey, optionally with a specified delay between press and
release.
Similarly, the method `tap_keys(key_a, key_b, key_c)` taps a sequence of
KeymapKeys.
* Use EXPECT_REPORT, tap_keys, etc. in most tests.
This commit uses EXPECT_REPORT, EXPECT_UNICODE, EXPECT_EMPTY_REPORT,
EXPECT_NO_REPORT, tap_key() and tap_keys() test utilities from the
previous two commits in most tests. Particularly the EXPECT_REPORT
macro is frequently useful and makes a nice reduction in boilerplate
needed to express many tests.
Co-authored-by: David Kosorin <david@kosorin.net>
260 lines
7.9 KiB
C++
260 lines
7.9 KiB
C++
/* Copyright 2021 Stefan Kerkmann
|
|
*
|
|
* 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 "keyboard_report_util.hpp"
|
|
#include "test_common.hpp"
|
|
|
|
using testing::_;
|
|
using testing::AnyNumber;
|
|
using testing::InSequence;
|
|
|
|
class Secure : public TestFixture {
|
|
public:
|
|
void SetUp() override {
|
|
secure_lock();
|
|
}
|
|
};
|
|
|
|
TEST_F(Secure, test_lock) {
|
|
TestDriver driver;
|
|
|
|
// Don't allow empty reports.
|
|
EXPECT_NO_REPORT(driver);
|
|
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
secure_unlock();
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
run_one_scan_loop();
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
secure_lock();
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_timeout) {
|
|
TestDriver driver;
|
|
|
|
// Don't allow empty reports.
|
|
EXPECT_NO_REPORT(driver);
|
|
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
secure_unlock();
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
idle_for(SECURE_IDLE_TIMEOUT + 1);
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request) {
|
|
TestDriver driver;
|
|
auto key_mo = KeymapKey(0, 0, 0, MO(1));
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_mo, key_a, key_b, key_c, key_d});
|
|
|
|
// Don't allow empty reports.
|
|
EXPECT_NO_REPORT(driver);
|
|
|
|
EXPECT_TRUE(secure_is_locked());
|
|
secure_request_unlock();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_a, key_b, key_c, key_d);
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_fail) {
|
|
TestDriver driver;
|
|
auto key_e = KeymapKey(0, 0, 0, KC_E);
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_e, key_a, key_b, key_c, key_d});
|
|
|
|
// Allow any number of empty reports.
|
|
EXPECT_EMPTY_REPORT(driver).Times(AnyNumber());
|
|
{ // Expect the following reports in this order.
|
|
InSequence s;
|
|
EXPECT_REPORT(driver, (KC_A));
|
|
EXPECT_REPORT(driver, (KC_B));
|
|
EXPECT_REPORT(driver, (KC_C));
|
|
EXPECT_REPORT(driver, (KC_D));
|
|
}
|
|
EXPECT_TRUE(secure_is_locked());
|
|
secure_request_unlock();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_e, key_a, key_b, key_c, key_d);
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_timeout) {
|
|
TestDriver driver;
|
|
|
|
// Don't allow empty reports.
|
|
EXPECT_NO_REPORT(driver);
|
|
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
secure_request_unlock();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
idle_for(SECURE_UNLOCK_TIMEOUT + 1);
|
|
EXPECT_FALSE(secure_is_unlocking());
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_fail_mid) {
|
|
TestDriver driver;
|
|
auto key_e = KeymapKey(0, 0, 0, KC_E);
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_e, key_a, key_b, key_c, key_d});
|
|
|
|
// Allow any number of empty reports.
|
|
EXPECT_EMPTY_REPORT(driver).Times(AnyNumber());
|
|
{ // Expect the following reports in this order.
|
|
InSequence s;
|
|
EXPECT_REPORT(driver, (KC_C));
|
|
EXPECT_REPORT(driver, (KC_D));
|
|
}
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
secure_request_unlock();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_a, key_b, key_e, key_c, key_d);
|
|
EXPECT_FALSE(secure_is_unlocking());
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_fail_out_of_order) {
|
|
TestDriver driver;
|
|
auto key_e = KeymapKey(0, 0, 0, KC_E);
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_e, key_a, key_b, key_c, key_d});
|
|
|
|
// Allow any number of empty reports.
|
|
EXPECT_EMPTY_REPORT(driver).Times(AnyNumber());
|
|
{ // Expect the following reports in this order.
|
|
InSequence s;
|
|
EXPECT_REPORT(driver, (KC_B));
|
|
EXPECT_REPORT(driver, (KC_C));
|
|
}
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
secure_request_unlock();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_a, key_d, key_b, key_c);
|
|
EXPECT_TRUE(secure_is_locked());
|
|
EXPECT_FALSE(secure_is_unlocking());
|
|
EXPECT_FALSE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_on_layer) {
|
|
TestDriver driver;
|
|
auto key_mo = KeymapKey(0, 0, 0, MO(1));
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_mo, key_a, key_b, key_c, key_d});
|
|
|
|
// Don't allow empty reports.
|
|
EXPECT_NO_REPORT(driver);
|
|
|
|
EXPECT_TRUE(secure_is_locked());
|
|
key_mo.press();
|
|
run_one_scan_loop();
|
|
secure_request_unlock();
|
|
key_mo.release();
|
|
run_one_scan_loop();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_a, key_b, key_c, key_d);
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
EXPECT_FALSE(layer_state_is(1));
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_mid_stroke) {
|
|
TestDriver driver;
|
|
auto key_e = KeymapKey(0, 0, 0, KC_E);
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_e, key_a, key_b, key_c, key_d});
|
|
|
|
EXPECT_REPORT(driver, (KC_E));
|
|
EXPECT_EMPTY_REPORT(driver);
|
|
EXPECT_TRUE(secure_is_locked());
|
|
key_e.press();
|
|
run_one_scan_loop();
|
|
secure_request_unlock();
|
|
key_e.release();
|
|
run_one_scan_loop();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_a, key_b, key_c, key_d);
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|
|
|
|
TEST_F(Secure, test_unlock_request_mods) {
|
|
TestDriver driver;
|
|
auto key_lsft = KeymapKey(0, 0, 0, KC_LSFT);
|
|
auto key_a = KeymapKey(0, 1, 0, KC_A);
|
|
auto key_b = KeymapKey(0, 2, 0, KC_B);
|
|
auto key_c = KeymapKey(0, 3, 0, KC_C);
|
|
auto key_d = KeymapKey(0, 4, 0, KC_D);
|
|
|
|
set_keymap({key_lsft, key_a, key_b, key_c, key_d});
|
|
|
|
EXPECT_REPORT(driver, (key_lsft.report_code));
|
|
EXPECT_EMPTY_REPORT(driver);
|
|
EXPECT_TRUE(secure_is_locked());
|
|
key_lsft.press();
|
|
run_one_scan_loop();
|
|
secure_request_unlock();
|
|
key_lsft.release();
|
|
run_one_scan_loop();
|
|
EXPECT_TRUE(secure_is_unlocking());
|
|
tap_keys(key_a, key_b, key_c, key_d);
|
|
EXPECT_TRUE(secure_is_unlocked());
|
|
|
|
testing::Mock::VerifyAndClearExpectations(&driver);
|
|
}
|