#!/usr/bin/env python # -*- coding: utf-8 -*- from proxy import Proxy, NoTalkProxy import sys from time import time if sys.version_info[0] == 2: from StringIO import StringIO else: from io import StringIO if sys.version_info < (2, 7): import unittest2 as unittest else: import unittest class ProxyTest(unittest.TestCase): @classmethod def setUpClass(cls): """ Class scope setup. """ cls.p = Proxy() def setUp(cls): """ Function/test case scope setup. """ cls.output = StringIO() cls.saved_stdout = sys.stdout sys.stdout = cls.output def tearDown(cls): """ Function/test case scope teardown. """ cls.output.close() sys.stdout = cls.saved_stdout def test_sales_manager_shall_talk_through_proxy_with_delay(cls): cls.p.busy = 'No' start_time = time() cls.p.talk() end_time = time() execution_time = end_time - start_time print_output = cls.output.getvalue() expected_print_output = 'Proxy checking for Sales Manager availability\n\ Sales Manager ready to talk\n' cls.assertEqual(print_output, expected_print_output) expected_execution_time = 1 cls.assertEqual(int(execution_time*10), expected_execution_time) def test_sales_manager_shall_respond_through_proxy_with_delay(cls): cls.p.busy = 'Yes' start_time = time() cls.p.talk() end_time = time() execution_time = end_time - start_time print_output = cls.output.getvalue() expected_print_output = 'Proxy checking for Sales Manager availability\n\ Sales Manager is busy\n' cls.assertEqual(print_output, expected_print_output) expected_execution_time = 1 cls.assertEqual(int(execution_time*10), expected_execution_time) class NoTalkProxyTest(unittest.TestCase): @classmethod def setUpClass(cls): """ Class scope setup. """ cls.ntp = NoTalkProxy() def setUp(cls): """ Function/test case scope setup. """ cls.output = StringIO() cls.saved_stdout = sys.stdout sys.stdout = cls.output def tearDown(cls): """ Function/test case scope teardown. """ cls.output.close() sys.stdout = cls.saved_stdout def test_sales_manager_shall_not_talk_through_proxy_with_delay(cls): cls.ntp.busy = 'No' start_time = time() cls.ntp.talk() end_time = time() execution_time = end_time - start_time print_output = cls.output.getvalue() expected_print_output = 'Proxy checking for Sales Manager availability\n\ This Sales Manager will not talk to you whether he/she is busy or not\n' cls.assertEqual(print_output, expected_print_output) expected_execution_time = 1 cls.assertEqual(int(execution_time*10), expected_execution_time) def test_sales_manager_shall_not_respond_through_proxy_with_delay(cls): cls.ntp.busy = 'Yes' start_time = time() cls.ntp.talk() end_time = time() execution_time = end_time - start_time print_output = cls.output.getvalue() expected_print_output = 'Proxy checking for Sales Manager availability\n\ This Sales Manager will not talk to you whether he/she is busy or not\n' cls.assertEqual(print_output, expected_print_output) expected_execution_time = 1 cls.assertEqual(int(execution_time*10), expected_execution_time) if __name__ == "__main__": unittest.main()