63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
import unittest
|
|
|
|
from api.services.memoir_images.schema import (
|
|
IMAGE_STATUS_FAILED,
|
|
IMAGE_STATUS_PENDING,
|
|
normalize_image_asset,
|
|
)
|
|
|
|
|
|
class MemoirImageSchemaTest(unittest.TestCase):
|
|
def test_normalize_image_asset_coerces_invalid_status_to_failed(self):
|
|
asset = normalize_image_asset(
|
|
{
|
|
"index": 0,
|
|
"placeholder": "{{IMAGE:南方小镇的青石板路}}",
|
|
"description": "南方小镇的青石板路",
|
|
"status": "mystery",
|
|
"url": "https://cos.example.com/0.png",
|
|
}
|
|
)
|
|
|
|
self.assertEqual(asset["status"], IMAGE_STATUS_FAILED)
|
|
self.assertEqual(asset["error"], "invalid image status: mystery")
|
|
|
|
def test_normalize_image_asset_requires_placeholder_and_description(self):
|
|
asset = normalize_image_asset(
|
|
{
|
|
"index": 0,
|
|
"status": IMAGE_STATUS_PENDING,
|
|
}
|
|
)
|
|
|
|
self.assertIsNone(asset)
|
|
|
|
def test_normalize_image_asset_preserves_retryable_for_failed_assets(self):
|
|
asset = normalize_image_asset(
|
|
{
|
|
"index": 0,
|
|
"placeholder": "{{IMAGE:南方小镇的青石板路}}",
|
|
"description": "南方小镇的青石板路",
|
|
"status": IMAGE_STATUS_FAILED,
|
|
"error": "upload denied",
|
|
"retryable": False,
|
|
}
|
|
)
|
|
|
|
self.assertEqual(asset["status"], IMAGE_STATUS_FAILED)
|
|
self.assertFalse(asset["retryable"])
|
|
|
|
def test_normalize_image_asset_clears_retryable_for_non_failed_assets(self):
|
|
asset = normalize_image_asset(
|
|
{
|
|
"index": 0,
|
|
"placeholder": "{{IMAGE:南方小镇的青石板路}}",
|
|
"description": "南方小镇的青石板路",
|
|
"status": IMAGE_STATUS_PENDING,
|
|
"retryable": True,
|
|
}
|
|
)
|
|
|
|
self.assertEqual(asset["status"], IMAGE_STATUS_PENDING)
|
|
self.assertIsNone(asset["retryable"])
|