2222import unittest
2323
2424from transip import TransIP
25- from transip .v6 .objects import Invoice
25+ from transip .v6 .objects import Invoice , InvoiceItem
2626from tests .utils import load_responses_fixtures
2727
2828class InvoicesTest (unittest .TestCase ):
@@ -44,12 +44,24 @@ def test_list(self) -> None:
4444 invoices : List [Invoice ] = self .client .invoices .list () # type: ignore
4545 invoice : Invoice = invoices [0 ]
4646
47- assert len (invoices ) == 1
48- assert invoice .get_id () == "F0000.1911.0000.0004" # type: ignore
47+ self . assertEqual ( len (invoices ), 1 )
48+ self . assertEqual ( invoice .get_id (), "F0000.1911.0000.0004" ) # type: ignore
4949
5050 @responses .activate
5151 def test_get (self ) -> None :
5252 invoice_id : str = "F0000.1911.0000.0004"
5353 invoice : Invoice = self .client .invoices .get (invoice_id ) # type: ignore
5454
55- assert invoice .get_id () == "F0000.1911.0000.0004" # type: ignore
55+ self .assertEqual (invoice .get_id (), "F0000.1911.0000.0004" ) # type: ignore
56+
57+ @responses .activate
58+ def test_items_list (self ) -> None :
59+ invoice : Invoice = self .client .invoices .get ("F0000.1911.0000.0004" ) # type: ignore
60+ items : List [InvoiceItem ] = invoice .items .list () # type: ignore
61+
62+ self .assertEqual (len (items ), 1 )
63+ self .assertEqual (items [0 ].product , "Big Storage Disk 2000 GB" ) # type: ignore
64+
65+ # Expect the get_id() method to return None as the invoice items don't
66+ # have a specific ID attribute.
67+ self .assertIsNone (items [0 ].get_id ()) # type: ignore
0 commit comments