# unittest Module
from OrderSystem import Contact, Order

# Test adding a contact and adding an order for a contact
import unittest # Standard Library

class TestContactOrderSystem(unittest.TestCase):

    def test_contact_creation(self):
        i_name = input("Enter you name: ")
        i_email = input("Enter your email: ")

        contact = Contact(i_name, i_email)  # {}
        self.assertEqual(contact.name, i_name)
        self.assertEqual(contact.email, i_email)  


    def test_order_creation(self):
        contact = Contact("Bob Smith", "bob@1234.com")
        order = Order(1234, contact)
        self.assertEqual(order.order_id, 1234)
        self.assertEqual(order.contact, contact)

if __name__ == "__main__":
    unittest.main()