Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add InvoiceLineItem class #452

Merged
merged 1 commit into from
Jul 25, 2018
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions stripe/api_resources/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
from stripe.api_resources.file_upload import FileUpload
from stripe.api_resources.invoice import Invoice
from stripe.api_resources.invoice_item import InvoiceItem
from stripe.api_resources.invoice_line_item import InvoiceLineItem
from stripe.api_resources.issuer_fraud_record import IssuerFraudRecord
from stripe.api_resources.login_link import LoginLink
from stripe.api_resources.order import Order
Expand Down
7 changes: 7 additions & 0 deletions stripe/api_resources/invoice_line_item.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
from __future__ import absolute_import, division, print_function

from stripe.stripe_object import StripeObject


class InvoiceLineItem(StripeObject):
OBJECT_NAME = 'line_item'
2 changes: 2 additions & 0 deletions stripe/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,8 @@ def load_object_classes():
api_resources.FileUpload.OBJECT_NAME: api_resources.FileUpload,
api_resources.Invoice.OBJECT_NAME: api_resources.Invoice,
api_resources.InvoiceItem.OBJECT_NAME: api_resources.InvoiceItem,
api_resources.InvoiceLineItem.OBJECT_NAME:
api_resources.InvoiceLineItem,
api_resources.IssuerFraudRecord.OBJECT_NAME:
api_resources.IssuerFraudRecord,
api_resources.LoginLink.OBJECT_NAME: api_resources.LoginLink,
Expand Down
69 changes: 69 additions & 0 deletions tests/api_resources/test_invoice_item.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
from __future__ import absolute_import, division, print_function

import stripe


TEST_RESOURCE_ID = 'ii_123'


class TestInvoiceItem(object):
def test_is_listable(self, request_mock):
resources = stripe.InvoiceItem.list()
request_mock.assert_requested(
'get',
'/v1/invoiceitems'
)
assert isinstance(resources.data, list)
assert isinstance(resources.data[0], stripe.InvoiceItem)

def test_is_retrievable(self, request_mock):
resource = stripe.InvoiceItem.retrieve(TEST_RESOURCE_ID)
request_mock.assert_requested(
'get',
'/v1/invoiceitems/%s' % TEST_RESOURCE_ID
)
assert isinstance(resource, stripe.InvoiceItem)

def test_is_creatable(self, request_mock):
resource = stripe.InvoiceItem.create(
customer='cus_123',
amount=123,
currency='usd'
)
request_mock.assert_requested(
'post',
'/v1/invoiceitems'
)
assert isinstance(resource, stripe.InvoiceItem)

def test_is_saveable(self, request_mock):
resource = stripe.InvoiceItem.retrieve(TEST_RESOURCE_ID)
resource.metadata['key'] = 'value'
resource.save()
request_mock.assert_requested(
'post',
'/v1/invoiceitems/%s' % resource.id
)

def test_is_modifiable(self, request_mock):
resource = stripe.InvoiceItem.modify(
TEST_RESOURCE_ID,
metadata={'key': 'value'}
)
request_mock.assert_requested(
'post',
'/v1/invoiceitems/%s' % TEST_RESOURCE_ID
)
assert isinstance(resource, stripe.InvoiceItem)

def test_is_deletable(self, request_mock):
resource = stripe.InvoiceItem.retrieve(TEST_RESOURCE_ID)
# Unfortunately stripe-mock will return a resource with a different
# ID, so we need to store the original ID for the request assertion
resource_id = resource.id
resource.delete()
request_mock.assert_requested(
'delete',
'/v1/invoiceitems/%s' % resource_id
)
assert resource.deleted is True
13 changes: 13 additions & 0 deletions tests/api_resources/test_invoice_line_item.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
from __future__ import absolute_import, division, print_function

import stripe


TEST_INVOICE_ID = 'in_123'


class TestInvoiceLineItem(object):
def test_deserialize(self, request_mock):
invoice = stripe.Invoice.retrieve(TEST_INVOICE_ID)
assert isinstance(invoice.lines.data, list)
assert isinstance(invoice.lines.data[0], stripe.InvoiceLineItem)