Skip to content

Commit fb17c04

Browse files
committed
Fix tests
1 parent 4ac794e commit fb17c04

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

test/custom_header_test.py

+8-10
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,11 @@ def test_custom_headers_set(self):
1616
sender = smarty.RequestsSender
1717
header = {'Test-User-Agent': 'Test-Agent', 'Test-Content-Type': 'Test-Type'}
1818
sender = smarty.CustomHeaderSender(header, sender)
19-
smartyrequest = smarty.Request()
20-
smartyrequest.url_prefix = "http://localhost"
21-
smartyrequest.payload = "This is the test content."
19+
smarty_request = smarty.Request()
20+
smarty_request.url_prefix = "http://localhost"
21+
smarty_request.payload = "This is the test content."
2222

23-
request = sender.build_request(smartyrequest)
23+
request = sender.build_request(smarty_request)
2424

2525
self.assertEqual('Test-Agent', request.headers['Test-User-Agent'])
2626
self.assertEqual('Test-Type', request.headers['Test-Content-Type'])
@@ -29,13 +29,11 @@ def test_custom_headers_used(self):
2929
sender = smarty.RequestsSender()
3030
header = {'User-Agent': 'Test-Agent', 'Content-Type': 'Test-Type'}
3131
sender = smarty.CustomHeaderSender(header, sender)
32-
smartyrequest = smarty.Request()
33-
smartyrequest.url_prefix = "http://localhost"
34-
smartyrequest.payload = "This is the test content."
32+
smarty_request = smarty.Request()
33+
smarty_request.url_prefix = "http://localhost"
34+
smarty_request.payload = "This is the test content."
3535

36-
request = sender.build_request(smartyrequest)
37-
38-
request = smarty.requests_sender.build_request(request)
36+
request = sender.build_request(smarty_request)
3937

4038
self.assertEqual('Test-Agent', request.headers['User-Agent'])
4139
self.assertEqual('Test-Type', request.headers['Content-Type'])

0 commit comments

Comments
 (0)