Exemplo n.º 1
0
TEST(RepoBSONTest, GetListStringPairField)
{
	std::vector<std::vector<std::string>> vecIn;

	size_t size = 10;
	mongo::BSONObjBuilder builder, arrbuilder;

	vecIn.reserve(size);
	for (size_t i = 0; i < size; ++i)
	{
		int n1 = rand();
		int n2 = rand();
		vecIn.push_back({ std::to_string(n1), std::to_string(n2) });

		arrbuilder << std::to_string(i) << BSON("first" << std::to_string(n1) << "second" << std::to_string(n2) << "third" << 1);
	}

	builder.appendArray("bsonArr", arrbuilder.obj());

	RepoBSON bson = builder.obj();

	std::list<std::pair<std::string, std::string>> vecOut = 
		bson.getListStringPairField("bsonArr", "first", "second");

	EXPECT_EQ(vecIn.size(), vecOut.size());
	auto listIt = vecOut.begin();

	for (size_t i = 0; i < size; i++)
	{

		EXPECT_EQ(vecIn[i][0], listIt->first);
		EXPECT_EQ(vecIn[i][1], listIt->second);
		++listIt;

	}

	//Shouldn't fail if trying to get a uuid field that doesn't exist
	EXPECT_EQ(0, bson.getListStringPairField("hello", "first", "third").size());
	EXPECT_EQ(0, bson.getListStringPairField("hello", "hi", "bye").size());
	EXPECT_EQ(0, bson.getListStringPairField("hello", "first", "second").size());
	EXPECT_EQ(0, testBson.getListStringPairField("ice", "first", "second").size());
	EXPECT_EQ(0, emptyBson.getListStringPairField("ice", "first", "second").size());
}