API Updates (stripe#728)

richardm-stripe · web-flow · commit fb0f7c0fd4ac · 2021-07-14T12:14:22.000-04:00
diff --git a/stripe/api_resources/quote.py b/stripe/api_resources/quote.py
@@ -13,6 +13,11 @@
 @custom_method("accept", http_verb="post")
 @custom_method("cancel", http_verb="post")
 @custom_method("finalize_quote", http_verb="post", http_path="finalize")
+@custom_method(
+    "list_computed_upfront_line_items",
+    http_verb="get",
+    http_path="computed_upfront_line_items",
+)
 @custom_method("list_line_items", http_verb="get", http_path="line_items")
 class Quote(CreateableAPIResource, ListableAPIResource, UpdateableAPIResource):
     OBJECT_NAME = "quote"
@@ -35,6 +40,12 @@ def finalize_quote(self, idempotency_key=None, **params):
         self.refresh_from(self.request("post", url, params, headers))
         return self
 
+    def list_computed_upfront_line_items(self, idempotency_key=None, **params):
+        url = self.instance_url() + "/computed_upfront_line_items"
+        headers = util.populate_headers(idempotency_key)
+        self.refresh_from(self.request("get", url, params, headers))
+        return self
+
     def list_line_items(self, idempotency_key=None, **params):
         url = self.instance_url() + "/line_items"
         headers = util.populate_headers(idempotency_key)
diff --git a/tests/api_resources/test_quote.py b/tests/api_resources/test_quote.py
@@ -95,6 +95,45 @@ def test_can_accept_classmethod(self, request_mock):
         )
         assert isinstance(resource, stripe.Quote)
 
+    def test_can_list_line_items(self, request_mock):
+        resources = stripe.Quote.list_line_items(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "get", "/v1/quotes/%s/line_items" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resources.data, list)
+        assert isinstance(resources.data[0], stripe.LineItem)
+
+    def test_can_list_line_items_classmethod(self, request_mock):
+        resources = stripe.Quote.list_line_items(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "get", "/v1/quotes/%s/line_items" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resources.data, list)
+        assert isinstance(resources.data[0], stripe.LineItem)
+
+    def test_can_list_computed_upfront_line_items(self, request_mock):
+        resources = stripe.Quote.list_computed_upfront_line_items(
+            TEST_RESOURCE_ID
+        )
+        request_mock.assert_requested(
+            "get",
+            "/v1/quotes/%s/computed_upfront_line_items" % TEST_RESOURCE_ID,
+        )
+        assert isinstance(resources.data, list)
+        assert isinstance(resources.data[0], stripe.LineItem)
+
+    def test_can_list_computed_upfront_line_items_classmethod(
+        self, request_mock
+    ):
+        resources = stripe.Quote.list_computed_upfront_line_items(
+            TEST_RESOURCE_ID
+        )
+        request_mock.assert_requested(
+            "get",
+            "/v1/quotes/%s/computed_upfront_line_items" % TEST_RESOURCE_ID,
+        )
+        assert isinstance(resources.data[0], stripe.LineItem)
+
     def test_can_pdf(self, setup_upload_api_base, request_mock):
         resource = stripe.Quote.retrieve(TEST_RESOURCE_ID)
         stream, _ = resource.pdf()
