test-threadpool.c 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
  2. *
  3. * Permission is hereby granted, free of charge, to any person obtaining a copy
  4. * of this software and associated documentation files (the "Software"), to
  5. * deal in the Software without restriction, including without limitation the
  6. * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  7. * sell copies of the Software, and to permit persons to whom the Software is
  8. * furnished to do so, subject to the following conditions:
  9. *
  10. * The above copyright notice and this permission notice shall be included in
  11. * all copies or substantial portions of the Software.
  12. *
  13. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  18. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  19. * IN THE SOFTWARE.
  20. */
  21. #include "uv.h"
  22. #include "task.h"
  23. static int work_cb_count;
  24. static int after_work_cb_count;
  25. static uv_work_t work_req;
  26. static char data;
  27. static void work_cb(uv_work_t* req) {
  28. ASSERT(req == &work_req);
  29. ASSERT(req->data == &data);
  30. work_cb_count++;
  31. }
  32. static void after_work_cb(uv_work_t* req, int status) {
  33. ASSERT(status == 0);
  34. ASSERT(req == &work_req);
  35. ASSERT(req->data == &data);
  36. after_work_cb_count++;
  37. }
  38. TEST_IMPL(threadpool_queue_work_simple) {
  39. int r;
  40. work_req.data = &data;
  41. r = uv_queue_work(uv_default_loop(), &work_req, work_cb, after_work_cb);
  42. ASSERT(r == 0);
  43. uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  44. ASSERT(work_cb_count == 1);
  45. ASSERT(after_work_cb_count == 1);
  46. MAKE_VALGRIND_HAPPY();
  47. return 0;
  48. }
  49. TEST_IMPL(threadpool_queue_work_einval) {
  50. int r;
  51. work_req.data = &data;
  52. r = uv_queue_work(uv_default_loop(), &work_req, NULL, after_work_cb);
  53. ASSERT(r == -1);
  54. uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  55. ASSERT(uv_last_error(uv_default_loop()).code == UV_EINVAL);
  56. ASSERT(work_cb_count == 0);
  57. ASSERT(after_work_cb_count == 0);
  58. MAKE_VALGRIND_HAPPY();
  59. return 0;
  60. }