folly/executors/test/UnboundedBlockingQueueTest.cpp

   1 /*
   2  * Copyright 2017-present Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 #include <folly/Baton.h>
  17 #include <folly/executors/UnboundedBlockingQueue.h>
  18 #include <gtest/gtest.h>
  19 #include <thread>
  20
  21 using namespace folly;
  22
  23 TEST(UnboundedQueuee, push_pop) {
  24   UnboundedBlockingQueue<int> q;
  25   q.add(42);
  26   EXPECT_EQ(42, q.take());
  27 }
  28 TEST(UnboundedBlockingQueue, size) {
  29   UnboundedBlockingQueue<int> q;
  30   EXPECT_EQ(0, q.size());
  31   q.add(42);
  32   EXPECT_EQ(1, q.size());
  33   q.take();
  34   EXPECT_EQ(0, q.size());
  35 }
  36
  37 TEST(UnboundedBlockingQueue, concurrent_push_pop) {
  38   UnboundedBlockingQueue<int> q;
  39   Baton<> b1, b2;
  40   std::thread t([&] {
  41     b1.post();
  42     EXPECT_EQ(42, q.take());
  43     EXPECT_EQ(0, q.size());
  44     b2.post();
  45   });
  46   b1.wait();
  47   q.add(42);
  48   b2.wait();
  49   EXPECT_EQ(0, q.size());
  50   t.join();
  51 }