Skip to content
Snippets Groups Projects
Commit 45f71f4a authored by Haibo Huang's avatar Haibo Huang Committed by TensorFlower Gardener
Browse files

Add a little bit more test for `BoundedStreamPool` return

PiperOrigin-RevId: 735560832
parent 3decad65
No related merge requests found
......@@ -50,10 +50,13 @@ class StreamPoolTest : public ::testing::Test {
TEST_F(StreamPoolTest, Borrow) {
BoundedStreamPool pool(executor_, 1);
EXPECT_EQ(pool.GetAvailableStreamNum(), 1);
ASSERT_OK_AND_ASSIGN(BoundedStreamPool::Handle handle, pool.Borrow());
EXPECT_NE(handle.get(), nullptr);
EXPECT_EQ(handle.get(), &*handle);
EXPECT_EQ(pool.GetAvailableStreamNum(), 0);
{
ASSERT_OK_AND_ASSIGN(BoundedStreamPool::Handle handle, pool.Borrow());
EXPECT_NE(handle.get(), nullptr);
EXPECT_EQ(handle.get(), &*handle);
EXPECT_EQ(pool.GetAvailableStreamNum(), 0);
}
EXPECT_EQ(pool.GetAvailableStreamNum(), 1);
}
} // namespace
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment