使用C++ GMock匹配器测试unordered_map的键是否来自一组键

#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <unordered_map>
#include <string>
#include <vector>

using namespace testing;

// 假设的处理函数
void processMap(std::unordered_map<int, std::string>& map) {
    // ... 进行一些处理 ...
}

// 自定义匹配器，用于检查unordered_map的所有键是否都在给定的键集合中
MATCHER_P(KeysInSet, keys, "") {
    for (const auto& pair : arg) {
        if (std::find(keys.begin(), keys.end(), pair.first) == keys.end()) {
            *result_listener << "Key " << pair.first << " not found in the set.";
            return false;
        }
    }
    return true;
}

TEST(ProcessMapTest, KeysAreInExpectedSet) {
    std::unordered_map<int, std::string> map = {{1, "one"}, {2, "two"}, {3, "three"}};
    std::vector<int> expectedKeys = {1, 2, 3};

    processMap(map);

    EXPECT_THAT(map, KeysInSet(expectedKeys));
}