class TestMailList(unittest.TestCase):
def setUp(self):
self.obj = MailList("someName")
self.obj.add_subscriber("Pepa", "pepa@pepa")
def test_atributes(self):
self.assertEqual("someName", self.obj.getName())
def test_add_subscriber(self):
self.obj.add_subscriber("Tsveta", "tsveta@tsveta")
expected = [("Pepa", "pepa@pepa"), ("Tsveta", "tsveta@tsveta")]
self.assertEqual(expected, self.obj.get_subscribers())
def test_get_subscribers(self):
self.obj.add_subscriber("Tsveta", "tsveta@tsveta")
expected = [("Pepa", "pepa@pepa"), ("Tsveta", "tsveta@tsveta")]
self.assertEqual(expected, self.obj.get_subscribers())
class FileMailListAdapter():
"""docstring for FileMailListAdapter"""
def __init__(self, filename):
self.__mail = MailList(filename)
def make_subscribers(self):
file = open(self.__mail.getName(), "r")
contents = file.read()
file.close()
if not len(contents) == 0:
contents = list(map(lambda x: x.split(), contents.split('\n')))
# contents = list(map(lambda x: (x[0], x[2]), contents))
# self.__mail.subscribers = contents
# return contents
dict_ = {}
for item in contents:
dict_[item[0]] = item[2]
return dict_
return {}
def getMail(self):
self.__mail.subscribers = self.make_subscribers()
return self.__mail
