// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#pragma once

#include <map>

#include "garnet/bin/zxdb/console/format_value.h"
#include "garnet/bin/zxdb/symbols/location.h"

namespace zxdb {

class Process;

class MockFormatValueProcessContext : public FormatValue::ProcessContext {
 public:
  MockFormatValueProcessContext();
  ~MockFormatValueProcessContext() override;

  // Adds a mock result for a given address query. Must be an exact match.
  void AddResult(uint64_t address, Location location);

  // FormatValue::ProcessContext implementation.
  Location GetLocationForAddress(uint64_t address) const override;

 private:
  std::map<uint64_t, Location> locations_;
};

}  // namespace zxdb