11import unittest
22import os
33import lob
4+ import pytest
45
56
67class TestAddressFunctions (unittest .TestCase ):
78 def setUp (self ):
89 lob .api_key = os .environ .get ('LOB_API_KEY' )
910
11+
12+ def test_create_address (self ):
13+ address = lob .Address .create (
14+ name = 'Lob' ,
15+ address_line1 = '185 Berry Street' ,
16+ address_line2 = 'Suite 1510' ,
17+ address_city = 'San Francisco' ,
18+ address_zip = '94017' ,
19+ address_state = 'CA' ,
20+ address_country = 'US'
21+ )
22+
23+ self .assertTrue (isinstance (address , lob .Address ))
24+ self .assertEqual (address .name , 'LOB' )
25+
1026 def test_list_addresses (self ):
1127 addresses = lob .Address .list ()
1228 self .assertTrue (isinstance (addresses .data [0 ], lob .Address ))
@@ -24,21 +40,8 @@ def test_list_addresses_limit(self):
2440 self .assertEqual (len (addresses .data ), 2 )
2541
2642 def test_list_address_fail (self ):
27- self .assertRaises (lob .error .InvalidRequestError , lob .Address .list , limit = 1000 )
28-
29- def test_create_address (self ):
30- address = lob .Address .create (
31- name = 'Lob' ,
32- address_line1 = '185 Berry Street' ,
33- address_line2 = 'Suite 1510' ,
34- address_city = 'San Francisco' ,
35- address_zip = '94017' ,
36- address_state = 'CA' ,
37- address_country = 'US'
38- )
39-
40- self .assertTrue (isinstance (address , lob .Address ))
41- self .assertEqual (address .name , 'LOB' )
43+ with pytest .raises (lob .error .InvalidRequestError ):
44+ lob .Address .list (foobar = 1000 )
4245
4346 def test_create_addresss_fail (self ):
4447 self .assertRaises (lob .error .InvalidRequestError , lob .Address .create )
@@ -51,6 +54,15 @@ def test_retrieve_address_fail(self):
5154 self .assertRaises (lob .error .InvalidRequestError , lob .Address .retrieve , id = 'test' )
5255
5356 def test_delete_address (self ):
54- addr = lob .Address .list ().data [0 ].id
55- delAddr = lob .Address .delete (id = addr )
56- self .assertEqual (addr , delAddr .id )
57+ address = lob .Address .create (
58+ name = 'Lob' ,
59+ address_line1 = '185 Berry Street' ,
60+ address_line2 = 'Suite 1510' ,
61+ address_city = 'San Francisco' ,
62+ address_zip = '94017' ,
63+ address_state = 'CA' ,
64+ address_country = 'US'
65+ )
66+
67+ delAddr = lob .Address .delete (id = address .id )
68+ self .assertEqual (address .id , delAddr .id )
0 commit comments