chromium/chrome/browser/ash/extensions/test_external_cache.h

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

#ifndef CHROME_BROWSER_ASH_EXTENSIONS_TEST_EXTERNAL_CACHE_H_
#define CHROME_BROWSER_ASH_EXTENSIONS_TEST_EXTERNAL_CACHE_H_

#include <map>
#include <memory>
#include <set>
#include <string>
#include <vector>

#include "base/memory/raw_ptr.h"
#include "base/values.h"
#include "chrome/browser/ash/extensions/external_cache.h"
#include "extensions/browser/updater/extension_downloader_delegate.h"

namespace chromeos {

class ExternalCacheDelegate;

// External cache implementation to be used in tests - instead of using a
// real local extension cache, which saves cached extension data to disk, it
// keeps cached extension CRX info in memory.
class TestExternalCache : public ExternalCache {
 public:
  TestExternalCache(ExternalCacheDelegate* delegate,
                    bool always_check_for_updates);

  TestExternalCache(const TestExternalCache&) = delete;
  TestExternalCache& operator=(const TestExternalCache&) = delete;

  ~TestExternalCache() override;

  // ExternalCache:
  const base::Value::Dict& GetCachedExtensions() override;
  void Shutdown(base::OnceClosure callback) override;
  void UpdateExtensionsList(base::Value::Dict prefs) override;
  void OnDamagedFileDetected(const base::FilePath& path) override;
  void RemoveExtensions(const std::vector<std::string>& ids) override;
  bool GetExtension(const std::string& id,
                    base::FilePath* file_path,
                    std::string* version) override;
  bool ExtensionFetchPending(const std::string& id) override;
  void PutExternalExtension(const std::string& id,
                            const base::FilePath& crx_file_path,
                            const std::string& version,
                            PutExternalExtensionCallback callback) override;
  void SetBackoffPolicy(
      std::optional<net::BackoffEntry::Policy> new_backoff_policy) override;

  // Simulates extension CRX download succeeding - it adds the extension
  // information to |cache_|.
  // |id| - the "downloaded" extension ID.
  // |crx_path| - the path to which the CRX is "downloaded".
  // |version| - the "downloaded" extension version.
  // |is_update| - whether the CRX download is started by extension update.
  // Returns whether the extension information has actually been saved. This
  // will return false if the cache does not think that the extension's
  // downaload is pending.
  bool SimulateExtensionDownloadFinished(const std::string& id,
                                         const std::string& crx_path,
                                         const std::string& version,
                                         bool is_update);

  // Simulates the extension's download failure. This will remove the extension
  // info from the set of extensions tracked in this external cache.
  // Returns whether the extension information has actually been changed. This
  // will return false if the cache does not think that the extension's
  // downaload is pending.
  bool SimulateExtensionDownloadFailed(
      const std::string& id,
      extensions::ExtensionDownloaderDelegate::Error error);

  // Set of the extension IDs in download pending state. Note that |this| does
  // not actually initiate the download, the user is expected to call either
  // SimulateExtensionDownloadFinished(), or SimulateExtensionDownloadFailed()
  // in order to remove an extension from this state and update the extension's
  // CRX information tracked by the external cache.
  const std::set<std::string>& pending_downloads() const {
    return pending_downloads_;
  }

  const std::optional<net::BackoffEntry::Policy>& backoff_policy() const {
    return backoff_policy_;
  }

 private:
  // An extension CRX information tracked by |this|.
  struct CrxCacheEntry {
    // The extension CRX path.
    std::string path;

    // The extension version associtated with the extension CRX.
    std::string version;
  };

  // Updates locally tracked state - called when the set of
  // |configured_extensions_| changes. It updates |cached_extensions_| and
  // adds extensions to |pending_downloads_| as needed.
  void UpdateCachedExtensions();

  // Adds an extension's CRX information to |crx_cache_|.
  void AddEntryToCrxCache(const std::string& id,
                          const std::string& crx_path,
                          const std::string& version);

  const raw_ptr<ExternalCacheDelegate> delegate_;
  const bool always_check_for_updates_;

  std::optional<net::BackoffEntry::Policy> backoff_policy_;

  base::Value::Dict configured_extensions_;
  base::Value::Dict cached_extensions_;

  std::set<std::string> pending_downloads_;
  std::map<std::string, CrxCacheEntry> crx_cache_;
};

}  // namespace chromeos

#endif  // CHROME_BROWSER_ASH_EXTENSIONS_TEST_EXTERNAL_CACHE_H_