intern/cycles/test/util_task_test.cpp

   1 /*
   2  * Copyright 2011-2016 Blender Foundation
   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
  17 #include "testing/testing.h"
  18
  19 #include "util/util_task.h"
  20
  21 CCL_NAMESPACE_BEGIN
  22
  23 namespace {
  24
  25 void task_run() {
  26 }
  27
  28 }  // namespace
  29
  30 TEST(util_task, basic) {
  31         TaskScheduler::init(0);
  32         TaskPool pool;
  33         for(int i = 0; i < 100; ++i) {
  34                 pool.push(function_bind(task_run));
  35         }
  36         TaskPool::Summary summary;
  37         pool.wait_work(&summary);
  38         TaskScheduler::exit();
  39         EXPECT_EQ(summary.num_tasks_handled, 100);
  40 }
  41
  42 TEST(util_task, multiple_times) {
  43         for(int N = 0; N < 1000; ++N) {
  44                 TaskScheduler::init(0);
  45                 TaskPool pool;
  46                 for(int i = 0; i < 100; ++i) {
  47                         pool.push(function_bind(task_run));
  48                 }
  49                 TaskPool::Summary summary;
  50                 pool.wait_work(&summary);
  51                 TaskScheduler::exit();
  52                 EXPECT_EQ(summary.num_tasks_handled, 100);
  53         }
  54 }
  55
  56 CCL_NAMESPACE_END