testing.py

# This is just for testing
from LU import badSmell
from LU.badSmell import OrderItem

class Testing:

    def __init__(self):
        self.__run_tests()

    @staticmethod
    def assert_equal(expect, actual):
        if expect != actual:
            raise Exception('Test Failed!', str(expect) + ' != ' + str(actual))

    def __run_tests(self):
        self.__test_create_order()
        self.__test_handle_post_request()
        self.__test_order_item()
        self.__test_order_item_tax()
        self.__test_order_item_tax_for_vat_free_items()

    def __test_create_order(self):
        order_item = OrderItem('Test Item', 1, 1)
        order = badSmell.create_new_order_and_print('Test Order', [order_item], 'ABC', 1.16)
        self.assert_equal(order['name'], 'Test Order')
        self.assert_equal(len(order['items']), 1)
        self.assert_equal(order['items'][0].name, 'Test Item')
        self.assert_equal(order['items'][0].price, 1)
        self.assert_equal(order['items'][0].priceWithTax, 1.16)


    def __test_handle_post_request(self):
        order = badSmell.handlePostRequest({
            'name': 'My Order',
            'items': [
                {
                    'name': 'Milk',
                    'price': 1.15,
                },
                {
                    'name': 'super_special_vat_free_item',
                    'price': 1.50,
                }
            ],
            'buyerId': 'abc123'
        })
        self.assert_equal(order['name'], 'My Order')
        self.assert_equal(len(order['items']), 2)
        self.assert_equal(order['items'][0].name, 'Milk')
        self.assert_equal(order['buyerId'], 'abc123')


    def __test_order_item(self):
        orderItem = OrderItem('Test Item', 100, 1)
        self.assert_equal(orderItem.name, 'Test Item')
        self.assert_equal(orderItem.price, 100)
        self.assert_equal(orderItem.id, 1)


    def __test_order_item_tax(self):
        order_item = OrderItem('Test Item', 1, 1)
        order = badSmell.create_new_order_and_print('Test Order', [order_item], 'ABC', 1.16)
        self.assert_equal(order['items'][0].price, 1)
        self.assert_equal(order['items'][0].priceWithTax, 1.16)

    def __test_order_item_tax_for_vat_free_items(self):
        order_item = OrderItem('post_stamps', 0.3, 1)
        order = badSmell.create_new_order_and_print('Order Post Stamps', [order_item], 'ABC', 1.16)
        self.assert_equal(order['items'][0].price, 0.3)
        self.assert_equal(order['items'][0].priceWithTax, 0.3)


Testing()