def test_does_nothing_if_empty_tempdirs(self):
system = System()
system._tempdirs = set()
with patch.object(reactor, "callInThread") as callInThread:
system._remove_tempdirs()
self.assertEqual(callInThread.call_count, 0)
def test_calls_removes_directories(self):
system = System()
system._tempdirs = set([None])
with patch.object(reactor, "callInThread") as callInThread:
system._remove_tempdirs()
self.assertEqual(callInThread.call_count, 1)
self.assertEqual(system._tempdirs, set())
callInThread.assert_called_with(system._remove_directories, set([None]))
def test_remote_tempdirs_assertion(self):
system = System()
for entry in ("", 1, None, [], tuple(), dict()):
system._tempdirs = entry
with self.assertRaises(AssertionError):
system._remove_tempdirs()
