def test_getlogin_bad(self):
with mock.patch("os.getenv") as os_getenv:
with mock.patch("os.getlogin") as os_getlogin:
os_getenv.side_effect = [None, None, "bar"]
os_getlogin.side_effect = OSError("[Errno 22] Invalid argument")
self.assertEqual(wrapper._getlogin(), "bar")
os_getlogin.assert_called_once_with()
self.assertEqual(os_getenv.call_count, 3)
def test_getlogin(self):
with mock.patch("os.getlogin") as os_getlogin:
os_getlogin.return_value = "foo"
self.assertEqual(wrapper._getlogin(), "foo")
