X-Git-Url: https://git.saurik.com/apt.git/blobdiff_plain/76abe9a5aad69eb7e67295588c6825cdae0341af..b58e2c7c56b1416a343e81f9f80cb1f02c128e25:/test/libapt/srvrecs_test.cc?ds=inline diff --git a/test/libapt/srvrecs_test.cc b/test/libapt/srvrecs_test.cc index 4b63d2ccd..b3fa8102d 100644 --- a/test/libapt/srvrecs_test.cc +++ b/test/libapt/srvrecs_test.cc @@ -1,41 +1,75 @@ #include #include +#include -#include +#include #include +#include #include TEST(SrvRecTest, PopFromSrvRecs) { - // the PopFromSrvRecs() is using a random number so we must - // run it a bunch of times to ensure we are not fooled by randomness - std::set selected; - for(size_t i = 0; i < 100; ++i) + std::vector Meep; + Meep.emplace_back("foo", 20, 0, 80); + Meep.emplace_back("bar", 20, 0, 80); + Meep.emplace_back("baz", 30, 0, 80); + + EXPECT_EQ(Meep.size(), 3); + SrvRec const result = PopFromSrvRecs(Meep); + // ensure that pop removed one element + EXPECT_EQ(Meep.size(), 2); + EXPECT_NE(result.target, "baz"); + + SrvRec const result2 = PopFromSrvRecs(Meep); + EXPECT_NE(result.target, result2.target); + EXPECT_NE(result2.target, "baz"); + EXPECT_EQ(Meep.size(), 1); + + SrvRec const result3 = PopFromSrvRecs(Meep); + EXPECT_EQ(result3.target, "baz"); + EXPECT_TRUE(Meep.empty()); +} + +TEST(SrvRecTest,Randomness) +{ + constexpr unsigned int testLength = 100; + std::vector base1; + std::vector base2; + std::vector base3; + for (unsigned int i = 0; i < testLength; ++i) { - std::vector Meep; - Meep.emplace_back("foo", 20, 0, 80); - Meep.emplace_back("bar", 20, 0, 80); - Meep.emplace_back("baz", 30, 0, 80); - - EXPECT_EQ(Meep.size(), 3); - SrvRec const result = PopFromSrvRecs(Meep); - selected.insert(result.target); - // ensure that pop removed one element - EXPECT_EQ(Meep.size(), 2); - EXPECT_NE(result.target, "baz"); - - SrvRec const result2 = PopFromSrvRecs(Meep); - EXPECT_NE(result.target, result2.target); - EXPECT_NE(result2.target, "baz"); - EXPECT_EQ(Meep.size(), 1); - - SrvRec const result3 = PopFromSrvRecs(Meep); - EXPECT_EQ(result3.target, "baz"); - EXPECT_TRUE(Meep.empty()); + std::string name; + strprintf(name, "foo%d", i); + base1.emplace_back(name, 20, 0, 80); + base2.emplace_back(name, 20, 0, 80); + base3.emplace_back(name, 30, 0, 80); } + EXPECT_EQ(testLength, base1.size()); + EXPECT_EQ(testLength, base2.size()); + EXPECT_EQ(testLength, base3.size()); + std::move(base3.begin(), base3.end(), std::back_inserter(base2)); + EXPECT_EQ(testLength*2, base2.size()); + + std::vector first_pull; + for (unsigned int i = 0; i < testLength; ++i) + first_pull.push_back(PopFromSrvRecs(base1)); + EXPECT_TRUE(base1.empty()); + EXPECT_FALSE(first_pull.empty()); + EXPECT_EQ(testLength, first_pull.size()); + + std::vector second_pull; + for (unsigned int i = 0; i < testLength; ++i) + second_pull.push_back(PopFromSrvRecs(base2)); + EXPECT_FALSE(base2.empty()); + EXPECT_FALSE(second_pull.empty()); + EXPECT_EQ(testLength, second_pull.size()); + + EXPECT_EQ(first_pull.size(), second_pull.size()); + EXPECT_TRUE(std::all_of(first_pull.begin(), first_pull.end(), [](SrvRec const &R) { return R.priority == 20; })); + EXPECT_TRUE(std::all_of(second_pull.begin(), second_pull.end(), [](SrvRec const &R) { return R.priority == 20; })); + EXPECT_FALSE(std::equal(first_pull.begin(), first_pull.end(), second_pull.begin())); - // ensure that after enough runs we end up with both selected - EXPECT_EQ(selected.size(), 2); + EXPECT_TRUE(std::all_of(base2.begin(), base2.end(), [](SrvRec const &R) { return R.priority == 30; })); }