1 /*
2  * Copyright 2022 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include <gtest/gtest.h>
18 
19 #include "model/controller/link_layer_controller.h"
20 #include "test_helpers.h"
21 
22 namespace rootcanal {
23 
24 using namespace bluetooth::hci;
25 
26 class LeSetPeriodicAdvertisingEnableTest : public ::testing::Test {
27 public:
LeSetPeriodicAdvertisingEnableTest()28   LeSetPeriodicAdvertisingEnableTest() {}
29   ~LeSetPeriodicAdvertisingEnableTest() override = default;
30 
31 protected:
32   Address address_{0};
33   ControllerProperties properties_{};
34   LinkLayerController controller_{address_, properties_};
35 };
36 
TEST_F(LeSetPeriodicAdvertisingEnableTest,Enable)37 TEST_F(LeSetPeriodicAdvertisingEnableTest, Enable) {
38   ASSERT_EQ(controller_.LeSetExtendedAdvertisingParameters(
39                     0, MakeAdvertisingEventProperties(), 0x0800, 0x0800, 0x7,
40                     OwnAddressType::PUBLIC_DEVICE_ADDRESS,
41                     PeerAddressType::PUBLIC_DEVICE_OR_IDENTITY_ADDRESS, Address::kEmpty,
42                     AdvertisingFilterPolicy::ALL_DEVICES, 0x70, PrimaryPhyType::LE_1M, 0,
43                     SecondaryPhyType::LE_2M, 0x0, false),
44             ErrorCode::SUCCESS);
45 
46   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingParameters(0, 0x6, 0xffff, false),
47             ErrorCode::SUCCESS);
48 
49   std::vector<uint8_t> advertising_data = {1, 2, 3};
50   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingData(0, Operation::COMPLETE_ADVERTISEMENT,
51                                                      advertising_data),
52             ErrorCode::SUCCESS);
53 
54   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(true, false, 0), ErrorCode::SUCCESS);
55 }
56 
TEST_F(LeSetPeriodicAdvertisingEnableTest,Disable)57 TEST_F(LeSetPeriodicAdvertisingEnableTest, Disable) {
58   ASSERT_EQ(controller_.LeSetExtendedAdvertisingParameters(
59                     0, MakeAdvertisingEventProperties(), 0x0800, 0x0800, 0x7,
60                     OwnAddressType::PUBLIC_DEVICE_ADDRESS,
61                     PeerAddressType::PUBLIC_DEVICE_OR_IDENTITY_ADDRESS, Address::kEmpty,
62                     AdvertisingFilterPolicy::ALL_DEVICES, 0x70, PrimaryPhyType::LE_1M, 0,
63                     SecondaryPhyType::LE_2M, 0x0, false),
64             ErrorCode::SUCCESS);
65 
66   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingParameters(0, 0x6, 0xffff, false),
67             ErrorCode::SUCCESS);
68 
69   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(false, false, 0), ErrorCode::SUCCESS);
70 }
71 
TEST_F(LeSetPeriodicAdvertisingEnableTest,UnknownAdvertisingHandle)72 TEST_F(LeSetPeriodicAdvertisingEnableTest, UnknownAdvertisingHandle) {
73   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(true, false, 0),
74             ErrorCode::UNKNOWN_ADVERTISING_IDENTIFIER);
75 
76   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(false, false, 0),
77             ErrorCode::UNKNOWN_ADVERTISING_IDENTIFIER);
78 }
79 
TEST_F(LeSetPeriodicAdvertisingEnableTest,PartialAdvertisingData)80 TEST_F(LeSetPeriodicAdvertisingEnableTest, PartialAdvertisingData) {
81   ASSERT_EQ(controller_.LeSetExtendedAdvertisingParameters(
82                     0, MakeAdvertisingEventProperties(), 0x0800, 0x0800, 0x7,
83                     OwnAddressType::PUBLIC_DEVICE_ADDRESS,
84                     PeerAddressType::PUBLIC_DEVICE_OR_IDENTITY_ADDRESS, Address::kEmpty,
85                     AdvertisingFilterPolicy::ALL_DEVICES, 0x70, PrimaryPhyType::LE_1M, 0,
86                     SecondaryPhyType::LE_2M, 0x0, false),
87             ErrorCode::SUCCESS);
88 
89   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingParameters(0, 0x6, 0xffff, false),
90             ErrorCode::SUCCESS);
91 
92   std::vector<uint8_t> first_advertising_data_fragment = {1, 2, 3};
93   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingData(0, Operation::FIRST_FRAGMENT,
94                                                      first_advertising_data_fragment),
95             ErrorCode::SUCCESS);
96 
97   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(true, false, 0),
98             ErrorCode::COMMAND_DISALLOWED);
99 }
100 
TEST_F(LeSetPeriodicAdvertisingEnableTest,PeriodicAdvertisingNotConfigured)101 TEST_F(LeSetPeriodicAdvertisingEnableTest, PeriodicAdvertisingNotConfigured) {
102   ASSERT_EQ(controller_.LeSetExtendedAdvertisingParameters(
103                     0, MakeAdvertisingEventProperties(), 0x0800, 0x0800, 0x7,
104                     OwnAddressType::PUBLIC_DEVICE_ADDRESS,
105                     PeerAddressType::PUBLIC_DEVICE_OR_IDENTITY_ADDRESS, Address::kEmpty,
106                     AdvertisingFilterPolicy::ALL_DEVICES, 0x70, PrimaryPhyType::LE_1M, 0,
107                     SecondaryPhyType::LE_2M, 0x0, false),
108             ErrorCode::SUCCESS);
109 
110   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(true, false, 0),
111             ErrorCode::COMMAND_DISALLOWED);
112 }
113 
TEST_F(LeSetPeriodicAdvertisingEnableTest,InvalidAdvertisingEventProperties)114 TEST_F(LeSetPeriodicAdvertisingEnableTest, InvalidAdvertisingEventProperties) {
115   ASSERT_EQ(controller_.LeSetExtendedAdvertisingParameters(
116                     0, MakeAdvertisingEventProperties(), 0x0800, 0x0800, 0x7,
117                     OwnAddressType::PUBLIC_DEVICE_ADDRESS,
118                     PeerAddressType::PUBLIC_DEVICE_OR_IDENTITY_ADDRESS, Address::kEmpty,
119                     AdvertisingFilterPolicy::ALL_DEVICES, 0x70, PrimaryPhyType::LE_1M, 0,
120                     SecondaryPhyType::LE_2M, 0x0, false),
121             ErrorCode::SUCCESS);
122 
123   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingParameters(0, 0x6, 0xffff, false),
124             ErrorCode::SUCCESS);
125 
126   ASSERT_EQ(controller_.LeSetExtendedAdvertisingParameters(
127                     0, MakeAdvertisingEventProperties(CONNECTABLE), 0x0800, 0x0800, 0x7,
128                     OwnAddressType::PUBLIC_DEVICE_ADDRESS,
129                     PeerAddressType::PUBLIC_DEVICE_OR_IDENTITY_ADDRESS, Address::kEmpty,
130                     AdvertisingFilterPolicy::ALL_DEVICES, 0x70, PrimaryPhyType::LE_1M, 0,
131                     SecondaryPhyType::LE_2M, 0x0, false),
132             ErrorCode::SUCCESS);
133 
134   ASSERT_EQ(controller_.LeSetPeriodicAdvertisingEnable(true, false, 0),
135             ErrorCode::COMMAND_DISALLOWED);
136 }
137 
138 }  // namespace rootcanal
139