prometheus-cpp/tests/registry_test.cc

#include <vector>

#include <gmock/gmock.h>

#include "lib/collectable.h"
#include "lib/registry.h"

using namespace testing;
using namespace prometheus;

class MockCollectable : public Collectable {
 public:
  MOCK_METHOD0(collect, std::vector<io::prometheus::client::MetricFamily>());
};

class RegistryTest : public Test {};

TEST_F(RegistryTest, collectsSingleMetricFamily) {
  Registry registry{{}};

  auto counterFamily = registry.add_counter("test", "a test", {});
  counterFamily->add({{"name", "counter1"}});
  counterFamily->add({{"name", "counter2"}});
  auto collected = registry.collect();
  ASSERT_EQ(collected.size(), 1);
  EXPECT_EQ(collected[0].name(), "test");
  EXPECT_EQ(collected[0].help(), "a test");
  ASSERT_EQ(collected[0].metric_size(), 2);
  ASSERT_EQ(collected[0].metric(0).label_size(), 1);
  EXPECT_EQ(collected[0].metric(0).label(0).name(), "name");
  ASSERT_EQ(collected[0].metric(1).label_size(), 1);
  EXPECT_EQ(collected[0].metric(1).label(0).name(), "name");
}