1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63


ash / api / tasks / test_tasks_delegate.h [blame]

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef ASH_API_TASKS_TEST_TASKS_DELEGATE_H_
#define ASH_API_TASKS_TEST_TASKS_DELEGATE_H_

#include <optional>

#include "ash/api/tasks/tasks_delegate.h"
#include "ash/ash_export.h"
#include "base/containers/flat_map.h"
#include "components/account_id/account_id.h"

namespace ash::api {

class TasksClient;

class ASH_EXPORT TestTasksDelegate : public TasksDelegate {
 public:
  TestTasksDelegate();
  TestTasksDelegate(const TestTasksDelegate&) = delete;
  TestTasksDelegate& operator=(const TestTasksDelegate&) = delete;
  ~TestTasksDelegate() override;

  // TasksDelegate:
  void UpdateClientForProfileSwitch(const AccountId& account_id) override;
  void GetTaskLists(bool force_fetch,
                    TasksClient::GetTaskListsCallback callback) override;
  void GetTasks(const std::string& task_list_id,
                bool force_fetch,
                TasksClient::GetTasksCallback callback) override;
  void AddTask(const std::string& task_list_id,
               const std::string& title,
               TasksClient::OnTaskSavedCallback callback) override;
  void UpdateTask(const std::string& task_list_id,
                  const std::string& task_id,
                  const std::string& title,
                  bool completed,
                  TasksClient::OnTaskSavedCallback callback) override;

  // Helper function for adding pre-build `TasksClient` objects.
  void AddFakeTasksClient(const AccountId& account_id,
                          std::unique_ptr<TasksClient> tasks_client);

  // Switches the active account for the delegate.
  void SetActiveAccount(const AccountId& account_id);

 private:
  // Returns a reference to the client registered for a current
  // account. CHECK-fails if no client is registered for it.
  TasksClient& GetClientForActiveAccount();

  // The currently active account.
  std::optional<AccountId> active_account_id_;

  // Maps account IDs to clients.
  base::flat_map<AccountId, std::unique_ptr<TasksClient>> clients_;
};

}  // namespace ash::api

#endif  // ASH_API_TASKS_TEST_TASKS_DELEGATE_H_