card_credential_test.py

#   Copyright 2026 UCP Authors
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

"""Tests for Card Credential in UCP SDK Server."""

from absl.testing import absltest
import integration_test_utils
from ucp_sdk.models.schemas.shopping import checkout as checkout
from ucp_sdk.models.schemas.shopping.payment import (
  Payment,
)


# Rebuild models to resolve forward references
checkout.Checkout.model_rebuild(_types_namespace={"Payment": Payment})


class CardCredentialTest(integration_test_utils.IntegrationTestBase):
  """Tests for Card Credential.

  Validated Paths:
  - POST /checkout-sessions/{id}/complete
  """

  def test_card_credential_payment(self) -> None:
    """Test successful payment processing with raw Card Credential.

    Given a ready-to-complete checkout session,
    When a completion request is made using a valid CardCredential,
    Then the request should succeed with status 200.
    """
    response_json = self.create_checkout_session()
    checkout_id = checkout.Checkout(**response_json).id

    payment_instrument = {
      "id": "instr_card",
      "handler_id": "mock_payment_handler",
      "type": "card",
      "display": {
        "brand": "Visa",
        "last_digits": "1111",
      },
      "credential": {
        "type": "card",
        "card_number_type": "fpan",
        "number": "4242424242424242",
        "expiry_month": 12,
        "expiry_year": 2030,
        "cvc": "123",
        "name": "John Doe",
      },
    }
    payment_payload = {
      "payment": {"instruments": [payment_instrument]},
      "risk_signals": {},
    }

    response = self.client.post(
      self.get_shopping_url(f"/checkout-sessions/{checkout_id}/complete"),
      json=payment_payload,
      headers=integration_test_utils.get_headers(),
    )

    self.assert_response_status(response, 200)
    self.assertEqual(
      response.json().get("status"),
      "completed",
      msg="Checkout status not 'completed'",
    )


if __name__ == "__main__":
  absltest.main()