File: test_file_upload.py

package info (click to toggle)
python-stripe 13.2.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 14,476 kB
  • sloc: python: 187,843; makefile: 13; sh: 9
file content (51 lines) | stat: -rw-r--r-- 1,839 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
from __future__ import absolute_import, division, print_function

import tempfile

import stripe
from stripe._file import File
from stripe._multipart_data_generator import MultipartDataGenerator


TEST_RESOURCE_ID = "file_123"


class TestFileUpload(object):
    def test_is_listable(self, http_client_mock, stripe_mock_stripe_client):
        resources = stripe_mock_stripe_client.files.list()
        http_client_mock.assert_requested("get", path="/v1/files")
        assert isinstance(resources.data, list)
        assert isinstance(resources.data[0], File)

    def test_is_retrievable(self, http_client_mock, stripe_mock_stripe_client):
        resource = stripe_mock_stripe_client.files.retrieve(TEST_RESOURCE_ID)
        http_client_mock.assert_requested(
            "get", path="/v1/files/%s" % TEST_RESOURCE_ID
        )
        assert isinstance(resource, File)

    def test_is_creatable(
        self,
        file_stripe_mock_stripe_client,
        http_client_mock,
    ):
        MultipartDataGenerator._initialize_boundary = lambda self: 1234567890
        test_file = tempfile.TemporaryFile()

        # We create a new client here instead of re-using the stripe_mock_stripe_client fixture
        # because stripe_mock_stripe_client overrides the "api" base address, which we want to
        # be empty for this test.
        resource = file_stripe_mock_stripe_client.files.create(
            params={
                "purpose": "dispute_evidence",
                "file": test_file,
                "file_link_data": {"create": True},
            }
        )
        http_client_mock.assert_requested(
            "post",
            api_base=stripe.upload_api_base,
            path="/v1/files",
            content_type="multipart/form-data; boundary=1234567890",
        )
        assert isinstance(resource, File)