extensions/browser/mock_external_provider.h

// Copyright 2017 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 EXTENSIONS_BROWSER_MOCK_EXTERNAL_PROVIDER_H_
#define EXTENSIONS_BROWSER_MOCK_EXTERNAL_PROVIDER_H_

#include <map>
#include <memory>
#include <string>
#include <utility>

#include "base/files/file_path.h"
#include "base/macros.h"
#include "extensions/browser/external_provider_interface.h"
#include "extensions/common/extension_id.h"
#include "extensions/common/manifest.h"

namespace base {
class Version;
}

namespace extensions {

class MockExternalProvider : public ExternalProviderInterface {
 public:
  MockExternalProvider(VisitorInterface* visitor, Manifest::Location location);
  ~MockExternalProvider() override;

  void UpdateOrAddExtension(const ExtensionId& id,
                            const std::string& version,
                            const base::FilePath& path);
  void UpdateOrAddExtension(std::unique_ptr<ExternalInstallInfoFile> info);
  void UpdateOrAddExtension(std::unique_ptr<ExternalInstallInfoUpdateUrl> info);
  void RemoveExtension(const ExtensionId& id);

  // ExternalProviderInterface implementation:
  void VisitRegisteredExtension() override;
  bool HasExtension(const std::string& id) const override;
  bool GetExtensionDetails(
      const std::string& id,
      Manifest::Location* location,
      std::unique_ptr<base::Version>* version) const override;
  bool IsReady() const override;
  void ServiceShutdown() override {}

  int visit_count() const { return visit_count_; }
  void set_visit_count(int visit_count) { visit_count_ = visit_count; }

 private:
  using FileDataMap =
      std::map<ExtensionId, std::unique_ptr<ExternalInstallInfoFile>>;
  using UrlDataMap =
      std::map<ExtensionId, std::unique_ptr<ExternalInstallInfoUpdateUrl>>;
  FileDataMap file_extension_map_;
  UrlDataMap url_extension_map_;
  Manifest::Location location_;
  VisitorInterface* visitor_;

  // visit_count_ tracks the number of calls to VisitRegisteredExtension().
  // Mutable because it must be incremented on each call to
  // VisitRegisteredExtension(), which must be a const method to inherit
  // from the class being mocked.
  mutable int visit_count_;

  DISALLOW_COPY_AND_ASSIGN(MockExternalProvider);
};

}  // namespace extensions

#endif  // EXTENSIONS_BROWSER_MOCK_EXTERNAL_PROVIDER_H_