def test_imapwidget(fake_qtile, monkeypatch):
monkeypatch.setitem(sys.modules, "imaplib", FakeIMAP("imaplib"))
monkeypatch.setitem(sys.modules, "keyring", FakeKeyring("keyring"))
from libqtile.widget import imapwidget
imap = imapwidget.ImapWidget(user="******")
fakebar = Bar([imap], 24)
fakebar.window = FakeWindow()
fakebar.width = 10
fakebar.height = 10
fakebar.draw = no_op
imap._configure(fake_qtile, fakebar)
text = imap.poll()
assert text == "INBOX: 2"
def test_imapwidget_keyring_error(fake_qtile, monkeypatch, fake_window):
monkeypatch.setitem(sys.modules, "imaplib", FakeIMAP("imaplib"))
monkeypatch.setitem(sys.modules, "keyring", FakeKeyring("keyring"))
from libqtile.widget import imapwidget
imapwidget.keyring.valid = False
imap = imapwidget.ImapWidget(user="******")
fakebar = Bar([imap], 24)
fakebar.window = fake_window
fakebar.width = 10
fakebar.height = 10
fakebar.draw = no_op
imap._configure(fake_qtile, fakebar)
text = imap.poll()
assert text == "Gnome Keyring Error"
def test_imapwidget_password_none(fake_qtile, monkeypatch):
monkeypatch.setitem(sys.modules, "imaplib", FakeIMAP("imaplib"))
monkeypatch.setitem(sys.modules, "keyring", FakeKeyring("keyring"))
from libqtile.widget import imapwidget
imapwidget.keyring.valid = False
imapwidget.keyring.error = False
imap = imapwidget.ImapWidget(user="******")
fakebar = Bar([imap], 24)
fakebar.window = FakeWindow()
fakebar.width = 10
fakebar.height = 10
fakebar.draw = no_op
imap._configure(fake_qtile, fakebar)
with pytest.raises(AttributeError):
with pytest.raises(UnboundLocalError):
imap.poll()