From 511dc309c85e5590d6fa0982299ffdd01d5cf3d3 Mon Sep 17 00:00:00 2001
From: Michele Caini <michele.caini@gmail.com>
Date: Thu, 26 Jan 2017 14:51:33 +0100
Subject: [PATCH] minor changes

---
 test/uvw/work.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/test/uvw/work.cpp b/test/uvw/work.cpp
index aa4a5537..5f12c33c 100644
--- a/test/uvw/work.cpp
+++ b/test/uvw/work.cpp
@@ -4,10 +4,18 @@
 
 TEST(Work, RunTask) {
     auto loop = uvw::Loop::getDefault();
+    auto handle = loop->resource<uvw::CheckHandle>();
 
     bool checkWorkEvent = false;
     bool checkTask = false;
 
+    handle->on<uvw::CheckEvent>([&checkWorkEvent](const auto &, auto &hndl) {
+        if(checkWorkEvent) {
+            hndl.stop();
+            hndl.close();
+        }
+    });
+
     auto req = loop->resource<uvw::WorkReq>([&checkTask]() {
         ASSERT_FALSE(checkTask);
         checkTask = true;
@@ -20,6 +28,7 @@ TEST(Work, RunTask) {
         checkWorkEvent = true;
     });
 
+    handle->start();
     req->queue();
     loop->run();
 
@@ -29,11 +38,17 @@ TEST(Work, RunTask) {
 
 TEST(Work, Cancellation) {
     auto loop = uvw::Loop::getDefault();
+    auto handle = loop->resource<uvw::CheckHandle>();
 
     bool checkErrorEvent = false;
     bool checkWorkEvent = false;
     bool checkTask = false;
 
+    handle->on<uvw::CheckEvent>([](const auto &, auto &hndl) {
+        hndl.stop();
+        hndl.close();
+    });
+
     auto req = loop->resource<uvw::WorkReq>([&checkTask]() {
         ASSERT_FALSE(checkTask);
         checkTask = true;
@@ -49,6 +64,7 @@ TEST(Work, Cancellation) {
         checkWorkEvent = true;
     });
 
+    handle->start();
     req->queue();
     req->cancel();
     loop->run();