FilesExpand file tree

 master

/

test_proxy.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (30 loc) · 1.05 KB

 master

/

test_proxy.py

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (30 loc) · 1.05 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

import sys

import unittest

from io import StringIO

from patterns.structural.proxy import Proxy, client

class ProxyTest(unittest.TestCase):

@classmethod

def setUpClass(cls):

"""Class scope setup."""

cls.proxy = 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_do_the_job_for_admin_shall_pass(self):

client(self.proxy, "admin")

assert self.output.getvalue() == (

"[log] Doing the job for admin is requested.\n"

"I am doing the job for admin\n"

)

def test_do_the_job_for_anonymous_shall_reject(self):

client(self.proxy, "anonymous")

assert self.output.getvalue() == (

"[log] Doing the job for anonymous is requested.\n"

"[log] I can do the job just for `admins`.\n"

)