@@ -19,26 +19,26 @@ def test_index(self):
19
19
# Ensure Flask is setup.
20
20
response = self .app .get ('/' )
21
21
self .assertEqual (response .status_code , 200 )
22
- self .assertIn ('please enter some text to reverse...' , response .data )
22
+ self .assertIn (b 'please enter some text to reverse...' , response .data )
23
23
24
24
def test_text_is_reversed (self ):
25
25
# Ensure the '/reversed_input/<user_input>' route works.
26
26
response = self .app .get ('/reversed_input/FooBar' )
27
- self .assertIn ('raBooF' , response .data )
27
+ self .assertIn (b 'raBooF' , response .data )
28
28
29
29
def test_form_error (self ):
30
30
# Ensure error is populated correctly.
31
31
response = self .app .post ('/' , data = dict (
32
32
reverse_string = '' , follow_redirects = True )
33
33
)
34
- self .assertIn ('This field is required' , response .data )
34
+ self .assertIn (b 'This field is required' , response .data )
35
35
36
36
def test_functional (self ):
37
37
# Ensure form redirects and string is reversed.
38
38
response = self .app .post ('/' , data = dict (
39
39
reverse_string = "Real Python" ,), follow_redirects = True )
40
40
self .assertEqual (response .status_code , 200 )
41
- self .assertIn ('nohtyP laeR' , response .data )
41
+ self .assertIn (b 'nohtyP laeR' , response .data )
42
42
43
43
44
44
if __name__ == '__main__' :
0 commit comments