Skip to content
Snippets Groups Projects
Commit b5fbfa58 authored by shinedday's avatar shinedday
Browse files

Mail now have sending date

parent 850479ef
No related branches found
No related tags found
No related merge requests found
......@@ -17,10 +17,11 @@ class Mail:
Class message
"""
def __init__(self, id_sender: int, id_receiver: int, message: Any) -> None:
def __init__(self, id_sender: int, id_receiver: int, message: Any, sending_date: int) -> None:
self.__id_sender = id_sender
self.__id_receiver = id_receiver
self.__message = message
self.__sending_date = sending_date
def get_id_sender(self):
"""
......@@ -40,6 +41,12 @@ class Mail:
"""
return self.__message
def get_sending_date(self):
"""
return sending_date
"""
return self.__sending_date
class Mailbox:
"""
......@@ -72,7 +79,7 @@ class Mailbox:
"""
this method is called to send a message
"""
mail = Mail(self.__owner_id, id_receiver, message)
mail = Mail(self.__owner_id, id_receiver, message, self.__amas.get_cycle())
for agent in self.__amas.get_agents():
if agent.get_id() == id_receiver:
......
......@@ -15,17 +15,18 @@ class TestMail(TestCase):
"""
Test mail init
"""
mail = Mail(1, 5, None)
mail = Mail(1, 5, None, 0)
self.assertEqual(mail.get_id_sender(), 1)
self.assertEqual(mail.get_id_receiver(), 5)
self.assertEqual(mail.get_message(), None)
self.assertEqual(mail.get_sending_date(), 0)
mail = Mail(255, 0, "test")
mail = Mail(255, 0, "test", 12)
self.assertEqual(mail.get_id_sender(), 255)
self.assertEqual(mail.get_id_receiver(), 0)
self.assertEqual(mail.get_message(), "test")
self.assertEqual(mail.get_sending_date(), 12)
if __name__ == '__main__':
main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment