|
@@ -18,7 +18,9 @@
|
|
|
# You should have received a copy of the GNU Lesser General Public License along
|
|
|
# with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
|
import asyncio
|
|
|
+import asynctest
|
|
|
import unittest.mock
|
|
|
+
|
|
|
import qubesadmin.tests
|
|
|
import qubesadmin.tests.tools
|
|
|
import qubesadmin.tools.qvm_shutdown
|
|
@@ -85,9 +87,11 @@ class TC_00_qvm_shutdown(qubesadmin.tests.QubesTestCase):
|
|
|
loop = asyncio.new_event_loop()
|
|
|
asyncio.set_event_loop(loop)
|
|
|
|
|
|
+ mock_events = asynctest.CoroutineMock()
|
|
|
patch = unittest.mock.patch(
|
|
|
- 'qubesadmin.events.EventsDispatcher._get_events_reader')
|
|
|
- mock_events = patch.start()
|
|
|
+ 'qubesadmin.events.EventsDispatcher._get_events_reader',
|
|
|
+ mock_events)
|
|
|
+ patch.start()
|
|
|
self.addCleanup(patch.stop)
|
|
|
mock_events.side_effect = qubesadmin.tests.tools.MockEventsReader([
|
|
|
b'1\0\0connection-established\0\0',
|
|
@@ -114,9 +118,11 @@ class TC_00_qvm_shutdown(qubesadmin.tests.QubesTestCase):
|
|
|
loop = asyncio.new_event_loop()
|
|
|
asyncio.set_event_loop(loop)
|
|
|
|
|
|
+ mock_events = asynctest.CoroutineMock()
|
|
|
patch = unittest.mock.patch(
|
|
|
- 'qubesadmin.events.EventsDispatcher._get_events_reader')
|
|
|
- mock_events = patch.start()
|
|
|
+ 'qubesadmin.events.EventsDispatcher._get_events_reader',
|
|
|
+ mock_events)
|
|
|
+ patch.start()
|
|
|
self.addCleanup(patch.stop)
|
|
|
mock_events.side_effect = qubesadmin.tests.tools.MockEventsReader([
|
|
|
b'1\0\0connection-established\0\0',
|
|
@@ -159,9 +165,11 @@ class TC_00_qvm_shutdown(qubesadmin.tests.QubesTestCase):
|
|
|
loop = asyncio.new_event_loop()
|
|
|
asyncio.set_event_loop(loop)
|
|
|
|
|
|
+ mock_events = asynctest.CoroutineMock()
|
|
|
patch = unittest.mock.patch(
|
|
|
- 'qubesadmin.events.EventsDispatcher._get_events_reader')
|
|
|
- mock_events = patch.start()
|
|
|
+ 'qubesadmin.events.EventsDispatcher._get_events_reader',
|
|
|
+ mock_events)
|
|
|
+ patch.start()
|
|
|
self.addCleanup(patch.stop)
|
|
|
mock_events.side_effect = qubesadmin.tests.tools.MockEventsReader([
|
|
|
b'1\0\0connection-established\0\0',
|