1 // Copyright 2021 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14
15 #include "pw_i2c/device.h"
16
17 #include <chrono>
18
19 #include "gtest/gtest.h"
20 #include "pw_bytes/byte_builder.h"
21 #include "pw_containers/algorithm.h"
22 #include "pw_i2c/initiator_mock.h"
23
24 using namespace std::literals::chrono_literals;
25
26 namespace pw::i2c {
27 namespace {
28
TEST(Device,WriteReadForOk)29 TEST(Device, WriteReadForOk) {
30 constexpr Address kTestDeviceAddress = Address::SevenBit<0x3F>();
31
32 constexpr auto kExpectWrite1 = bytes::Array<1, 2, 3>();
33 constexpr auto kExpectRead1 = bytes::Array<1, 2>();
34
35 auto expected_transactions = MakeExpectedTransactionArray({Transaction(
36 OkStatus(), kTestDeviceAddress, kExpectWrite1, kExpectRead1, 2ms)});
37
38 MockInitiator initiator(expected_transactions);
39
40 Device device = Device(initiator, kTestDeviceAddress);
41
42 std::array<std::byte, kExpectRead1.size()> read1;
43 EXPECT_EQ(device.WriteReadFor(kExpectWrite1, read1, 2ms), OkStatus());
44 EXPECT_TRUE(pw::containers::Equal(read1, kExpectRead1));
45 EXPECT_EQ(initiator.Finalize(), OkStatus());
46 }
47
TEST(Device,WriteForOk)48 TEST(Device, WriteForOk) {
49 constexpr Address kTestDeviceAddress = Address::SevenBit<0x3F>();
50
51 constexpr auto kExpectWrite1 = bytes::Array<1, 2, 3>();
52
53 auto expected_transactions = MakeExpectedTransactionArray(
54 {WriteTransaction(OkStatus(), kTestDeviceAddress, kExpectWrite1, 2ms)});
55
56 MockInitiator initiator(expected_transactions);
57 Device device = Device(initiator, kTestDeviceAddress);
58
59 EXPECT_EQ(device.WriteFor(kExpectWrite1, 2ms), OkStatus());
60 EXPECT_EQ(initiator.Finalize(), OkStatus());
61 }
62
TEST(Device,ReadForOk)63 TEST(Device, ReadForOk) {
64 constexpr Address kTestDeviceAddress = Address::SevenBit<0x3F>();
65
66 constexpr auto kExpectRead1 = bytes::Array<1, 2, 3>();
67
68 auto expected_transactions = MakeExpectedTransactionArray(
69 {ReadTransaction(OkStatus(), kTestDeviceAddress, kExpectRead1, 2ms)});
70
71 MockInitiator initiator(expected_transactions);
72 Device device = Device(initiator, kTestDeviceAddress);
73
74 std::array<std::byte, kExpectRead1.size()> read1;
75 EXPECT_EQ(device.ReadFor(read1, 2ms), OkStatus());
76 EXPECT_EQ(initiator.Finalize(), OkStatus());
77 }
78
TEST(Device,ProbeDeviceForOk)79 TEST(Device, ProbeDeviceForOk) {
80 constexpr Address kTestDeviceAddress = Address::SevenBit<0x3F>();
81
82 auto expected_transactions = MakeExpectedTransactionArray(
83 {ProbeTransaction(OkStatus(), kTestDeviceAddress, 2ms)});
84
85 MockInitiator initiator(expected_transactions);
86 Device device = Device(initiator, kTestDeviceAddress);
87
88 EXPECT_EQ(initiator.ProbeDeviceFor(kTestDeviceAddress, 2ms), OkStatus());
89 EXPECT_EQ(initiator.Finalize(), OkStatus());
90 }
91
92 } // namespace
93 } // namespace pw::i2c
94