feat: Airwallex 发卡管理后台完整实现
- 后端: FastAPI + SQLAlchemy + SQLite, JWT认证, 代理支持的AirwallexClient - 前端: React 18 + Vite + Ant Design 5, 中文界面 - 功能: 卡片管理, 持卡人管理, 交易记录, API令牌, 系统设置, 审计日志 - 第三方API: X-API-Key认证, 权限控制 - Docker部署: docker-compose编排前后端
This commit is contained in:
zqq61
168
airwallex-sdk/tests/test_invoice.py
Normal file
168
airwallex-sdk/tests/test_invoice.py
Normal file
@@ -0,0 +1,168 @@
|
||||
"""
|
||||
Tests for the Airwallex SDK invoice API.
|
||||
"""
|
||||
import unittest
|
||||
from unittest.mock import patch, MagicMock
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from airwallex import AirwallexClient
|
||||
from airwallex.models.invoice import (
|
||||
Invoice,
|
||||
InvoiceItem,
|
||||
InvoicePreviewRequest,
|
||||
InvoicePreviewResponse
|
||||
)
|
||||
|
||||
|
||||
class TestInvoiceAPI(unittest.TestCase):
|
||||
"""Tests for the Invoice API."""
|
||||
|
||||
def setUp(self):
|
||||
"""Set up test fixtures."""
|
||||
self.client_id = "test_client_id"
|
||||
self.api_key = "test_api_key"
|
||||
|
||||
# Create a client with authentication mocked
|
||||
patcher = patch.object(AirwallexClient, 'authenticate')
|
||||
self.mock_auth = patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
|
||||
self.client = AirwallexClient(client_id=self.client_id, api_key=self.api_key)
|
||||
self.client._token = "test_token"
|
||||
self.client._token_expiry = datetime.now() + timedelta(minutes=30)
|
||||
|
||||
@patch('httpx.Client.request')
|
||||
def test_list_invoices(self, mock_request):
|
||||
"""Test listing invoices."""
|
||||
# Mock response
|
||||
mock_response = MagicMock()
|
||||
mock_response.status_code = 200
|
||||
mock_response.json.return_value = {
|
||||
"items": [
|
||||
{
|
||||
"id": "inv_test123",
|
||||
"customer_id": "cus_test456",
|
||||
"currency": "USD",
|
||||
"total_amount": 100.00,
|
||||
"status": "PAID",
|
||||
"period_start_at": "2023-01-01T00:00:00Z",
|
||||
"period_end_at": "2023-02-01T00:00:00Z",
|
||||
"created_at": "2023-01-01T00:00:00Z"
|
||||
}
|
||||
],
|
||||
"has_more": False
|
||||
}
|
||||
mock_request.return_value = mock_response
|
||||
|
||||
# Call list method
|
||||
invoices = self.client.invoice.list()
|
||||
|
||||
# Check request was made correctly
|
||||
mock_request.assert_called_once()
|
||||
args, kwargs = mock_request.call_args
|
||||
self.assertEqual(args[0], "GET")
|
||||
self.assertTrue(args[1].endswith("/invoices"))
|
||||
|
||||
# Check response parsing
|
||||
self.assertEqual(len(invoices), 1)
|
||||
self.assertIsInstance(invoices[0], Invoice)
|
||||
self.assertEqual(invoices[0].id, "inv_test123")
|
||||
self.assertEqual(invoices[0].currency, "USD")
|
||||
self.assertEqual(invoices[0].total_amount, 100.00)
|
||||
|
||||
@patch('httpx.Client.request')
|
||||
def test_get_invoice(self, mock_request):
|
||||
"""Test fetching a single invoice."""
|
||||
# Mock response
|
||||
mock_response = MagicMock()
|
||||
mock_response.status_code = 200
|
||||
mock_response.json.return_value = {
|
||||
"id": "inv_test123",
|
||||
"customer_id": "cus_test456",
|
||||
"currency": "USD",
|
||||
"total_amount": 100.00,
|
||||
"status": "PAID",
|
||||
"period_start_at": "2023-01-01T00:00:00Z",
|
||||
"period_end_at": "2023-02-01T00:00:00Z",
|
||||
"created_at": "2023-01-01T00:00:00Z"
|
||||
}
|
||||
mock_request.return_value = mock_response
|
||||
|
||||
# Call fetch method
|
||||
invoice = self.client.invoice.fetch("inv_test123")
|
||||
|
||||
# Check request was made correctly
|
||||
mock_request.assert_called_once()
|
||||
args, kwargs = mock_request.call_args
|
||||
self.assertEqual(args[0], "GET")
|
||||
self.assertTrue(args[1].endswith("/invoices/inv_test123"))
|
||||
|
||||
# Check response parsing
|
||||
self.assertIsInstance(invoice, Invoice)
|
||||
self.assertEqual(invoice.id, "inv_test123")
|
||||
self.assertEqual(invoice.currency, "USD")
|
||||
self.assertEqual(invoice.total_amount, 100.00)
|
||||
|
||||
@patch('httpx.Client.request')
|
||||
def test_preview_invoice(self, mock_request):
|
||||
"""Test previewing an invoice."""
|
||||
# Mock response
|
||||
mock_response = MagicMock()
|
||||
mock_response.status_code = 200
|
||||
mock_response.json.return_value = {
|
||||
"customer_id": "cus_test456",
|
||||
"currency": "USD",
|
||||
"total_amount": 100.00,
|
||||
"created_at": "2023-01-01T00:00:00Z",
|
||||
"items": [
|
||||
{
|
||||
"id": "item_test123",
|
||||
"invoice_id": "inv_test123",
|
||||
"amount": 100.00,
|
||||
"currency": "USD",
|
||||
"period_start_at": "2023-01-01T00:00:00Z",
|
||||
"period_end_at": "2023-02-01T00:00:00Z",
|
||||
"price": {
|
||||
"id": "pri_test123",
|
||||
"name": "Test Price",
|
||||
"active": True,
|
||||
"currency": "USD",
|
||||
"product_id": "prod_test123",
|
||||
"pricing_model": "flat"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
mock_request.return_value = mock_response
|
||||
|
||||
# Create preview request
|
||||
preview_request = InvoicePreviewRequest(
|
||||
customer_id="cus_test456",
|
||||
items=[
|
||||
InvoicePreviewRequest.SubscriptionItem(
|
||||
price_id="pri_test123",
|
||||
quantity=1
|
||||
)
|
||||
]
|
||||
)
|
||||
|
||||
# Call preview method
|
||||
preview = self.client.invoice.preview(preview_request)
|
||||
|
||||
# Check request was made correctly
|
||||
mock_request.assert_called_once()
|
||||
args, kwargs = mock_request.call_args
|
||||
self.assertEqual(args[0], "POST")
|
||||
self.assertTrue(args[1].endswith("/invoices/preview"))
|
||||
|
||||
# Check response parsing
|
||||
self.assertIsInstance(preview, InvoicePreviewResponse)
|
||||
self.assertEqual(preview.customer_id, "cus_test456")
|
||||
self.assertEqual(preview.currency, "USD")
|
||||
self.assertEqual(preview.total_amount, 100.00)
|
||||
self.assertEqual(len(preview.items), 1)
|
||||
self.assertEqual(preview.items[0].amount, 100.00)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user