|
| 1 | +# |
| 2 | +# Copyright (c) 2021 Airbyte, Inc., all rights reserved. |
| 3 | +# |
| 4 | + |
| 5 | + |
| 6 | +from airbyte_cdk.models import SyncMode |
| 7 | +from source_stripe.streams import CheckoutSessionsLineItems |
| 8 | + |
| 9 | + |
| 10 | +def test_missed_id_child_stream(requests_mock): |
| 11 | + |
| 12 | + session_id_missed = "cs_test_a165K4wNihuJlp2u3tknuohrvjAxyXFUB7nxZH3lwXRKJsadNEvIEWMUJ9" |
| 13 | + session_id_exists = "cs_test_a1RjRHNyGUQOFVF3OkL8V8J0lZUASyVoCtsnZYG74VrBv3qz4245BLA1BP" |
| 14 | + |
| 15 | + response_sessions = { |
| 16 | + "data": [{"id": session_id_missed}, {"id": session_id_exists}], |
| 17 | + "has_more": False, |
| 18 | + "object": "list", |
| 19 | + "url": "/v1/checkout/sessions", |
| 20 | + } |
| 21 | + |
| 22 | + response_sessions_line_items = { |
| 23 | + "data": [{"id": "li_1JpAUUIEn5WyEQxnfGJT5MbL"}], |
| 24 | + "has_more": False, |
| 25 | + "object": "list", |
| 26 | + "url": "/v1/checkout/sessions/{}/line_items".format(session_id_exists), |
| 27 | + } |
| 28 | + |
| 29 | + response_error = { |
| 30 | + "error": { |
| 31 | + "code": "resource_missing", |
| 32 | + "doc_url": "https://stripe.com/docs/error-codes/resource-missing", |
| 33 | + "message": "No such checkout session: '{}'".format(session_id_missed), |
| 34 | + "param": "session", |
| 35 | + "type": "invalid_request_error", |
| 36 | + } |
| 37 | + } |
| 38 | + |
| 39 | + requests_mock.get("https://api.stripe.com/v1/checkout/sessions", json=response_sessions) |
| 40 | + requests_mock.get( |
| 41 | + "https://api.stripe.com/v1/checkout/sessions/{}/line_items".format(session_id_exists), json=response_sessions_line_items |
| 42 | + ) |
| 43 | + requests_mock.get( |
| 44 | + "https://api.stripe.com/v1/checkout/sessions/{}/line_items".format(session_id_missed), json=response_error, status_code=404 |
| 45 | + ) |
| 46 | + |
| 47 | + stream = CheckoutSessionsLineItems(start_date=None, account_id=None) |
| 48 | + records = list(stream.read_records(sync_mode=SyncMode.full_refresh)) |
| 49 | + assert len(records) == 1 |
0 commit comments