TEST(RepoBSONTest, CloneAndShrink)
{
//shrinking a bson without any binary fields should yield an identical bson
RepoBSON shrunkBson = testBson.cloneAndShrink();
EXPECT_EQ(testBson, shrunkBson);
EXPECT_EQ(testBson.getFilesMapping().size(), shrunkBson.getFilesMapping().size());
mongo::BSONObjBuilder builder;
std::vector < uint8_t > in, out, ref;
size_t size = 100;
in.resize(size);
ref.resize(size);
builder << "stringTest" << "hello";
builder << "numTest" << 1.35;
builder.appendBinData("binDataTest", in.size(), mongo::BinDataGeneral, in.data());
std::unordered_map < std::string, std::pair<std::string, std::vector<uint8_t>>> mapping, outMapping;
mapping["orgRef"] = std::pair<std::string, std::vector<uint8_t>>("blah", ref);
RepoBSON binBson(builder.obj(), mapping);
shrunkBson = binBson.cloneAndShrink();
outMapping = shrunkBson.getFilesMapping();
EXPECT_NE(shrunkBson, binBson);
EXPECT_FALSE(shrunkBson.hasField("binDataTest"));
EXPECT_EQ(2, outMapping.size());
EXPECT_TRUE(outMapping.find("orgRef") != outMapping.end());
//Check the binary still obtainable
EXPECT_TRUE(shrunkBson.getBinaryFieldAsVector("binDataTest", out));
ASSERT_EQ(in.size(), out.size());
for (size_t i = 0; i < out.size(); ++i)
{
EXPECT_EQ(in[i], out[i]);
}
//Check the out referenced bigfile is still sane
EXPECT_EQ(ref.size(), outMapping["orgRef"].second.size());
for (size_t i = 0; i < ref.size(); ++i)
{
EXPECT_EQ(ref[i], outMapping["orgRef"].second[i]);
}
}
