#!/usr/bin/python3 # -*- coding: utf-8 -*- # key-mapper - GUI for device specific keyboard mappings # Copyright (C) 2020 sezanzeb # # This file is part of key-mapper. # # key-mapper 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 3 of the License, or # (at your option) any later version. # # key-mapper 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 key-mapper. If not, see . import os import shutil import json import unittest from evdev.ecodes import EV_KEY, EV_ABS, ABS_HAT0X from keymapper.mapping import Mapping from keymapper.state import SystemMapping from keymapper.config import config from keymapper.paths import get_preset_path from tests.test import tmp, cleanup class TestSystemMapping(unittest.TestCase): def tearDown(self): cleanup() def test_update(self): system_mapping = SystemMapping() system_mapping.update({ 'foo1': 101, 'bar1': 102 }) system_mapping.update({ 'foo2': 201, 'bar2': 202 }) self.assertEqual(system_mapping.get('foo1'), 101) self.assertEqual(system_mapping.get('bar2'), 202) def test_system_mapping(self): system_mapping = SystemMapping() self.assertGreater(len(system_mapping._mapping), 100) self.assertEqual(system_mapping.get('1'), 2) self.assertEqual(system_mapping.get('KeY_1'), 2) self.assertEqual(system_mapping.get('AlT_L'), 56) self.assertEqual(system_mapping.get('KEy_LEFtALT'), 56) self.assertEqual(system_mapping.get('kEY_LeFTSHIFT'), 42) self.assertEqual(system_mapping.get('ShiFt_L'), 42) self.assertEqual(system_mapping.get('BTN_left'), 272) self.assertIsNotNone(system_mapping.get('kp_1')) self.assertIsNotNone(system_mapping.get('KP_1')) self.assertEqual( system_mapping.get('KP_Left'), system_mapping.get('KP_4') ) self.assertEqual( system_mapping.get('KP_Left'), system_mapping.get('KEY_KP4') ) names = system_mapping.list_names() self.assertIn('key_kp1', names) self.assertIn('key_nextsong', names) self.assertIn('2', names) self.assertIn('key_3', names) self.assertIn('c', names) self.assertIn('key_d', names) self.assertIn('f4', names) self.assertIn('key_f5', names) self.assertIn('minus', names) self.assertIn('btn_left', names) self.assertIn('btn_right', names) class TestMapping(unittest.TestCase): def setUp(self): self.mapping = Mapping() self.assertFalse(self.mapping.changed) def tearDown(self): cleanup() def test_config(self): self.mapping.save(get_preset_path('foo', 'bar2')) self.assertEqual(self.mapping.get('a'), None) self.mapping.set('a', 1) self.assertEqual(self.mapping.get('a'), 1) self.mapping.remove('a') self.mapping.set('a.b', 2) self.assertEqual(self.mapping.get('a.b'), 2) self.assertEqual(self.mapping._config['a']['b'], 2) self.mapping.remove('a.b') self.mapping.set('a.b.c', 3) self.assertEqual(self.mapping.get('a.b.c'), 3) self.assertEqual(self.mapping._config['a']['b']['c'], 3) # setting mapping.whatever does not overwrite the mapping # after saving. It should be ignored. self.mapping.change((EV_KEY, 81, 1), 'a') self.mapping.set('mapping.a', 2) self.mapping.save(get_preset_path('foo', 'bar')) self.mapping.load(get_preset_path('foo', 'bar')) self.assertEqual(self.mapping.get_character((EV_KEY, 81, 1)), 'a') self.assertIsNone(self.mapping.get('mapping.a')) # loading a different preset also removes the configs from memory self.mapping.set('a.b.c', 6) self.mapping.load(get_preset_path('foo', 'bar2')) self.assertIsNone(self.mapping.get('a.b.c')) def test_fallback(self): config.set('d.e.f', 5) self.assertEqual(self.mapping.get('d.e.f'), 5) self.mapping.set('d.e.f', 3) self.assertEqual(self.mapping.get('d.e.f'), 3) def test_clone(self): ev_1 = (EV_KEY, 1, 1) ev_2 = (EV_KEY, 2, 0) mapping1 = Mapping() mapping1.change(ev_1, 'a') mapping2 = mapping1.clone() mapping1.change(ev_2, 'b') self.assertEqual(mapping1.get_character(ev_1), 'a') self.assertEqual(mapping1.get_character(ev_2), 'b') self.assertEqual(mapping2.get_character(ev_1), 'a') self.assertIsNone(mapping2.get_character(ev_2)) self.assertIsNone(mapping2.get_character((EV_KEY, 2, 3))) self.assertIsNone(mapping2.get_character((EV_KEY, 1, 3))) def test_save_load(self): one = (EV_KEY, 10, 1) two = (EV_KEY, 11, 1) three = (EV_KEY, 12, 1) self.mapping.change(one, '1') self.mapping.change(two, '2') self.mapping.change(three, '3') self.mapping._config['foo'] = 'bar' self.mapping.save(get_preset_path('device 1', 'test')) path = os.path.join(tmp, 'presets', 'device 1', 'test.json') self.assertTrue(os.path.exists(path)) loaded = Mapping() self.assertEqual(len(loaded), 0) loaded.load(get_preset_path('device 1', 'test')) self.assertEqual(len(loaded), 3) self.assertEqual(loaded.get_character(one), '1') self.assertEqual(loaded.get_character(two), '2') self.assertEqual(loaded.get_character(three), '3') self.assertEqual(loaded._config['foo'], 'bar') def test_save_load_2(self): # loads mappings with only (type, code) as the key path = os.path.join(tmp, 'presets', 'device 1', 'test.json') os.makedirs(os.path.dirname(path), exist_ok=True) with open(path, 'w') as file: json.dump({ 'mapping': { f'{EV_KEY},3': 'a', f'{EV_ABS},{ABS_HAT0X},-1': 'b', f'{EV_ABS},{ABS_HAT0X},1': 'c', } }, file) loaded = Mapping() loaded.load(get_preset_path('device 1', 'test')) self.assertEqual(loaded.get_character((EV_KEY, 3, 1)), 'a') self.assertEqual(loaded.get_character((EV_ABS, ABS_HAT0X, -1)), 'b') self.assertEqual(loaded.get_character((EV_ABS, ABS_HAT0X, 1)), 'c') def test_change(self): ev_1 = (EV_KEY, 1, 111) ev_2 = (EV_KEY, 1, 222) ev_3 = (EV_KEY, 2, 111) ev_4 = (EV_ABS, 1, 111) # 1 is not assigned yet, ignore it self.mapping.change(ev_1, 'a', ev_2) self.assertTrue(self.mapping.changed) self.assertIsNone(self.mapping.get_character(ev_2)) self.assertEqual(self.mapping.get_character(ev_1), 'a') self.assertEqual(len(self.mapping), 1) # change ev_1 to ev_3 and change a to b self.mapping.change(ev_3, 'b', ev_1) self.assertIsNone(self.mapping.get_character(ev_1)) self.assertEqual(self.mapping.get_character(ev_3), 'b') self.assertEqual(len(self.mapping), 1) # add 4 self.mapping.change(ev_4, 'c', None) self.assertEqual(self.mapping.get_character(ev_3), 'b') self.assertEqual(self.mapping.get_character(ev_4), 'c') self.assertEqual(len(self.mapping), 2) # change the mapping of 4 to d self.mapping.change(ev_4, 'd', None) self.assertEqual(self.mapping.get_character(ev_4), 'd') self.assertEqual(len(self.mapping), 2) # this also works in the same way self.mapping.change(ev_4, 'e', ev_4) self.assertEqual(self.mapping.get_character(ev_4), 'e') self.assertEqual(len(self.mapping), 2) def test_clear(self): # does nothing self.mapping.clear((EV_KEY, 40, 1)) self.assertFalse(self.mapping.changed) self.assertEqual(len(self.mapping), 0) self.mapping._mapping[(EV_KEY, 40, 1)] = 'b' self.assertEqual(len(self.mapping), 1) self.mapping.clear((EV_KEY, 40, 1)) self.assertEqual(len(self.mapping), 0) self.assertTrue(self.mapping.changed) self.mapping.change((EV_KEY, 10, 1), 'KP_1', None) self.assertTrue(self.mapping.changed) self.mapping.change((EV_KEY, 20, 1), 'KP_2', None) self.mapping.change((EV_KEY, 30, 1), 'KP_3', None) self.assertEqual(len(self.mapping), 3) self.mapping.clear((EV_KEY, 20, 1)) self.assertEqual(len(self.mapping), 2) self.assertEqual(self.mapping.get_character((EV_KEY, 10, 1)), 'KP_1') self.assertIsNone(self.mapping.get_character((EV_KEY, 20, 1))) self.assertEqual(self.mapping.get_character((EV_KEY, 30, 1)), 'KP_3') def test_empty(self): self.mapping.change((EV_KEY, 10, 1), '1') self.mapping.change((EV_KEY, 11, 1), '2') self.mapping.change((EV_KEY, 12, 1), '3') self.assertEqual(len(self.mapping), 3) self.mapping.empty() self.assertEqual(len(self.mapping), 0) if __name__ == "__main__": unittest.main()