chrome/browser/chromeos/app_mode/test_kiosk_extension_builder.h

// Copyright 2018 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 CHROME_BROWSER_CHROMEOS_APP_MODE_TEST_KIOSK_EXTENSION_BUILDER_H_
#define CHROME_BROWSER_CHROMEOS_APP_MODE_TEST_KIOSK_EXTENSION_BUILDER_H_

#include <string>
#include <vector>

#include "base/memory/ref_counted.h"
#include "extensions/common/manifest.h"
#include "extensions/common/manifest_handlers/kiosk_mode_info.h"

namespace extensions {
class Extension;
}

namespace chromeos {

// Wrapper around extensions::ExtensionBuilder for creating extension::Extension
// instances for usage in kiosk app tests.
class TestKioskExtensionBuilder {
 public:
  TestKioskExtensionBuilder(extensions::Manifest::Type type,
                            const std::string& extension_id);
  ~TestKioskExtensionBuilder();

  const std::string& extension_id() const { return extension_id_; }
  const std::string& version() const { return version_; }

  void set_kiosk_enabled(bool enabled) { kiosk_enabled_ = enabled; }
  void set_offline_enabled(bool enabled) { offline_enabled_ = enabled; }
  void set_version(const std::string& version) { version_ = version; }

  void AddSecondaryExtension(const std::string& id);
  void AddSecondaryExtensionWithEnabledOnLaunch(const std::string& id,
                                                bool enabled_on_launch);

  scoped_refptr<const extensions::Extension> Build() const;

 private:
  const extensions::Manifest::Type type_;
  const std::string extension_id_;

  bool kiosk_enabled_ = true;
  bool offline_enabled_ = true;
  std::vector<extensions::SecondaryKioskAppInfo> secondary_extensions_;
  std::string version_ = "1.0";

  DISALLOW_COPY_AND_ASSIGN(TestKioskExtensionBuilder);
};

}  // namespace chromeos

#endif  // CHROME_BROWSER_CHROMEOS_APP_MODE_TEST_KIOSK_EXTENSION_BUILDER_H_