fake_camera_roll_download_manager.h 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef ASH_COMPONENTS_PHONEHUB_FAKE_CAMERA_ROLL_DOWNLOAD_MANAGER_H_
  5. #define ASH_COMPONENTS_PHONEHUB_FAKE_CAMERA_ROLL_DOWNLOAD_MANAGER_H_
  6. #include <vector>
  7. #include "ash/components/phonehub/camera_roll_download_manager.h"
  8. #include "ash/components/phonehub/proto/phonehub_api.pb.h"
  9. #include "ash/services/secure_channel/public/mojom/secure_channel_types.mojom.h"
  10. #include "base/containers/flat_map.h"
  11. namespace ash {
  12. namespace phonehub {
  13. class FakeCameraRollDownloadManager : public CameraRollDownloadManager {
  14. public:
  15. FakeCameraRollDownloadManager();
  16. ~FakeCameraRollDownloadManager() override;
  17. // CameraRollDownloadManager:
  18. void CreatePayloadFiles(
  19. int64_t payload_id,
  20. const phonehub::proto::CameraRollItemMetadata& item_metadata,
  21. CreatePayloadFilesCallback payload_files_callback) override;
  22. void UpdateDownloadProgress(
  23. secure_channel::mojom::FileTransferUpdatePtr update) override;
  24. void DeleteFile(int64_t payload_id) override;
  25. void set_expected_create_payload_files_result(
  26. CreatePayloadFilesResult expected_create_payload_files_result) {
  27. expected_create_payload_files_result_ =
  28. expected_create_payload_files_result;
  29. }
  30. const std::vector<secure_channel::mojom::FileTransferUpdatePtr>&
  31. GetFileTransferUpdates(int64_t payload_id) const;
  32. private:
  33. CreatePayloadFilesResult expected_create_payload_files_result_ =
  34. CreatePayloadFilesResult::kSuccess;
  35. // A map from payload IDs to the list of FileTransferUpdate received for each
  36. // payload.
  37. base::flat_map<int64_t,
  38. std::vector<secure_channel::mojom::FileTransferUpdatePtr>>
  39. payload_update_map_;
  40. };
  41. } // namespace phonehub
  42. } // namespace ash
  43. #endif // ASH_COMPONENTS_PHONEHUB_FAKE_CAMERA_ROLL_DOWNLOAD_MANAGER_H_