comparison wibble/tests.py @ 11:dbd55e6af4f6

Added test
author Ben Croston <ben@croston.org>
date Mon, 05 Sep 2011 12:13:45 +0100
parents 58e764e39492
children 4ac14cfadc15
comparison
equal deleted inserted replaced
10:58e764e39492 11:dbd55e6af4f6
35 def echo(self, mystring): 35 def echo(self, mystring):
36 return 'ECHO: ' + mystring 36 return 'ECHO: ' + mystring
37 37
38 def raiseexception(self): 38 def raiseexception(self):
39 dividebyzeroerror = 1/0 39 dividebyzeroerror = 1/0
40
41 def returnnothing(self):
42 pass
40 43
41 def myauth(username, password, useragent=None): 44 def myauth(username, password, useragent=None):
42 return username == 'testuser' and \ 45 return username == 'testuser' and \
43 hashlib.md5('s3cr3t').hexdigest() == password and \ 46 hashlib.md5('s3cr3t').hexdigest() == password and \
44 useragent == 'wibble_unittest' 47 useragent == 'wibble_unittest'
92 POUND = '\u00A3' 95 POUND = '\u00A3'
93 else: 96 else:
94 POUND = unicode('\u00A3') 97 POUND = unicode('\u00A3')
95 self.assertEqual(self.client.echo(POUND), 'ECHO: ' + POUND) 98 self.assertEqual(self.client.echo(POUND), 'ECHO: ' + POUND)
96 self.assertEqual(self.client.echo('hello mum!'), 'ECHO: hello mum!') 99 self.assertEqual(self.client.echo('hello mum!'), 'ECHO: hello mum!')
100
101 class ReturnNothing(WibbleTests):
102 def runTest(self):
103 self.assertEqual(self.client.returnnothing(), None)
97 ##### client ^^^ ##### 104 ##### client ^^^ #####
98 105
99 finished = False 106 finished = False
100 def suite(): 107 def suite():
101 if platform.python_version().startswith('2'): 108 if platform.python_version().startswith('2'):
113 suite.addTest(NotAuthTest()) 120 suite.addTest(NotAuthTest())
114 suite.addTest(IgnoreClassNameTest()) 121 suite.addTest(IgnoreClassNameTest())
115 suite.addTest(ExceptionTest()) 122 suite.addTest(ExceptionTest())
116 suite.addTest(BadRequestTest()) 123 suite.addTest(BadRequestTest())
117 suite.addTest(EchoTest()) 124 suite.addTest(EchoTest())
125 suite.addTest(ReturnNothing())
118 return suite 126 return suite
119 127
120 if __name__ == '__main__': 128 if __name__ == '__main__':
121 import sys 129 import sys
122 if platform.python_version().startswith('2') and 'serve' in sys.argv: 130 if platform.python_version().startswith('2') and 'serve' in sys.argv: