python-patterns/test_proxy.py at master · sd2017/python-patterns · GitHub

History

executable file

109 lines (91 loc) · 3.47 KB

Raw

100

101

102

103

104

105

106

107

108

109

#!/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()

python-patterns/test_proxy.py at master · sd2017/python-patterns · GitHub

Expand file tree

Breadcrumbs

test_proxy.py

Latest commit

History

Breadcrumbs

test_proxy.py

File metadata and controls