chrome/browser/web_applications/test/fake_environment.h

// Copyright 2024 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_WEB_APPLICATIONS_TEST_FAKE_ENVIRONMENT_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_ENVIRONMENT_H_

#include <map>
#include <string>

#include "base/environment.h"
#include "base/strings/cstring_view.h"

namespace web_app {

// Provides mock environment variables values based on a stored map.
class FakeEnvironment : public base::Environment {
 public:
  FakeEnvironment();
  FakeEnvironment(const FakeEnvironment&) = delete;
  FakeEnvironment& operator=(const FakeEnvironment&) = delete;
  ~FakeEnvironment() override;

  void Set(base::cstring_view name, const std::string& value);

  // base::Environment:
  std::optional<std::string> GetVar(base::cstring_view variable_name) override;
  bool SetVar(base::cstring_view variable_name,
              const std::string& new_value) override;
  bool UnSetVar(base::cstring_view variable_name) override;

 private:
  std::map<std::string, std::string> variables_;
};

}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_ENVIRONMENT_H_