55import unittest
66import responses
77
8- try :
9- from unittest .mock import patch
10- except ImportError :
11- from mock import patch
12-
138from dynatademand .api import DemandAPIClient
149
1510BASE_HOST = "http://test-url.example"
@@ -24,7 +19,11 @@ def setUp(self):
2419 def test_get_line_item (self ):
2520 with open ('./tests/test_files/get_line_item.json' , 'r' ) as line_item_file :
2621 line_item_json = json .load (line_item_file )
27- responses .add (responses .GET , '{}/sample/v1/projects/1/lineItems/100' .format (BASE_HOST ), json = line_item_json , status = 200 )
22+ responses .add (
23+ responses .GET ,
24+ '{}/sample/v1/projects/1/lineItems/100' .format (BASE_HOST ),
25+ json = line_item_json ,
26+ status = 200 )
2827 self .api .get_line_item (1 , 100 )
2928 self .assertEqual (len (responses .calls ), 1 )
3029 self .assertEqual (responses .calls [0 ].response .json (), line_item_json )
@@ -42,7 +41,11 @@ def test_get_line_items(self):
4241 def test_get_line_item_detailed_report (self ):
4342 with open ('./tests/test_files/get_line_item_detailed_report.json' , 'r' ) as line_item_detailed_report_file :
4443 line_item_detailed_report_json = json .load (line_item_detailed_report_file )
45- responses .add (responses .GET , '{}/sample/v1/projects/1/lineItems/100/detailedReport' .format (BASE_HOST ), json = line_item_detailed_report_json , status = 200 )
44+ responses .add (
45+ responses .GET ,
46+ '{}/sample/v1/projects/1/lineItems/100/detailedReport' .format (BASE_HOST ),
47+ json = line_item_detailed_report_json ,
48+ status = 200 )
4649 self .api .get_line_item_detailed_report (1 , 100 )
4750 self .assertEqual (len (responses .calls ), 1 )
4851 self .assertEqual (responses .calls [0 ].response .json (), line_item_detailed_report_json )
0 commit comments