1//
2// Copyright (C) 2015 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 "shill/arp_client_test_helper.h"
18
19#include <gtest/gtest.h>
20
21using testing::_;
22using testing::Invoke;
23using testing::Return;
24
25namespace shill {
26
27ArpClientTestHelper::ArpClientTestHelper(MockArpClient* client)
28    : client_(client) {}
29
30ArpClientTestHelper::~ArpClientTestHelper() {}
31
32void ArpClientTestHelper::GeneratePacket(uint16_t operation,
33                                         const IPAddress& local_ip,
34                                         const ByteString& local_mac,
35                                         const IPAddress& remote_ip,
36                                         const ByteString& remote_mac) {
37  packet_.set_operation(operation);
38  packet_.set_local_ip_address(local_ip);
39  packet_.set_local_mac_address(local_mac);
40  packet_.set_remote_ip_address(remote_ip);
41  packet_.set_remote_mac_address(remote_mac);
42
43  EXPECT_CALL(*client_, ReceivePacket(_, _))
44      .WillOnce(Invoke(this, &ArpClientTestHelper::SimulateReceivePacket));
45}
46
47bool ArpClientTestHelper::SimulateReceivePacket(ArpPacket* packet,
48                                                ByteString* sender) {
49  packet->set_operation(packet_.operation());
50  packet->set_local_ip_address(packet_.local_ip_address());
51  packet->set_local_mac_address(packet_.local_mac_address());
52  packet->set_remote_ip_address(packet_.remote_ip_address());
53  packet->set_remote_mac_address(packet_.remote_mac_address());
54  return true;
55}
56
57}  // namespace shill
58