TEST_F(QueueTest, UpdateWeightExclDequeued) {
buildSimpleTree();
signalEgress(5, false);
updatePriority(5, {1, true, 7});
signalEgress(1, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{9, 40}, {7, 40}, {3, 20}}));
}
TEST_F(QueueTest, UpdateWeightNotEnqueued) {
addTransaction(1, {0, false, 7});
addTransaction(3, {0, false, 7});
signalEgress(1, false);
signalEgress(3, false);
updatePriority(1, {3, false, 7});
dump();
EXPECT_EQ(nodes_, IDList({{3, 100}, {1, 100}}));
}
TEST_F(QueueTest, Misc) {
buildSimpleTree();
signalEgress(1, true);
EXPECT_FALSE(q_.empty());
EXPECT_EQ(q_.numPendingEgress(), 1);
signalEgress(1, false);
EXPECT_EQ(q_.numPendingEgress(), 0);
EXPECT_TRUE(q_.empty());
removeTransaction(9);
removeTransaction(1);
dump();
EXPECT_EQ(nodes_, IDList({{3, 25}, {5, 25}, {7, 50}}));
}
TEST_F(QueueTest, UpdateParentDescendantExcl) {
buildSimpleTree();
updatePriority(1, {5, true, 7});
dump();
EXPECT_EQ(nodes_, IDList({{5, 100}, {1, 100}, {3, 20}, {7, 40}, {9, 40}}));
nextEgress();
EXPECT_EQ(nodes_, IDList({{5, 100}}));
signalEgress(5, false);
signalEgress(1, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 40}, {9, 40}, {3, 20}}));
}
TEST_F(QueueTest, UpdateParentSiblingExcl) {
buildSimpleTree();
updatePriority(7, {5, true, 3});
dump();
EXPECT_EQ(nodes_, IDList({{1, 100}, {3, 50}, {5, 50},
{7, 100}, {9, 100}}));
signalEgress(1, false);
signalEgress(3, false);
signalEgress(5, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 100}}));
}
TEST_F(QueueTest, nextEgressExclusiveAddWithEgress) {
buildSimpleTree();
// clear all egress, except 3
signalEgress(1, false);
signalEgress(5, false);
signalEgress(7, false);
signalEgress(9, false);
// Add a transaction with exclusive dependency, clear its egress
addTransaction(11, {1, true, 100});
signalEgress(11, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{3, 100}}));
}
TEST_F(QueueTest, nextEgress) {
buildSimpleTree();
nextEgress();
EXPECT_EQ(nodes_, IDList({{1, 100}}));
addTransaction(11, {7, false, 15});
signalEgress(1, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 50}, {3, 25}, {5, 25}}));
signalEgress(5, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 50}, {3, 25}, {9, 25}}));
signalEgress(5, true);
signalEgress(3, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 66}, {5, 33}}));
signalEgress(5, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 66}, {9, 33}}));
signalEgress(7, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{11, 66}, {9, 33}}));
signalEgress(9, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{11, 100}}));
signalEgress(3, true);
signalEgress(7, true);
signalEgress(9, true);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 50}, {3, 25}, {9, 25}}));
}
TEST_F(QueueTest, UpdateParentSibling) {
buildSimpleTree();
updatePriority(5, {3, false, 3});
dump();
EXPECT_EQ(nodes_, IDList({{1, 100}, {3, 33}, {5, 100},
{9, 100}, {7, 66}}));
signalEgress(1, false);
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 66}, {3, 33}}));
// Clear 5's egress (so it is only in the tree because 9 has egress) and move
// it back. Hit's a slightly different code path in reparent
signalEgress(5, false);
updatePriority(5, {1, false, 3});
dump();
EXPECT_EQ(nodes_, IDList({{1, 100}, {3, 25}, {7, 50}, {5, 25}, {9, 100}}));
nextEgress();
EXPECT_EQ(nodes_, IDList({{7, 50}, {3, 25}, {9, 25}}));
}
TEST_F(QueueTest, nextEgressExclusiveAdd) {
buildSimpleTree();
// clear all egress
signalEgress(1, false);
signalEgress(3, false);
signalEgress(5, false);
signalEgress(7, false);
signalEgress(9, false);
// Add a transaction with exclusive dependency, clear its egress
addTransaction(11, {1, true, 100});
signalEgress(11, false);
// signal egress for a child that got moved via exclusive dep
signalEgress(3, true);
nextEgress();
EXPECT_EQ(nodes_, IDList({{3, 100}}));
}
TEST_F(QueueTest, nextEgressRemoveParent) {
buildSimpleTree();
// Clear egress for all except txn=9
signalEgress(1, false);
signalEgress(3, false);
signalEgress(5, false);
signalEgress(7, false);
// Remove parent of 9 (5)
removeTransaction(5);
nextEgress();
EXPECT_EQ(nodes_, IDList({{9, 100}}));
// signal egress for 9's new siblings to verify weights
signalEgress(3, true);
signalEgress(7, true);
nextEgress();
EXPECT_EQ(nodes_, IDList({{9, 40}, {7, 40}, {3, 20}}));
}
void addTransaction(HTTPCodec::StreamID id, http2::PriorityUpdate pri) {
HTTP2PriorityQueue::Handle h = q_.addTransaction(id, pri, makeFakeTxn(id));
handles_.insert(std::make_pair(id, h));
signalEgress(id, 1);
}
