Something went wrong. Try again.
keyboard stuff
Something went wrong. Try again.
7.8 kB · 260 lines
C++
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261/* 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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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));
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(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());
VERIFY_AND_CLEAR(driver);}