Skip to content

Commit e4f054c

Browse files
committed
feat: Update the unit tests for use type changes
1 parent 7197f68 commit e4f054c

File tree

5 files changed

+9
-1
lines changed

5 files changed

+9
-1
lines changed

test/Unit/test_checks_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import lob_python
1616
from lob_python.api.checks_api import ChecksApi # noqa: E501
1717
from lob_python.model.check_editable import CheckEditable
18+
from lob_python.model.chk_use_type import ChkUseType
1819
from lob_python.model.mail_type import MailType
1920
from lob_python.model.sort_by3 import SortBy3
2021
from lob_python.model.include_model import IncludeModel
@@ -39,6 +40,7 @@ def setUp(self):
3940
to = "adr_fakeId2",
4041
bank_account = "bank_fakeId",
4142
amount = 100.0,
43+
use_type = ChkUseType("marketing")
4244
)
4345

4446
self.mock_list_of_checks = MagicMock(return_value={

test/Unit/test_intl_autocompletions_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def setUp(self):
4545
primary_line = "340 WATERSTONE PL SE",
4646
city = "AIRDRIE",
4747
state = "AB",
48-
country = CountryExtendedExpanded("CANADA"),
48+
country = CountryExtended("CA"),
4949
zip_code = "T4B 2G7",
5050
)
5151

test/Unit/test_letters_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from lob_python.api.letters_api import LettersApi # noqa: E501
1717
from lob_python.model.letter_editable import LetterEditable
1818
from lob_python.model.address_editable import AddressEditable
19+
from lob_python.model.ltr_use_type import LtrUseType
1920
from lob_python.model.mail_type import MailType
2021
from lob_python.model.sort_by3 import SortBy3
2122
from lob_python.model.metadata_model import MetadataModel
@@ -43,6 +44,7 @@ def setUp(self):
4344
_from = "adr_fakeId2",
4445
file = "https://s3-us-west-2.amazonaws.com/public.lob.com/assets/us_letter_1pg.pdf",
4546
color = True,
47+
use_type = LtrUseType("marketing")
4648
)
4749

4850
def test_letter_retrieve_error_handle(self):

test/Unit/test_postcards_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from lob_python.api.postcards_api import PostcardsApi # noqa: E501
1717
from lob_python.model.postcard_editable import PostcardEditable
1818
from lob_python.model.mail_type import MailType
19+
from lob_python.model.psc_use_type import PscUseType
1920
from lob_python.model.sort_by3 import SortBy3
2021
from lob_python.model.metadata_model import MetadataModel
2122
from lob_python.model.include_model import IncludeModel
@@ -43,6 +44,7 @@ def setUp(self):
4344
_from = "adr_fakeId2",
4445
front = "https://s3-us-west-2.amazonaws.com/public.lob.com/assets/templates/4x6_pc_template.pdf",
4546
back = "https://s3-us-west-2.amazonaws.com/public.lob.com/assets/templates/4x6_pc_template.pdf",
47+
use_type = PscUseType("marketing")
4648
)
4749

4850
def test_postcard_retrieve_error_handle(self):

test/Unit/test_self_mailers_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from lob_python.model.include_model import IncludeModel
2020
from lob_python.model.self_mailer_size import SelfMailerSize
2121
from lob_python.model.mail_type import MailType
22+
from lob_python.model.sfm_use_type import SfmUseType
2223
from lob_python.model.sort_by3 import SortBy3
2324
from lob_python.exceptions import UnauthorizedException, NotFoundException, ApiException
2425
from unittest.mock import Mock, MagicMock
@@ -39,6 +40,7 @@ def setUp(self):
3940
_from = "adr_fakeId2",
4041
inside = "https://s3.us-west-2.amazonaws.com/public.lob.com/assets/templates/self_mailers/6x18_sfm_inside.pdf",
4142
outside = "https://s3.us-west-2.amazonaws.com/public.lob.com/assets/templates/self_mailers/6x18_sfm_inside.pdf",
43+
use_type = SfmUseType("marketing")
4244
)
4345

4446
self.mock_list_of_self_mailers = MagicMock(return_value={

0 commit comments

Comments
 (0)