// Copyright 2021-2023 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef THIRD_PARTY_NEARBY_SHARING_INTERNAL_TEST_FAKE_CONTEXT_H_
#define THIRD_PARTY_NEARBY_SHARING_INTERNAL_TEST_FAKE_CONTEXT_H_

#include <stdint.h>

#include <memory>

#include "internal/platform/clock.h"
#include "internal/platform/task_runner.h"
#include "internal/platform/timer.h"
#include "internal/test/fake_clock.h"
#include "internal/test/fake_task_runner.h"
#include "sharing/internal/api/bluetooth_adapter.h"
#include "sharing/internal/api/fast_initiation_manager.h"
#include "sharing/internal/public/connectivity_manager.h"
#include "sharing/internal/public/context.h"
#include "sharing/internal/test/fake_bluetooth_adapter.h"
#include "sharing/internal/test/fake_connectivity_manager.h"
#include "sharing/internal/test/fake_fast_initiation_manager.h"

namespace nearby {

class FakeContext : public Context {
 public:
  FakeContext();
  ~FakeContext() override = default;

  Clock* GetClock() const override;
  std::unique_ptr<Timer> CreateTimer() override;
  ConnectivityManager* GetConnectivityManager() const override;
  sharing::api::BluetoothAdapter& GetBluetoothAdapter() const override;
  api::FastInitiationManager& GetFastInitiationManager() const override;
  std::unique_ptr<TaskRunner> CreateSequencedTaskRunner() const override;
  std::unique_ptr<TaskRunner> CreateConcurrentTaskRunner(
      uint32_t concurrent_count) const override;
  TaskRunner* GetTaskRunner() override;

  FakeClock* fake_clock() const { return fake_clock_.get(); }
  FakeConnectivityManager* fake_connectivity_manager() const {
    return fake_connectivity_manager_.get();
  }
  FakeBluetoothAdapter* fake_bluetooth_adapter() const {
    return fake_bluetooth_adapter_.get();
  }
  FakeFastInitiationManager* fake_fast_initiation_manager() const {
    return fake_fast_initiation_manager_.get();
  }
  FakeTaskRunner* fake_task_runner() const {
    return executor_.get();
  }

  FakeTaskRunner* last_sequenced_task_runner() const {
    return last_sequenced_task_runner_;
  }

 private:
  std::unique_ptr<FakeClock> fake_clock_;
  std::unique_ptr<FakeConnectivityManager> fake_connectivity_manager_;
  std::unique_ptr<FakeBluetoothAdapter> fake_bluetooth_adapter_;
  std::unique_ptr<FakeFastInitiationManager> fake_fast_initiation_manager_;
  std::unique_ptr<FakeTaskRunner> executor_;
  mutable FakeTaskRunner* last_sequenced_task_runner_ = nullptr;
};

}  // namespace nearby

#endif  // THIRD_PARTY_NEARBY_SHARING_INTERNAL_TEST_FAKE_CONTEXT_H_