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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
|
import json
from typing import Callable
import pytest
import stripe
from stripe import ThinEvent
from stripe.events._v1_billing_meter_error_report_triggered_event import (
V1BillingMeterErrorReportTriggeredEvent,
)
from tests.test_webhook import DUMMY_WEBHOOK_SECRET, generate_header
EventParser = Callable[[str], ThinEvent]
class TestV2Event(object):
@pytest.fixture(scope="function")
def v2_payload_no_data(self):
return json.dumps(
{
"id": "evt_234",
"object": "v2.core.event",
"type": "v1.billing.meter.error_report_triggered",
"livemode": True,
"created": "2022-02-15T00:27:45.330Z",
"related_object": {
"id": "mtr_123",
"type": "billing.meter",
"url": "/v1/billing/meters/mtr_123",
"stripe_context": "acct_123",
},
"reason": {
"id": "foo",
"idempotency_key": "bar",
},
}
)
@pytest.fixture(scope="function")
def v2_payload_with_data(self):
return json.dumps(
{
"id": "evt_234",
"object": "v2.core.event",
"type": "v1.billing.meter.error_report_triggered",
"livemode": False,
"created": "2022-02-15T00:27:45.330Z",
"context": "acct_123",
"related_object": {
"id": "mtr_123",
"type": "billing.meter",
"url": "/v1/billing/meters/mtr_123",
},
"data": {
"reason": {
"error_count": 1,
}
},
}
)
@pytest.fixture(scope="function")
def stripe_client(self, http_client_mock):
return stripe.StripeClient(
api_key="keyinfo_test_123",
stripe_context="wksp_123",
http_client=http_client_mock.get_mock_http_client(),
)
@pytest.fixture(scope="function")
def parse_thin_event(
self, stripe_client: stripe.StripeClient
) -> EventParser:
"""
helper to simplify parsing and validating events given a payload
returns a function that has the client pre-bound
"""
def _parse_thin_event(payload: str):
return stripe_client.parse_thin_event(
payload, generate_header(payload=payload), DUMMY_WEBHOOK_SECRET
)
return _parse_thin_event
def test_parses_thin_event(
self, parse_thin_event: EventParser, v2_payload_no_data: str
):
event = parse_thin_event(v2_payload_no_data)
assert isinstance(event, ThinEvent)
assert event.id == "evt_234"
assert event.related_object
assert event.related_object.id == "mtr_123"
assert event.reason
assert event.reason.id == "foo"
def test_parses_thin_event_with_data(
self, parse_thin_event: EventParser, v2_payload_with_data: str
):
event = parse_thin_event(v2_payload_with_data)
assert isinstance(event, ThinEvent)
assert not hasattr(event, "data")
assert event.reason is None
def test_validates_signature(
self, stripe_client: stripe.StripeClient, v2_payload_no_data
):
with pytest.raises(stripe.error.SignatureVerificationError):
stripe_client.parse_thin_event(
v2_payload_no_data, "bad header", DUMMY_WEBHOOK_SECRET
)
def test_v2_events_data_type(self, http_client_mock, v2_payload_with_data):
method = "get"
path = "/v2/core/events/evt_123"
http_client_mock.stub_request(
method,
path=path,
rbody=v2_payload_with_data,
rcode=200,
rheaders={},
)
client = stripe.StripeClient(
api_key="keyinfo_test_123",
http_client=http_client_mock.get_mock_http_client(),
)
event = client.v2.core.events.retrieve("evt_123")
http_client_mock.assert_requested(
method,
api_base=stripe.DEFAULT_API_BASE,
path=path,
api_key="keyinfo_test_123",
)
assert event.id is not None
assert isinstance(event, V1BillingMeterErrorReportTriggeredEvent)
assert event.data is not None
assert isinstance(
event.data,
V1BillingMeterErrorReportTriggeredEvent.V1BillingMeterErrorReportTriggeredEventData,
)
assert event.data.reason.error_count == 1
|