1//===- llvm/Testing/ADT/StringMap.h ---------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef LLVM_TESTING_ADT_STRINGMAP_H_
10#define LLVM_TESTING_ADT_STRINGMAP_H_
11
12#include "llvm/ADT/STLExtras.h"
13#include "llvm/ADT/StringMap.h"
14#include "llvm/Support/FormatVariadic.h"
15#include "llvm/Testing/ADT/StringMapEntry.h"
16#include <ostream>
17#include <sstream>
18
19namespace llvm {
20
21/// Support for printing to std::ostream, for use with e.g. producing more
22/// useful error messages with Google Test.
23template <typename T>
24std::ostream &operator<<(std::ostream &OS, const StringMap<T> &M) {
25  if (M.empty()) {
26    return OS << "{ }";
27  }
28
29  std::vector<std::string> Lines;
30  for (const auto &E : M) {
31    std::ostringstream SS;
32    SS << E << ",";
33    Lines.push_back(SS.str());
34  }
35  llvm::sort(Lines);
36  Lines.insert(Lines.begin(), "{");
37  Lines.insert(Lines.end(), "}");
38
39  return OS << llvm::formatv("{0:$[\n]}",
40                             make_range(Lines.begin(), Lines.end()))
41                   .str();
42}
43
44} // namespace llvm
45
46#endif
47