~aleteoryx/muditaos

muditaos/module-bluetooth/tests/tests-BTKeysStorage.cpp -rw-r--r-- 4.1 KiB
a405cad6 — Aleteoryx trim readme 6 days ago 
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

// Copyright (c) 2017-2024, Mudita Sp. z.o.o. All rights reserved.
// For licensing, see https://github.com/mudita/MuditaOS/blob/master/LICENSE.md

#include <catch2/catch.hpp>
#include <service-bluetooth/SettingsHolder.hpp>
#include <BtKeysStorage.hpp>

namespace bluetooth
{
    class TestSettingsHolder : public SettingsHolder
    {
      public:
        std::string entryValue;
        Settings settingType;
        auto getValue(const Settings setting) -> SettingEntry override
        {
            if (setting == settingType) {
                return entryValue;
            }
            else {
                return std::string{};
            }
        }
        void setValue(const Settings &newSetting, const SettingEntry &value) override
        {
            settingType = newSetting;
            entryValue  = std::get<std::string>(value);
        }
    };

} // namespace bluetooth

using namespace bluetooth;

TEST_CASE("BT Keys storage")
{
    KeyStorage storage;
    auto settingHolder = std::make_shared<TestSettingsHolder>();

    storage.settings = settingHolder;

    auto btKeyStorage = storage.getKeyStorage();
    SECTION("put key to the storage")
    {
        bd_addr_t addr;
        sscanf_bd_addr("94:B2:CC:F1:92:B1", addr);
        link_key_t key       = "12345";
        link_key_type_t type = COMBINATION_KEY;
        btKeyStorage->put_link_key(addr, key, type);

        REQUIRE(settingHolder->settingType == Settings::BtKeys);
        REQUIRE(settingHolder->entryValue ==
                "{\"keys\": [{\"bd_addr\": \"94:B2:CC:F1:92:B1\", \"link_key\": \"12345\", \"type\": 0}]}");
    }
    SECTION("get key from the storage")
    {
        bd_addr_t addr;
        sscanf_bd_addr("94:B2:CC:F1:92:B1", addr);
        link_key_t key;
        link_key_type_t type;
        settingHolder->entryValue =
            "{\"keys\": [{\"bd_addr\": \"94:B2:CC:F1:92:B1\", \"link_key\": \"12345\", \"type\": 0}]}";

        btKeyStorage->open();
        auto ret = btKeyStorage->get_link_key(addr, key, &type);

        REQUIRE(std::string(reinterpret_cast<char *>(key)) == "12345");
        REQUIRE(type == COMBINATION_KEY);
        REQUIRE(ret == 1);
    }
    SECTION("get key from the storage when there's no key")
    {
        bd_addr_t addr;
        sscanf_bd_addr("94:B2:CC:F1:92:B1", addr);
        link_key_t key;
        link_key_type_t type;
        settingHolder->entryValue = "";
        btKeyStorage->open();
        btKeyStorage->delete_link_key(addr);
        auto ret = btKeyStorage->get_link_key(addr, key, &type);

        REQUIRE(ret != 1);
    }
    SECTION("put key to the storage few times")
    {
        bd_addr_t addr;
        sscanf_bd_addr("94:B2:CC:F1:92:B1", addr);
        link_key_t key       = "12345";
        link_key_type_t type = COMBINATION_KEY;

        btKeyStorage->open();
        btKeyStorage->put_link_key(addr, key, type);
        btKeyStorage->put_link_key(addr, key, type);
        btKeyStorage->put_link_key(addr, key, type);

        REQUIRE(settingHolder->settingType == Settings::BtKeys);
        REQUIRE(settingHolder->entryValue ==
                "{\"keys\": [{\"bd_addr\": \"94:B2:CC:F1:92:B1\", \"link_key\": \"12345\", \"type\": 0}]}");
    }
    SECTION("delete key from the storage")
    {
        bd_addr_t addr;
        sscanf_bd_addr("94:B2:CC:F1:92:B1", addr);
        settingHolder->entryValue =
            "{\"keys\": [{\"bd_addr\": \"94:B2:CC:F1:92:B1\", \"link_key\": \"12345\", \"type\": 0}]}";

        btKeyStorage->open();
        btKeyStorage->delete_link_key(addr);

        REQUIRE(settingHolder->settingType == Settings::BtKeys);
        REQUIRE(settingHolder->entryValue == "{\"keys\": []}");
    }
    SECTION("double-delete key from the storage")
    {
        bd_addr_t addr;
        sscanf_bd_addr("94:B2:CC:F1:92:B1", addr);
        settingHolder->entryValue =
            "{\"keys\": [{\"bd_addr\": \"94:B2:CC:F1:92:B1\", \"link_key\": \"12345\", \"type\": 0}]}";

        btKeyStorage->open();
        btKeyStorage->delete_link_key(addr);
        btKeyStorage->delete_link_key(addr);

        REQUIRE(settingHolder->settingType == Settings::BtKeys);
        REQUIRE(settingHolder->entryValue == "{\"keys\": []}");
    }
}