import unittest

from aprsd import email


class TestEmail(unittest.TestCase):
    def test_get_email_from_shortcut(self):
        email.CONFIG = {"aprsd": {"email": {"shortcuts": {}}}}
        email_address = "something@something.com"
        addr = "-{}".format(email_address)
        actual = email.get_email_from_shortcut(addr)
        self.assertEqual(addr, actual)

        email.CONFIG = {"aprsd": {"email": {"nothing": "nothing"}}}
        actual = email.get_email_from_shortcut(addr)
        self.assertEqual(addr, actual)

        email.CONFIG = {"aprsd": {"email": {"shortcuts": {"not_used": "empty"}}}}
        actual = email.get_email_from_shortcut(addr)
        self.assertEqual(addr, actual)

        email.CONFIG = {"aprsd": {"email": {"shortcuts": {"-wb": email_address}}}}
        short = "-wb"
        actual = email.get_email_from_shortcut(short)
        self.assertEqual(email_address, actual)