| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef COMPONENTS_CRYPTAUTH_BLE_MOCK_FOREGROUND_EID_GENERATOR_H_ |
| #define COMPONENTS_CRYPTAUTH_BLE_MOCK_FOREGROUND_EID_GENERATOR_H_ |
| |
| #include <memory> |
| #include <string> |
| #include <vector> |
| |
| #include "components/cryptauth/foreground_eid_generator.h" |
| |
| namespace cryptauth { |
| |
| class BeaconSeed; |
| struct RemoteDevice; |
| |
| // Mock class for ForegroundEidGenerator. Note that GoogleMock cannot be used to |
| // mock this class because GoogleMock's mock functions cannot return a |
| // |std::unique_ptr|. |
| class MockForegroundEidGenerator : public ForegroundEidGenerator { |
| public: |
| MockForegroundEidGenerator(); |
| ~MockForegroundEidGenerator() override; |
| |
| // Setters for the return values of the overridden functions below. |
| void set_background_scan_filter( |
| std::unique_ptr<EidData> background_scan_filter) { |
| background_scan_filter_ = std::move(background_scan_filter); |
| } |
| |
| void set_advertisement(std::unique_ptr<DataWithTimestamp> advertisement) { |
| advertisement_ = std::move(advertisement); |
| } |
| |
| void set_possible_advertisements( |
| std::unique_ptr<std::vector<std::string>> possible_advertisements) { |
| possible_advertisements_ = std::move(possible_advertisements); |
| } |
| |
| void set_identified_device(RemoteDevice* identified_device) { |
| identified_device_ = identified_device; |
| } |
| |
| // ForegroundEidGenerator: |
| std::unique_ptr<EidData> GenerateBackgroundScanFilter( |
| const std::vector<BeaconSeed>& scanning_device_beacon_seeds) |
| const override; |
| std::unique_ptr<DataWithTimestamp> GenerateAdvertisement( |
| const std::string& advertising_device_public_key, |
| const std::vector<BeaconSeed>& scanning_device_beacon_seeds) |
| const override; |
| std::vector<std::string> GeneratePossibleAdvertisements( |
| const std::string& advertising_device_public_key, |
| const std::vector<BeaconSeed>& scanning_device_beacon_seeds) |
| const override; |
| RemoteDevice const* IdentifyRemoteDeviceByAdvertisement( |
| const std::string& advertisement_service_data, |
| const std::vector<RemoteDevice>& device_list, |
| const std::vector<BeaconSeed>& scanning_device_beacon_seeds) |
| const override; |
| |
| int num_identify_calls() { return num_identify_calls_; } |
| |
| private: |
| std::unique_ptr<EidData> background_scan_filter_; |
| std::unique_ptr<DataWithTimestamp> advertisement_; |
| std::unique_ptr<std::vector<std::string>> possible_advertisements_; |
| const RemoteDevice* identified_device_; |
| |
| int num_identify_calls_; |
| }; |
| |
| } // namespace cryptauth |
| |
| #endif // COMPONENTS_CRYPTAUTH_BLE_MOCK_FOREGROUND_EID_GENERATOR_H_ |