import unittest
from unittest import TestCase

from design_pattern_restuarant.Cook import Cook
from design_pattern_restuarant.Waitress import Waitress

from design_pattern_restuarant.Order import Order


class TestOrder(TestCase):
    def test_order_pizza(self):

        # Arrange
        cook = Cook('Bob')
        order = Order(cook)
        waitress = Waitress('Stacy')


        # Act
        waitress.take_order(order)
        waitress.place_order()

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