def test_driver_setup_and_teardown(self, Popen, call): Popen.return_value = 1 dl = DesktopLibrary() dl.driver_setup() self.assertTrue(dl.winappdriver.process) dl.driver_teardown() self.assertFalse(dl.winappdriver.process)
def test_driver_child_process_teardown(self): mock_child = MagicMock() dl = DesktopLibrary() dl.winappdriver.process = MagicMock() dl.winappdriver.process.pid = 1 psutil.Process.create_time = MagicMock() psutil.Process.children = MagicMock(return_value=[mock_child]) self.assertFalse(dl.winappdriver.process is None) dl.driver_teardown() self.assertTrue(dl.winappdriver.process is None)
def test_teardown_without_setup(self, call): dl = DesktopLibrary() dl.driver_teardown() self.assertFalse(dl.winappdriver.process)