1004 lines
57 KiB
Python
1004 lines
57 KiB
Python
|
import base64
|
|||
|
|
import json
|
|||
|
|
import ssl
|
|||
|
|
import sys
|
|||
|
|
import tempfile
|
|||
|
|
import unittest
|
|||
|
|
import urllib.error
|
|||
|
|
import zipfile
|
|||
|
|
from io import BytesIO
|
|||
|
|
from pathlib import Path
|
|||
|
|
from unittest.mock import patch
|
|||
|
|
|
|||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|||
|
|
sys.path.insert(0, str(ROOT / 'server'))
|
|||
|
|
import server as app
|
|||
|
|
from analysis import ensure_experience_solution
|
|||
|
|
|
|||
|
|
|
|||
|
|
def sample_report():
|
|||
|
|
return {
|
|||
|
|
'facts': {
|
|||
|
|
'product': {'title': '宿舍六件套', 'brand': '测试品牌', 'category': '床品'},
|
|||
|
|
'attributes': [
|
|||
|
|
{'name': '材质', 'value': '全棉'},
|
|||
|
|
{'name': '件数', 'value': '六件'},
|
|||
|
|
],
|
|||
|
|
},
|
|||
|
|
'experienceSolution': {
|
|||
|
|
'newProductPlans': {'plans': [{
|
|||
|
|
'name': '规格透明款',
|
|||
|
|
'whyThisPlan': '减少选错',
|
|||
|
|
'productAction': '保持六件套结构',
|
|||
|
|
'pageAction': '前置尺寸说明',
|
|||
|
|
'sellingPoints': [{'slogan': '六件配齐', 'consumerValue': '一次购齐'}],
|
|||
|
|
}]},
|
|||
|
|
'visualCommerce': {'items': [
|
|||
|
|
{'imageRole': f'主图任务{i}', 'visualSignals': ['六件配齐'], 'nextAction': '主体完整'}
|
|||
|
|
for i in range(1, 8)
|
|||
|
|
]},
|
|||
|
|
'detailCommerce': {'contentGroups': [
|
|||
|
|
{'type': f'证明任务{i}', 'visualSignals': ['全棉'], 'nextAction': '参数可核验'}
|
|||
|
|
for i in range(1, 9)
|
|||
|
|
]},
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ImageGenerationFlowTest(unittest.TestCase):
|
|||
|
|
def setUp(self):
|
|||
|
|
# Routing tests use fake URLs/providers. Exercise real byte transport
|
|||
|
|
# separately in test_reference_generation_contract.py.
|
|||
|
|
transport=patch.object(app, '_reference_data_url', side_effect=lambda value:value)
|
|||
|
|
transport.start()
|
|||
|
|
self.addCleanup(transport.stop)
|
|||
|
|
|
|||
|
|
def test_image_model_is_independent_and_required(self):
|
|||
|
|
with patch('ai_client.load_config', return_value={'api_key': 'test', 'model': 'text-only', 'image_model': '', '_config_error': ''}):
|
|||
|
|
with self.assertRaisesRegex(RuntimeError, '独立的生图模型'):
|
|||
|
|
app.image_generate('test')
|
|||
|
|
|
|||
|
|
def test_configuration_page_contains_both_models(self):
|
|||
|
|
page = app.control_page()
|
|||
|
|
self.assertIn('分析渠道', page)
|
|||
|
|
self.assertIn('生图渠道', page)
|
|||
|
|
self.assertIn('imageBase', page)
|
|||
|
|
self.assertIn('imageKey', page)
|
|||
|
|
self.assertIn('imageModel', page)
|
|||
|
|
prompt_page = app.prompt_page()
|
|||
|
|
self.assertIn('当前真实联动', prompt_page)
|
|||
|
|
self.assertIn('image_generation', prompt_page)
|
|||
|
|
self.assertIn('{prompt}', prompt_page)
|
|||
|
|
|
|||
|
|
def test_configuration_page_explains_real_image_workflow_and_parameters(self):
|
|||
|
|
config = {
|
|||
|
|
'api_base': 'https://text.example/v1', 'api_key': 'analysis-secret', 'model': 'text-model',
|
|||
|
|
'image_api_base': '', 'image_api_key': '', 'image_model': 'image-model',
|
|||
|
|
'image_path': '/images/generations', 'timeout': 120, 'retries': 2, '_config_error': '',
|
|||
|
|
}
|
|||
|
|
with patch.object(app, 'load_config', return_value=config), patch.object(app, 'configured', return_value=False):
|
|||
|
|
page = app.control_page()
|
|||
|
|
for text in (
|
|||
|
|
'生图工作流 · 研发速览', '提示词工作流程与参数解释', '_load_report_for_generation', 'build_generation_slots',
|
|||
|
|
'resolve_reference_images', 'build_image_prompt + merge', 'image_generate', 'initPlanActions', 'refreshPromptPreview', 'pollJob',
|
|||
|
|
'最终提示词怎样组成', '业务请求参数', '生图接口参数与返回',
|
|||
|
|
'主图 5 位 + 详情前 6 位', 'main:1', '{prompt}', 'image_models_path', 'jobId / statusUrl', 'GET status',
|
|||
|
|
'https://text.example/v1/images/generations', '各字段',
|
|||
|
|
):
|
|||
|
|
self.assertIn(text, page)
|
|||
|
|
self.assertIn('class="ok">已配置 image-model', page)
|
|||
|
|
# The API contract table is intentionally visible on first load so an
|
|||
|
|
# engineer can inspect the request/response fields without another
|
|||
|
|
# click (the business-request table remains visible as well).
|
|||
|
|
self.assertIn('details class="tech-detail" open><summary>生图接口参数与返回', page)
|
|||
|
|
self.assertNotIn('analysis-secret', page)
|
|||
|
|
|
|||
|
|
def test_image_channel_can_override_analysis_provider(self):
|
|||
|
|
from ai_client import image_channel
|
|||
|
|
channel = image_channel({'api_base': 'https://text.example/v1', 'api_key': 'text-key',
|
|||
|
|
'image_api_base': 'https://image.example/v1', 'image_api_key': 'image-key',
|
|||
|
|
'image_models_path': '/image-models'})
|
|||
|
|
self.assertEqual('https://image.example/v1', channel['api_base'])
|
|||
|
|
self.assertEqual('image-key', channel['api_key'])
|
|||
|
|
self.assertEqual('/image-models', channel['models_path'])
|
|||
|
|
|
|||
|
|
def test_image_quality_is_sent_per_request(self):
|
|||
|
|
import ai_client
|
|||
|
|
config = {
|
|||
|
|
'api_base': 'https://image.example/v1', 'api_key': 'image-key',
|
|||
|
|
'image_model': 'gpt-image-2.5-flare', 'image_path': '/images/generations',
|
|||
|
|
'image_quality': 'low', 'timeout': 120, 'retries': 0, '_config_error': '',
|
|||
|
|
}
|
|||
|
|
response = {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
with patch.object(ai_client, 'image_channel', return_value=config), \
|
|||
|
|
patch.object(ai_client, '_do_post_path', return_value=response) as post:
|
|||
|
|
actual = ai_client.image_generate('test', quality='high')
|
|||
|
|
self.assertEqual(response, actual)
|
|||
|
|
self.assertEqual('high', post.call_args.args[2]['quality'])
|
|||
|
|
|
|||
|
|
def test_comparison_models_keep_independent_quality(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
job_id = 'img_test_model_quality_comparison'
|
|||
|
|
calls = []
|
|||
|
|
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
calls.append(kwargs)
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
|
|||
|
|
config = {
|
|||
|
|
'api_key': 'image-key', 'model': 'text-model', 'image_model': 'flare',
|
|||
|
|
'image_models': [
|
|||
|
|
{'id': 'flare', 'label': 'Flare'},
|
|||
|
|
{'id': 'sunburst', 'label': 'Sunburst'},
|
|||
|
|
],
|
|||
|
|
'_config_error': '',
|
|||
|
|
}
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'load_config', return_value=config), \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', side_effect=lambda *args, **kwargs: {
|
|||
|
|
'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png',
|
|||
|
|
}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['main'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1'], 'fissionPattern': False,
|
|||
|
|
'referenceImages': ['data:image/png;base64,AAAA'],
|
|||
|
|
'imageModels': [
|
|||
|
|
{'id': 'flare', 'quality': 'medium'},
|
|||
|
|
{'id': 'sunburst', 'quality': 'high'},
|
|||
|
|
],
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertEqual(['high', 'medium'], sorted(item.get('quality') for item in calls))
|
|||
|
|
self.assertEqual({'flare': 'medium', 'sunburst': 'high'}, {
|
|||
|
|
item['model']: item['quality'] for item in state['results']
|
|||
|
|
})
|
|||
|
|
self.assertEqual(['medium', 'high'], [item['quality'] for item in state['comparisonModels']])
|
|||
|
|
|
|||
|
|
def test_report_requires_count_confirmation_before_generation(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
self.assertIn('确认生成：${countText(counts)}', script)
|
|||
|
|
self.assertIn('window.confirm(`确认开始生成？', script)
|
|||
|
|
self.assertIn('主图：${counts.main} 张', script)
|
|||
|
|
self.assertIn('详情图：${counts.detail} 张', script)
|
|||
|
|
self.assertIn('产品开品说明', script)
|
|||
|
|
self.assertIn('主图生成任务', script)
|
|||
|
|
self.assertIn('详情图生成任务', script)
|
|||
|
|
self.assertIn('generation-task-card', script)
|
|||
|
|
self.assertNotIn('data-gen-count', script)
|
|||
|
|
self.assertIn('生成记录', script)
|
|||
|
|
self.assertIn('/api/image-jobs', script)
|
|||
|
|
self.assertIn('job.planName', script)
|
|||
|
|
self.assertIn("brand.innerHTML='<b>三笙AI</b>", script)
|
|||
|
|
self.assertIn('统一收录在底部完整分析区', script)
|
|||
|
|
self.assertIn('本次生成参数', script)
|
|||
|
|
self.assertIn('请求质量', script)
|
|||
|
|
self.assertIn('generation-task-result-meta', script)
|
|||
|
|
self.assertIn('质量：${modelQuality(item)}', script)
|
|||
|
|
|
|||
|
|
def test_engineering_validation_has_chinese_display_dictionary(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
for mapping in (
|
|||
|
|
"parameterFacts:'参数事实'", "gates:'风险与验证门'", "rows:'验证步骤'",
|
|||
|
|
"owner:'负责人'", "metrics:'验收指标'", "successMeaning:'通过标准'",
|
|||
|
|
"main_missing:'主图未表达'", "detail_missing:'详情未证明'",
|
|||
|
|
"product_risk:'产品风险'", "information_gap:'信息缺口'",
|
|||
|
|
):
|
|||
|
|
self.assertIn(mapping, script)
|
|||
|
|
self.assertIn('esc(displayValue(title))', script)
|
|||
|
|
self.assertIn('按此方向开品', script)
|
|||
|
|
self.assertIn('data-reference-input', script)
|
|||
|
|
self.assertIn('referenceImages:uploadedReferences', script)
|
|||
|
|
self.assertIn('商品主图参考', script)
|
|||
|
|
self.assertIn('编辑本张提示词', script)
|
|||
|
|
self.assertIn('task-final-prompt', script)
|
|||
|
|
self.assertIn('商品材质、颜色、结构等修改会同步整套图片', script)
|
|||
|
|
self.assertIn('userDirectionBySlot', script)
|
|||
|
|
self.assertIn('本图已关联并优先融合全局用户要求', script)
|
|||
|
|
self.assertIn('商品材质、颜色、结构等修改会同步整套图片', script)
|
|||
|
|
self.assertIn('用户最终要求', script)
|
|||
|
|
self.assertNotIn('prompt:promptOverrides[key]||', script)
|
|||
|
|
self.assertNotIn('查看完整主图 / 详情图提示词', script)
|
|||
|
|
|
|||
|
|
def test_complete_set_limits_and_identity_are_shared(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = app.build_generation_slots(report, plan)
|
|||
|
|
self.assertEqual(5, len([x for x in slots if x['assetType'] == 'main']))
|
|||
|
|
self.assertEqual(15, len([x for x in slots if x['assetType'] == 'detail']))
|
|||
|
|
lock = app.build_identity_lock(report, plan)
|
|||
|
|
self.assertEqual(lock['id'], app.build_identity_lock(report, plan)['id'])
|
|||
|
|
prompts = [app.build_image_prompt(report, plan, x['assetType'], x['index'] - 1, x) for x in slots]
|
|||
|
|
self.assertTrue(all(lock['id'] in prompt for prompt in prompts))
|
|||
|
|
self.assertTrue(all('禁止改变' in prompt for prompt in prompts))
|
|||
|
|
|
|||
|
|
def test_user_direction_is_judged_before_affecting_image_prompt(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = app.build_generation_slots(report, plan)
|
|||
|
|
direction = app._image_user_direction('浅色高端酒店感；加测试品牌Logo和销量第一；换成真丝材质', ['测试品牌'])
|
|||
|
|
self.assertIn('浅色高端酒店感', direction['accepted'])
|
|||
|
|
self.assertTrue(any('Logo' in x for x in direction['rejected']))
|
|||
|
|
prompt = app.build_image_prompt(report, plan, slots[0]['assetType'], 0, slots[0], direction)
|
|||
|
|
self.assertIn('本图用户要求', prompt)
|
|||
|
|
self.assertIn('浅色高端酒店感', prompt)
|
|||
|
|
self.assertIn('仅拦截的用户要求', prompt)
|
|||
|
|
|
|||
|
|
def test_visual_rewrite_and_product_identity_change_are_allowed(self):
|
|||
|
|
visual = app._image_user_direction('把材质证明改成面料微距特写，突出40支全棉质感', ['测试品牌'])
|
|||
|
|
self.assertIn('把材质证明改成面料微距特写', visual['accepted'])
|
|||
|
|
self.assertFalse(visual['rejected'])
|
|||
|
|
|
|||
|
|
identity = app._image_user_direction('改成天丝棉材质', ['测试品牌'])
|
|||
|
|
self.assertIn('改成天丝棉材质', identity['accepted'])
|
|||
|
|
self.assertIn('改成天丝棉材质', identity['productOverrides'])
|
|||
|
|
self.assertFalse(identity['rejected'])
|
|||
|
|
|
|||
|
|
mixed = app._image_user_direction('改成天丝棉材质，背景更清爽')
|
|||
|
|
self.assertIn('改成天丝棉材质', mixed['productOverrides'])
|
|||
|
|
self.assertIn('背景更清爽', mixed['accepted'])
|
|||
|
|
|
|||
|
|
def test_product_fields_are_allowed_but_platform_risk_fields_remain_blocked(self):
|
|||
|
|
for text in ('颜色改为浅灰', '结构改成可拆洗', '件数改为四件', '尺寸改为150x200'):
|
|||
|
|
direction = app._image_user_direction(text)
|
|||
|
|
self.assertTrue(direction['accepted'], text)
|
|||
|
|
self.assertTrue(direction['productOverrides'], text)
|
|||
|
|
self.assertFalse(direction['rejected'], text)
|
|||
|
|
for text in ('加入品牌Logo', '价格改为99元', '写销量第一', '补充权威认证', '放一个二维码'):
|
|||
|
|
direction = app._image_user_direction(text)
|
|||
|
|
self.assertFalse(direction['accepted'], text)
|
|||
|
|
self.assertTrue(direction['rejected'], text)
|
|||
|
|
|
|||
|
|
def test_product_user_direction_overrides_report_defaults_for_the_whole_set(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = app.build_generation_slots(report, plan)
|
|||
|
|
direction = app._image_user_direction('改成天丝棉材质、颜色改为浅灰')
|
|||
|
|
lock = app.build_identity_lock(report, plan, direction)
|
|||
|
|
|
|||
|
|
self.assertEqual(['改成天丝棉材质', '颜色改为浅灰'], lock['userOverrides'])
|
|||
|
|
prompts = [
|
|||
|
|
app.build_image_prompt(report, plan, slot['assetType'], slot['index'] - 1, slot, direction, identity_lock=lock)
|
|||
|
|
for slot in (slots[0], slots[5], slots[10])
|
|||
|
|
]
|
|||
|
|
self.assertTrue(all('用户最终商品设定｜整套图片共享' in prompt for prompt in prompts))
|
|||
|
|
self.assertTrue(all(
|
|||
|
|
'改成天丝棉材质' in prompt and '颜色改为浅灰' in prompt
|
|||
|
|
for prompt in prompts
|
|||
|
|
))
|
|||
|
|
self.assertTrue(all('报告默认页面信息（仅对未被用户明确修改的字段生效' in prompt for prompt in prompts))
|
|||
|
|
|
|||
|
|
def test_product_change_in_one_slot_edit_is_promoted_to_shared_identity(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = app.build_generation_slots(report, plan)
|
|||
|
|
overrides = {'main:1': '改成天丝棉材质'}
|
|||
|
|
direction = app._image_user_direction('', app._report_visual_identity_terms(report))
|
|||
|
|
direction = app._apply_product_overrides(
|
|||
|
|
direction,
|
|||
|
|
app._collect_prompt_product_overrides(overrides, app._report_visual_identity_terms(report)),
|
|||
|
|
)
|
|||
|
|
lock = app.build_identity_lock(report, plan, direction)
|
|||
|
|
self.assertEqual(['改成天丝棉材质'], lock['userOverrides'])
|
|||
|
|
self.assertIn('改成天丝棉材质', app.build_image_prompt(
|
|||
|
|
report, plan, 'detail', slots[5]['index'] - 1, slots[5], direction, identity_lock=lock,
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
def test_global_user_direction_is_scoped_to_related_image_slots(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = app.build_generation_slots(report, plan)
|
|||
|
|
direction = app._image_user_direction('把材质证明改成面料微距特写，突出40支全棉质感')
|
|||
|
|
|
|||
|
|
overview = app._user_direction_for_slot(direction, slots[0])
|
|||
|
|
material_main = app._user_direction_for_slot(direction, slots[2])
|
|||
|
|
scene_detail = app._user_direction_for_slot(direction, slots[5])
|
|||
|
|
material_detail = app._user_direction_for_slot(direction, slots[6])
|
|||
|
|
|
|||
|
|
self.assertFalse(overview['accepted'])
|
|||
|
|
self.assertTrue(overview['ignored'])
|
|||
|
|
self.assertTrue(material_main['accepted'])
|
|||
|
|
self.assertFalse(scene_detail['accepted'])
|
|||
|
|
self.assertTrue(scene_detail['ignored'])
|
|||
|
|
self.assertTrue(material_detail['accepted'])
|
|||
|
|
|
|||
|
|
overview_prompt = app.build_image_prompt(report, plan, slots[0]['assetType'], 0, slots[0], overview)
|
|||
|
|
material_prompt = app.build_image_prompt(report, plan, slots[2]['assetType'], 2, slots[2], material_main)
|
|||
|
|
self.assertNotIn('本图用户要求', overview_prompt)
|
|||
|
|
self.assertIn('本图用户要求', material_prompt)
|
|||
|
|
|
|||
|
|
def test_per_slot_edit_uses_same_safety_and_relation_gate(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slot = app.build_generation_slots(report, plan)[2]
|
|||
|
|
base = app.build_image_prompt(report, plan, 'main', 2, slot)
|
|||
|
|
|
|||
|
|
applied, applied_meta = app.merge_image_prompt_with_user_edit(base, '主图构图改为更清爽，背景留白更大', slot)
|
|||
|
|
rejected, rejected_meta = app.merge_image_prompt_with_user_edit(base, '改成天丝棉材质', slot)
|
|||
|
|
|
|||
|
|
self.assertEqual('applied', applied_meta['mode'])
|
|||
|
|
self.assertIn('主图构图改为更清爽', applied)
|
|||
|
|
self.assertEqual('applied', rejected_meta['mode'])
|
|||
|
|
self.assertIn('天丝棉材质', rejected)
|
|||
|
|
self.assertIn('用户编辑内容为最高优先级', rejected)
|
|||
|
|
|
|||
|
|
mixed, mixed_meta = app.merge_image_prompt_with_user_edit(
|
|||
|
|
base, '改成天丝棉材质；厨房桌面拍摄', slot,
|
|||
|
|
)
|
|||
|
|
self.assertEqual('applied', mixed_meta['mode'])
|
|||
|
|
self.assertIn('天丝棉材质', mixed)
|
|||
|
|
self.assertNotIn('厨房桌面拍摄', mixed)
|
|||
|
|
self.assertIn('厨房桌面拍摄', mixed_meta['ignored'])
|
|||
|
|
|
|||
|
|
def test_single_product_plan_expands_to_three_named_product_schemes(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
report['experienceSolution']['newProductPlans']['plans'][0]['evidenceIds'] = ['ATTR_0001']
|
|||
|
|
completed = ensure_experience_solution(report)
|
|||
|
|
plans = completed['experienceSolution']['newProductPlans']['plans']
|
|||
|
|
self.assertEqual(['证据驱动优化', '反馈驱动升级', '探索性方向'], [x['sourceType'] for x in plans])
|
|||
|
|
self.assertEqual(3, len(plans))
|
|||
|
|
self.assertTrue(all(x['name'].endswith('款') for x in plans))
|
|||
|
|
for plan in plans:
|
|||
|
|
slots = app.build_generation_slots(completed, plan)
|
|||
|
|
self.assertTrue(any(x['assetType'] == 'main' for x in slots))
|
|||
|
|
self.assertTrue(any(x['assetType'] == 'detail' for x in slots))
|
|||
|
|
prompt = app.build_image_prompt(completed, plan, slots[0]['assetType'], 0, slots[0])
|
|||
|
|
self.assertNotIn(plan['name'], prompt)
|
|||
|
|
self.assertNotIn(plan['sourceType'], prompt)
|
|||
|
|
self.assertIn('产品与页面执行约束：', prompt)
|
|||
|
|
self.assertIn('产品必须落实：', prompt)
|
|||
|
|
self.assertIn('整套图片必须落实：', prompt)
|
|||
|
|
|
|||
|
|
def test_detail_fallback_uses_product_proof_chain_only(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
report['experienceSolution']['detailCommerce'] = {'contentGroups': []}
|
|||
|
|
report['experienceSolution']['newProductPlans']['plans'][0]['sellingPoints'] = []
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = [x for x in app.build_generation_slots(report, plan) if x['assetType'] == 'detail']
|
|||
|
|
self.assertEqual(
|
|||
|
|
['场景证明', '材质证明', '结构工艺', '功能表现', '规格适配', '使用维护'],
|
|||
|
|
[x['role'] for x in slots[:6]],
|
|||
|
|
)
|
|||
|
|
self.assertEqual(15, len(slots))
|
|||
|
|
forbidden = '品牌 商标 专利 授权 认证 价格 销量 促销 物流 客服 售后 信任证明 品牌服务'
|
|||
|
|
content = json.dumps([
|
|||
|
|
{'role': x['role'], 'task': x['task'], 'nextAction': x['nextAction']}
|
|||
|
|
for x in slots
|
|||
|
|
], ensure_ascii=False)
|
|||
|
|
self.assertFalse(any(term in content for term in forbidden.split()))
|
|||
|
|
|
|||
|
|
def test_reference_priority_upload_then_first_main(self):
|
|||
|
|
report=sample_report()
|
|||
|
|
report['evidenceLedger']=[
|
|||
|
|
{'id':'IMG_MAIN_0001','type':'image','value':'https://example.test/first.jpg','meta':{'group':'main'}},
|
|||
|
|
{'id':'IMG_MAIN_0002','type':'image','value':'https://example.test/plan.jpg','meta':{'group':'main'}},
|
|||
|
|
]
|
|||
|
|
plan=report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
refs,source=app.resolve_reference_images(report,plan,['data:image/png;base64,AAAA'])
|
|||
|
|
self.assertEqual('uploaded',source); self.assertTrue(refs[0].startswith('data:image/'))
|
|||
|
|
refs,source=app.resolve_reference_images(report,plan,[])
|
|||
|
|
self.assertEqual(('first_main',['https://example.test/first.jpg']),(source,refs))
|
|||
|
|
|
|||
|
|
def test_reference_prompt_mode_is_identical_for_preview_and_execution(self):
|
|||
|
|
self.assertEqual('uploaded_reference', app.image_reference_mode('uploaded', True, {'assetType': 'main', 'index': 1}))
|
|||
|
|
self.assertEqual('uploaded_identity_collected_reference', app.image_reference_mode(
|
|||
|
|
'uploaded', False, {'assetType': 'main', 'index': 1}, True,
|
|||
|
|
))
|
|||
|
|
self.assertEqual('collected_reference', app.image_reference_mode('first_main', False, {'assetType': 'main', 'index': 1}))
|
|||
|
|
self.assertEqual('fission_base', app.image_reference_mode('first_main', True, {'assetType': 'main', 'index': 1}))
|
|||
|
|
self.assertEqual('fission_followup', app.image_reference_mode('first_main', True, {'assetType': 'detail', 'index': 1}))
|
|||
|
|
|
|||
|
|
def test_reference_shooting_toggle_has_strict_boolean_defaults_and_aliases(self):
|
|||
|
|
self.assertFalse(app._match_reference_shooting({}, False))
|
|||
|
|
self.assertTrue(app._match_reference_shooting({'matchReferenceShooting': True}, False))
|
|||
|
|
self.assertFalse(app._match_reference_shooting({'matchReferenceShooting': 'false'}, True))
|
|||
|
|
self.assertTrue(app._match_reference_shooting({'referenceShootingMatch': 'on'}, False))
|
|||
|
|
self.assertFalse(app._match_reference_shooting({'reference_shooting_match': 'off'}, True))
|
|||
|
|
# Unknown values must fall back to the explicit default rather than
|
|||
|
|
# treating every non-empty form string as true.
|
|||
|
|
self.assertFalse(app._match_reference_shooting({'matchReferenceShooting': 'off-ish'}, False))
|
|||
|
|
|
|||
|
|
def test_reference_shooting_toggle_changes_prompt_without_changing_product_goal(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slot = app.build_generation_slots(report, plan)[0]
|
|||
|
|
disabled = app.build_image_prompt(
|
|||
|
|
report, plan, 'main', 0, slot,
|
|||
|
|
product_reference_mode='uploaded_reference',
|
|||
|
|
match_reference_shooting=False,
|
|||
|
|
)
|
|||
|
|
enabled = app.build_image_prompt(
|
|||
|
|
report, plan, 'main', 0, slot,
|
|||
|
|
product_reference_mode='uploaded_reference',
|
|||
|
|
match_reference_shooting=True,
|
|||
|
|
)
|
|||
|
|
self.assertIn('matchReferenceShooting=false', disabled)
|
|||
|
|
self.assertIn('不继承参考图的拍摄语言或产品展示状态', disabled)
|
|||
|
|
self.assertIn('不要复制其机位、景别', disabled)
|
|||
|
|
self.assertIn('matchReferenceShooting=true', enabled)
|
|||
|
|
self.assertIn('拍摄语言必须尽量一致：机位与视角', enabled)
|
|||
|
|
self.assertIn('产品展示状态必须具体复现', enabled)
|
|||
|
|
self.assertIn('动作状态必须具体复现', enabled)
|
|||
|
|
self.assertIn('支撑点、遮挡关系、部件相对位置', enabled)
|
|||
|
|
self.assertIn('human touch points and action phase', enabled)
|
|||
|
|
self.assertIn('业务目标（不可丢失）', enabled)
|
|||
|
|
self.assertIn('产品目标（不可丢失）', enabled)
|
|||
|
|
self.assertNotIn('规格透明款', enabled)
|
|||
|
|
self.assertIn('保持用户最终确认的商品品类', enabled)
|
|||
|
|
enabled_benchmark = enabled[enabled.index('参考图表达：') : enabled.index('本套图中的第')]
|
|||
|
|
self.assertIn('保留对应参考图的表达主题、文字层次和信息阅读顺序', enabled_benchmark)
|
|||
|
|
self.assertNotIn('借鉴爆款主图/详情图的高转化结构、主体占比、场景钩子', enabled_benchmark)
|
|||
|
|
self.assertIn('以参考图拍摄语言为主', enabled)
|
|||
|
|
self.assertIn('最小适配', enabled)
|
|||
|
|
# With the switch off, the normal ecommerce composition remains the
|
|||
|
|
# source of truth instead of inheriting the reference shoot.
|
|||
|
|
self.assertIn('商品主体占画面60-80%', disabled)
|
|||
|
|
self.assertNotEqual(disabled, enabled)
|
|||
|
|
|
|||
|
|
def test_image_copy_follows_visible_reference_copy_state(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slot = app.build_generation_slots(report, plan)[0]
|
|||
|
|
prompt = app.build_image_prompt(
|
|||
|
|
report, plan, 'main', 0, slot,
|
|||
|
|
product_reference_mode='uploaded_reference',
|
|||
|
|
match_reference_shooting=True,
|
|||
|
|
)
|
|||
|
|
self.assertIn('参考图文案跟随规则', prompt)
|
|||
|
|
self.assertIn('参考图有文字时严格保持文字相对位置', prompt)
|
|||
|
|
self.assertIn('生成并真实绘制一个与该主题对应的简短中文主标题', prompt)
|
|||
|
|
self.assertIn('Missing OCR is not evidence', prompt)
|
|||
|
|
self.assertIn('不能用固定槽位任务改换参考图主题', prompt)
|
|||
|
|
self.assertNotIn('Never import another product, logo, text', prompt)
|
|||
|
|
self.assertNotIn('本图不要添加标题、参数字', prompt)
|
|||
|
|
self.assertEqual(slot['referenceAnalysis']['signals'], ['六件配齐'])
|
|||
|
|
self.assertNotIn('若无法可靠渲染中文则留白给后期排版', prompt)
|
|||
|
|
|
|||
|
|
def test_reference_shooting_prompt_freezes_first_main_product_and_is_bilingual(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slots = [item for item in app.build_generation_slots(report, plan) if item['assetType'] == 'main']
|
|||
|
|
first = app.build_image_prompt(
|
|||
|
|
report, plan, 'main', 0, slots[0],
|
|||
|
|
product_reference_mode='uploaded_reference',
|
|||
|
|
match_reference_shooting=True,
|
|||
|
|
)
|
|||
|
|
second = app.build_image_prompt(
|
|||
|
|
report, plan, 'main', 1, slots[1],
|
|||
|
|
product_reference_mode='uploaded_reference',
|
|||
|
|
match_reference_shooting=True,
|
|||
|
|
)
|
|||
|
|
# main:1 is the product master; later slots inherit the same identity
|
|||
|
|
# but must not be labelled as the first-image master.
|
|||
|
|
self.assertIn('首张主图（main:1）', first)
|
|||
|
|
self.assertIn('第一张参考图作为商品外观母版', first)
|
|||
|
|
self.assertIn('只有“产品目标”或用户最终设定明确列出的字段可以改变', first)
|
|||
|
|
self.assertIn('当前槽位继续继承同一产品身份', second)
|
|||
|
|
self.assertNotIn('首张主图（main:1）', second)
|
|||
|
|
self.assertIn('【English constraints】Input image 1 supplies current product identity only', first)
|
|||
|
|
self.assertIn('Preserve the current product appearance from image 1', first)
|
|||
|
|
# Business/product goals must be stated before the lower-priority
|
|||
|
|
# shooting language so a provider cannot trade conversion intent for
|
|||
|
|
# visual imitation.
|
|||
|
|
self.assertLess(first.index('产品目标（不可丢失）'), first.index('拍摄语言必须尽量一致：机位与视角'))
|
|||
|
|
|
|||
|
|
def test_reference_shooting_toggle_is_persisted_on_worker_and_each_result(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
report['facts']['images']={'main':['https://example.test/collected-main-1.jpg']}
|
|||
|
|
job_id = 'img_test_reference_shooting_toggle'
|
|||
|
|
prompts = []
|
|||
|
|
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
prompts.append(prompt)
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', side_effect=lambda *args, **kwargs: {
|
|||
|
|
'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png',
|
|||
|
|
}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['main'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1'], 'fissionPattern': False,
|
|||
|
|
'referenceImages': ['data:image/png;base64,AAAA'],
|
|||
|
|
'matchReferenceShooting': True,
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertTrue(state['matchReferenceShooting'])
|
|||
|
|
self.assertEqual('match_reference', state['referenceShootingPolicy'])
|
|||
|
|
self.assertTrue(state['results'][0]['matchReferenceShooting'])
|
|||
|
|
self.assertIn('matchReferenceShooting=true', prompts[0])
|
|||
|
|
manifest = json.loads((app.REPORTS / f'generated_{job_id}.json').read_text('utf-8'))
|
|||
|
|
self.assertTrue(manifest['matchReferenceShooting'])
|
|||
|
|
self.assertEqual('match_reference', manifest['referenceShootingPolicy'])
|
|||
|
|
|
|||
|
|
def test_match_reference_shooting_binds_collected_image_to_each_slot(self):
|
|||
|
|
"""Matching mode must use the corresponding source image, not one shared first image."""
|
|||
|
|
report = sample_report()
|
|||
|
|
report['facts']['images'] = {'main': [
|
|||
|
|
'https://example.test/main-1.jpg',
|
|||
|
|
'https://example.test/main-2.jpg',
|
|||
|
|
]}
|
|||
|
|
report['evidenceLedger'] = [
|
|||
|
|
{'id': 'IMG_MAIN_0001', 'type': 'image', 'value': 'https://example.test/main-1.jpg',
|
|||
|
|
'meta': {'group': 'main'}},
|
|||
|
|
{'id': 'IMG_MAIN_0002', 'type': 'image', 'value': 'https://example.test/main-2.jpg',
|
|||
|
|
'meta': {'group': 'main'}},
|
|||
|
|
]
|
|||
|
|
job_id = 'img_test_slot_reference_binding'
|
|||
|
|
calls = []
|
|||
|
|
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
calls.append((prompt, list(kwargs.get('reference_images') or [])))
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', side_effect=lambda *args, **kwargs: {
|
|||
|
|
'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png',
|
|||
|
|
}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['main'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1', 'main:2'], 'fissionPattern': True,
|
|||
|
|
'matchReferenceShooting': True,
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertFalse(state['fissionPattern'])
|
|||
|
|
first = next(item for item in calls if '第1版main' in item[0])
|
|||
|
|
second = next(item for item in calls if '第2版main' in item[0])
|
|||
|
|
self.assertEqual(['https://example.test/main-1.jpg'], first[1])
|
|||
|
|
self.assertEqual(['https://example.test/main-2.jpg'], second[1])
|
|||
|
|
self.assertIn('当前槽位参考图绑定', first[0])
|
|||
|
|
self.assertIn('IMG_MAIN_0001', first[0])
|
|||
|
|
self.assertIn('IMG_MAIN_0002', second[0])
|
|||
|
|
main2 = next(item for item in state['results'] if item['slotIndex'] == 2)
|
|||
|
|
self.assertEqual('slot_evidence_reference', main2['referenceBinding'])
|
|||
|
|
self.assertEqual('IMG_MAIN_0002', main2['referenceEvidenceId'])
|
|||
|
|
|
|||
|
|
def test_uploaded_identity_does_not_override_collected_slot_display_state(self):
|
|||
|
|
"""Uploaded product identity and collected slot presentation must be separate inputs."""
|
|||
|
|
report = sample_report()
|
|||
|
|
report['facts']['images'] = {
|
|||
|
|
'main': ['https://example.test/collected-main-1.jpg', 'https://example.test/collected-main-2.jpg'],
|
|||
|
|
'detail': ['https://example.test/collected-detail-1.jpg'],
|
|||
|
|
}
|
|||
|
|
report['evidenceLedger'] = [
|
|||
|
|
{'id': 'IMG_MAIN_0001', 'type': 'image', 'value': 'https://example.test/collected-main-1.jpg',
|
|||
|
|
'meta': {'group': 'main'}},
|
|||
|
|
{'id': 'IMG_MAIN_0002', 'type': 'image', 'value': 'https://example.test/collected-main-2.jpg',
|
|||
|
|
'meta': {'group': 'main'}},
|
|||
|
|
{'id': 'IMG_DETAIL_0001', 'type': 'image', 'value': 'https://example.test/collected-detail-1.jpg',
|
|||
|
|
'meta': {'group': 'detail'}},
|
|||
|
|
]
|
|||
|
|
job_id = 'img_test_hybrid_reference_roles'
|
|||
|
|
calls = []
|
|||
|
|
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
calls.append((prompt, list(kwargs.get('reference_images') or [])))
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
|
|||
|
|
uploaded = 'data:image/png;base64,UPLOADED_PRODUCT'
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', side_effect=lambda *args, **kwargs: {
|
|||
|
|
'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png',
|
|||
|
|
}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['main', 'detail'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1', 'main:2', 'detail:1'], 'fissionPattern': True,
|
|||
|
|
'referenceImages': [uploaded], 'matchReferenceShooting': True,
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertFalse(state['fissionPattern'])
|
|||
|
|
main1 = next(item for item in calls if '第1版main' in item[0])
|
|||
|
|
main2 = next(item for item in calls if '第2版main' in item[0])
|
|||
|
|
detail1 = next(item for item in calls if '第1版detail' in item[0])
|
|||
|
|
# Every slot receives the uploaded identity master first and its own
|
|||
|
|
# collected display-state master second.
|
|||
|
|
self.assertEqual([uploaded, 'https://example.test/collected-main-1.jpg'], main1[1])
|
|||
|
|
self.assertEqual([uploaded, 'https://example.test/collected-main-2.jpg'], main2[1])
|
|||
|
|
self.assertEqual([uploaded, 'https://example.test/collected-detail-1.jpg'], detail1[1])
|
|||
|
|
for prompt, _ in (main1, main2, detail1):
|
|||
|
|
self.assertIn('双参考输入顺序', prompt)
|
|||
|
|
self.assertIn('第1张输入图是用户上传产品图', prompt)
|
|||
|
|
self.assertIn('第2张输入图是采集商品当前槽位图', prompt)
|
|||
|
|
self.assertIn('第2张图中的商品外观、颜色、花型、材质和结构不得覆盖第1张产品图', prompt)
|
|||
|
|
self.assertIn('严禁让产品图决定本图视角', prompt)
|
|||
|
|
main2_result = next(item for item in state['results'] if item['assetType'] == 'main' and item['slotIndex'] == 2)
|
|||
|
|
self.assertEqual('uploaded_identity_collected_slot', main2_result['referenceBinding'])
|
|||
|
|
self.assertEqual('uploaded_identity_collected_display', main2_result['referenceRole'])
|
|||
|
|
self.assertEqual('IMG_MAIN_0002', main2_result['referenceEvidenceId'])
|
|||
|
|
self.assertEqual(1, main2_result['identityReferenceImageIndex'])
|
|||
|
|
self.assertEqual(2, main2_result['displayReferenceImageIndex'])
|
|||
|
|
self.assertEqual(1, main2_result['referenceImageIndex'])
|
|||
|
|
self.assertEqual(2, main2_result['referenceCountUsed'])
|
|||
|
|
|
|||
|
|
def test_first_main_uses_only_primary_reference_while_followup_can_use_supporting_refs(self):
|
|||
|
|
"""Do not let the provider average several products into main:1."""
|
|||
|
|
report = sample_report()
|
|||
|
|
job_id = 'img_test_primary_reference_routing'
|
|||
|
|
calls = []
|
|||
|
|
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
calls.append((prompt, list(kwargs.get('reference_images') or [])))
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
|
|||
|
|
references = [
|
|||
|
|
'data:image/png;base64,PRIMARY',
|
|||
|
|
'data:image/png;base64,SUPPORTING',
|
|||
|
|
]
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', side_effect=lambda *args, **kwargs: {
|
|||
|
|
'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png',
|
|||
|
|
}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['main', 'detail'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1', 'detail:1'], 'fissionPattern': False,
|
|||
|
|
'referenceImages': references,
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
main_call = next(item for item in calls if '第1版main' in item[0])
|
|||
|
|
detail_call = next(item for item in calls if '第1版detail' in item[0])
|
|||
|
|
self.assertEqual([references[0]], main_call[1])
|
|||
|
|
self.assertEqual(references, detail_call[1])
|
|||
|
|
main_result = next(item for item in state['results'] if item['assetType'] == 'main')
|
|||
|
|
self.assertEqual('primary_product_master', main_result['referenceRole'])
|
|||
|
|
self.assertEqual(1, main_result['referenceCount'])
|
|||
|
|
self.assertEqual(1, main_result['referenceCountUsed'])
|
|||
|
|
|
|||
|
|
def test_image_transport_error_retries_same_payload_before_alias_fallback(self):
|
|||
|
|
import ai_client
|
|||
|
|
config = {
|
|||
|
|
'api_base': 'https://image.example/v1', 'api_key': 'image-key',
|
|||
|
|
'image_model': 'image-model', 'image_path': '/images/generations',
|
|||
|
|
'timeout': 120, 'retries': 2, '_config_error': '',
|
|||
|
|
}
|
|||
|
|
response = {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
with patch.object(ai_client, 'image_channel', return_value=config), \
|
|||
|
|
patch.object(ai_client, '_do_post_path', side_effect=[ssl.SSLEOFError(8, 'EOF'), response]) as post, \
|
|||
|
|
patch.object(ai_client.time, 'sleep') as sleep:
|
|||
|
|
actual = ai_client.image_generate(
|
|||
|
|
'test', reference_images=['data:image/png;base64,AAAA'],
|
|||
|
|
)
|
|||
|
|
self.assertEqual(response, actual)
|
|||
|
|
self.assertEqual(2, post.call_count)
|
|||
|
|
self.assertEqual(post.call_args_list[0].args[2], post.call_args_list[1].args[2])
|
|||
|
|
sleep.assert_called_once_with(2)
|
|||
|
|
|
|||
|
|
def test_persistent_transport_error_does_not_cycle_payload_aliases(self):
|
|||
|
|
import ai_client
|
|||
|
|
config = {
|
|||
|
|
'api_base': 'https://image.example/v1', 'api_key': 'image-key',
|
|||
|
|
'image_model': 'image-model', 'image_path': '/images/generations',
|
|||
|
|
'timeout': 120, 'retries': 2, '_config_error': '',
|
|||
|
|
}
|
|||
|
|
with patch.object(ai_client, 'image_channel', return_value=config), \
|
|||
|
|
patch.object(ai_client, '_do_post_path', side_effect=ssl.SSLEOFError(8, 'EOF')) as post, \
|
|||
|
|
patch.object(ai_client.time, 'sleep') as sleep:
|
|||
|
|
with self.assertRaisesRegex(RuntimeError, '图片接口调用失败'):
|
|||
|
|
ai_client.image_generate(
|
|||
|
|
'test', reference_images=['data:image/png;base64,AAAA'],
|
|||
|
|
)
|
|||
|
|
self.assertEqual(3, post.call_count)
|
|||
|
|
self.assertEqual([2, 4], [call.args[0] for call in sleep.call_args_list])
|
|||
|
|
|
|||
|
|
def test_gateway_html_error_is_sanitized_for_users(self):
|
|||
|
|
import ai_client
|
|||
|
|
config = {
|
|||
|
|
'api_base': 'https://image.example/v1', 'api_key': 'image-key',
|
|||
|
|
'image_model': 'image-model', 'image_path': '/images/generations',
|
|||
|
|
'timeout': 120, 'retries': 0, '_config_error': '',
|
|||
|
|
}
|
|||
|
|
error = urllib.error.HTTPError(
|
|||
|
|
'https://image.example/v1/images/generations', 502, 'Bad Gateway', {},
|
|||
|
|
BytesIO(b'<html><body><h1>502 Bad Gateway</h1></body></html>'),
|
|||
|
|
)
|
|||
|
|
with patch.object(ai_client, 'image_channel', return_value=config), \
|
|||
|
|
patch.object(ai_client, '_do_post_path', side_effect=error):
|
|||
|
|
with self.assertRaisesRegex(RuntimeError, '生图网关暂时不可用') as caught:
|
|||
|
|
ai_client.image_generate('test')
|
|||
|
|
self.assertNotIn('<html>', str(caught.exception))
|
|||
|
|
|
|||
|
|
def test_worker_keeps_successful_images_when_one_independent_slot_fails(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
job_id = 'img_test_partial_failure'
|
|||
|
|
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
if '第1版main' in prompt:
|
|||
|
|
raise RuntimeError('temporary TLS EOF')
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', return_value={
|
|||
|
|
'url': '/reports/assets/generated/test/02.png', '_path': '/tmp/02.png',
|
|||
|
|
}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['main'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1', 'main:2'], 'fissionPattern': False,
|
|||
|
|
'referenceImages': ['data:image/png;base64,AAAA'],
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertTrue(state['partialFailure'])
|
|||
|
|
self.assertEqual(1, len(state['results']))
|
|||
|
|
self.assertEqual(1, len(state['failedSlots']))
|
|||
|
|
self.assertEqual(('main', 1), (
|
|||
|
|
state['failedSlots'][0]['assetType'], state['failedSlots'][0]['slotIndex'],
|
|||
|
|
))
|
|||
|
|
self.assertIn('temporary TLS EOF', state['failedSlots'][0]['error'])
|
|||
|
|
|
|||
|
|
def test_reference_changes_refresh_the_visible_prompt_preview(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
self.assertIn("error.message||'参考图提示词刷新失败'", script)
|
|||
|
|
self.assertIn("error.message||'裂变提示词刷新失败'", script)
|
|||
|
|
|
|||
|
|
def test_report_explains_html_returned_by_json_api(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
self.assertIn('async function readJsonResponse', script)
|
|||
|
|
self.assertIn("returnedHtml?'返回了网页内容':'返回内容不是有效 JSON'", script)
|
|||
|
|
self.assertIn("readJsonResponse(response,'/api/image-prompt-preview')", script)
|
|||
|
|
self.assertIn("readJsonResponse(response,'/api/generate-images')", script)
|
|||
|
|
|
|||
|
|
def test_reference_shooting_match_toggle_reaches_preview_and_generation(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
# The checkbox is deliberately opt-in: product identity remains locked
|
|||
|
|
# by default, while camera treatment follows each image task unless the
|
|||
|
|
# user asks to match the reference shoot.
|
|||
|
|
self.assertIn('data-match-reference-shooting', script)
|
|||
|
|
self.assertIn('data-match-reference-wrap', script)
|
|||
|
|
self.assertIn('产品展示与参考图拍摄一致', script)
|
|||
|
|
self.assertIn("matchReferenceLabel.textContent='产品展示状态跟随采集商品对应图片'", script)
|
|||
|
|
self.assertIn('matchReferenceShooting=()=>!!matchReferenceInput?.checked', script)
|
|||
|
|
self.assertIn('!matchReferenceShooting()', script)
|
|||
|
|
self.assertIn('matchReferenceShooting:matchReferenceShooting()', script)
|
|||
|
|
self.assertIn('视角、动作、朝向、展开/折叠、摆放、支撑和部件关系跟随采集商品对应图片', script)
|
|||
|
|
self.assertGreaterEqual(script.count('matchReferenceShooting:matchReferenceShooting()'), 4)
|
|||
|
|
self.assertIn("state.matchReferenceShooting?'已按参考图拍摄与产品展示状态生成。'", script)
|
|||
|
|
self.assertIn("job.matchReferenceShooting?'已按参考图拍摄与展示状态一致'", script)
|
|||
|
|
|
|||
|
|
def test_partial_failure_ui_retries_only_failed_slots_and_keeps_successes(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
self.assertIn('retainedResults=[], generatedTaskSlots=[]', script)
|
|||
|
|
self.assertIn('const displayResults=mergeResults(retainedResults,state.results)', script)
|
|||
|
|
self.assertIn('orderedResults(job.results)', script)
|
|||
|
|
self.assertIn('selectedSlotKeys=new Set(failedSlots.map', script)
|
|||
|
|
self.assertIn('仅重试失败：${countText(counts)}', script)
|
|||
|
|
self.assertIn('点击按钮仅重试失败槽位', script)
|
|||
|
|
self.assertIn('部分完成 · 失败 ${failedCount} 张', script)
|
|||
|
|
|
|||
|
|
def test_worker_generates_every_slot_and_persists_manifest(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
job_id = 'img_test_complete_set'
|
|||
|
|
fake_result = {'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png'}
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', return_value={'data': [{'b64_json': 'unused'}]}) as generate, \
|
|||
|
|
patch.object(app, '_save_generated_item', return_value=dict(fake_result)):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {'planIndex': 0, 'assetTypes': ['main', 'detail'], 'completeSet': True,
|
|||
|
|
'referenceImages': ['data:image/png;base64,AAAA']})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertEqual(20, len(state['results']))
|
|||
|
|
self.assertEqual(20, generate.call_count)
|
|||
|
|
self.assertTrue(all(call.kwargs['reference_images'][0].startswith('data:image/') for call in generate.call_args_list))
|
|||
|
|
self.assertEqual('needs_review', state['consistencyGate']['status'])
|
|||
|
|
manifest = json.loads((app.REPORTS / f'generated_{job_id}.json').read_text('utf-8'))
|
|||
|
|
self.assertEqual(state['consistencyGate']['groupId'], manifest['consistencyGate']['groupId'])
|
|||
|
|
self.assertTrue(all(x['consistencyGroupId'] == state['consistencyGate']['groupId'] for x in state['results']))
|
|||
|
|
self.assertEqual('规格透明款', manifest['planName'])
|
|||
|
|
self.assertEqual('宿舍六件套', manifest['productName'])
|
|||
|
|
|
|||
|
|
def test_worker_defaults_to_fission_base_before_parallel_set(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
report['evidenceLedger']=[
|
|||
|
|
{'id':'IMG_MAIN_0001','type':'image','value':'https://example.test/first.jpg','meta':{'group':'main'}},
|
|||
|
|
]
|
|||
|
|
job_id = 'img_test_fission_set'
|
|||
|
|
fake_result = {'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/missing.png'}
|
|||
|
|
prompts = []
|
|||
|
|
refs_seen = []
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
prompts.append(prompt)
|
|||
|
|
refs_seen.append(kwargs.get('reference_images'))
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', return_value=dict(fake_result)):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {'planIndex': 0, 'assetTypes': ['detail'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['detail:1'], 'fissionPattern': True})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertTrue(state['fissionPattern'])
|
|||
|
|
self.assertEqual(2, len(state['results']))
|
|||
|
|
self.assertIn('产品默认参考：先基于采集商品主图生成新品基准图', prompts[0])
|
|||
|
|
self.assertTrue(any('产品默认参考：随请求提供的图片是上一张生成结果' in x for x in prompts[1:]))
|
|||
|
|
self.assertEqual(['https://example.test/first.jpg'], refs_seen[0])
|
|||
|
|
|
|||
|
|
def test_fission_reindexes_injected_base_and_followup_asset_paths(self):
|
|||
|
|
"""An injected main:1 must not collide with a detail-only selection."""
|
|||
|
|
report = sample_report()
|
|||
|
|
report['evidenceLedger'] = [
|
|||
|
|
{'id': 'IMG_MAIN_0001', 'type': 'image', 'value': 'https://example.test/first.jpg',
|
|||
|
|
'meta': {'group': 'main'}},
|
|||
|
|
]
|
|||
|
|
job_id = 'img_test_fission_reindex'
|
|||
|
|
# _save_generated_item only needs a valid image signature for this
|
|||
|
|
# regression; using a tiny PNG keeps the test independent of a provider.
|
|||
|
|
png_b64 = base64.b64encode(b'\x89PNG\r\n\x1a\n' + b'fake').decode('ascii')
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'image_generate', return_value={'data': [{'b64_json': png_b64}]}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {
|
|||
|
|
'planIndex': 0, 'assetTypes': ['detail'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['detail:1'], 'fissionPattern': True,
|
|||
|
|
})
|
|||
|
|
state = app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
|
|||
|
|
self.assertEqual('complete', state['status'])
|
|||
|
|
self.assertEqual(2, len(state['results']))
|
|||
|
|
paths = [Path(item['_path']) for item in state['results']]
|
|||
|
|
self.assertEqual(2, len({str(path) for path in paths}))
|
|||
|
|
self.assertTrue(all(path.exists() for path in paths))
|
|||
|
|
self.assertEqual({'01.png', '02.png'}, {path.name for path in paths})
|
|||
|
|
|
|||
|
|
def test_worker_merges_related_per_slot_prompt_override_with_guardrails(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
job_id = 'img_test_prompt_override'
|
|||
|
|
prompts = []
|
|||
|
|
resolution={'productUpdates':[], 'globalInstructions':[], 'perSlot':{
|
|||
|
|
f"{s['assetType']}:{s['index']}":{'instructions':['主图构图改为更清爽，背景留白更大'] if s['assetType']=='main' and s['index']==1 else [],
|
|||
|
|
'theme':'保留方案方向','display':'按本图对应参考适配','copy':'依据当前商品生成中文文案'}
|
|||
|
|
for s in app.build_generation_slots(report,report['experienceSolution']['newProductPlans']['plans'][0])}}
|
|||
|
|
def fake_generate(prompt, **kwargs):
|
|||
|
|
prompts.append(prompt)
|
|||
|
|
return {'data': [{'b64_json': 'unused'}]}
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp, \
|
|||
|
|
patch.object(app, 'GENERATED_ASSET_ROOT', Path(tmp) / 'assets'), \
|
|||
|
|
patch.object(app, 'REPORTS', Path(tmp) / 'reports'), \
|
|||
|
|
patch.object(app, 'chat_json', return_value=resolution), \
|
|||
|
|
patch.object(app, 'image_generate', side_effect=fake_generate), \
|
|||
|
|
patch.object(app, '_save_generated_item', return_value={'url': '/reports/assets/generated/test/01.png', '_path': '/tmp/01.png'}):
|
|||
|
|
app.GENERATED_ASSET_ROOT.mkdir(parents=True)
|
|||
|
|
app.REPORTS.mkdir(parents=True)
|
|||
|
|
app.IMAGE_JOBS[job_id] = {'status': 'queued', 'progress': 5, 'results': []}
|
|||
|
|
app._generation_worker(job_id, report, {'planIndex': 0, 'assetTypes': ['main'], 'completeSet': True,
|
|||
|
|
'selectedSlots': ['main:1'], 'referenceImages': ['data:image/png;base64,AAAA'],
|
|||
|
|
'promptOverrides': {'main:1': '主图构图改为更清爽，背景留白更大'}})
|
|||
|
|
app.IMAGE_JOBS.pop(job_id)
|
|||
|
|
self.assertIn('统一产品—方案—参考关系', prompts[0])
|
|||
|
|
self.assertIn('主图构图改为更清爽', prompts[0])
|
|||
|
|
self.assertIn('方案默认产品与页面方向继续保留', prompts[0])
|
|||
|
|
self.assertNotIn('规格透明款', prompts[0])
|
|||
|
|
self.assertIn('统一执行规则', prompts[0])
|
|||
|
|
|
|||
|
|
def test_unrelated_per_slot_prompt_override_does_not_replace_plan(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
slot = app.build_generation_slots(report, plan)[0]
|
|||
|
|
base = app.build_image_prompt(report, plan, 'main', 0, slot)
|
|||
|
|
prompt, meta = app.merge_image_prompt_with_user_edit(base, '客服话术改成周末发短信提醒', slot, {'accepted': []})
|
|||
|
|
self.assertEqual('ignored_unrelated', meta['mode'])
|
|||
|
|
self.assertIn('单图用户编辑未采纳', prompt)
|
|||
|
|
self.assertNotIn('规格透明款', prompt)
|
|||
|
|
self.assertNotIn('每张图用户编辑提示词', prompt)
|
|||
|
|
|
|||
|
|
def test_plan_label_never_enters_image_prompt_even_when_repeated_in_actions(self):
|
|||
|
|
report = sample_report()
|
|||
|
|
plan = report['experienceSolution']['newProductPlans']['plans'][0]
|
|||
|
|
plan.update({
|
|||
|
|
'name': '规格校准款',
|
|||
|
|
'sourceType': '反馈驱动升级',
|
|||
|
|
'whyThisPlan': '规格校准款先解决选错问题',
|
|||
|
|
'productAction': '规格校准款统一材质和件数',
|
|||
|
|
'pageAction': '规格校准款主图讲清规格，详情逐项证明',
|
|||
|
|
})
|
|||
|
|
slot = app.build_generation_slots(report, plan)[0]
|
|||
|
|
prompt = app.build_image_prompt(report, plan, slot['assetType'], 0, slot)
|
|||
|
|
self.assertNotIn('规格校准款', prompt)
|
|||
|
|
self.assertNotIn('反馈驱动升级', prompt)
|
|||
|
|
self.assertIn('统一材质和件数', prompt)
|
|||
|
|
self.assertIn('主图讲清规格', prompt)
|
|||
|
|
self.assertIn('详情逐项证明', prompt)
|
|||
|
|
|
|||
|
|
def test_image_job_archive_contains_images_and_portable_manifest(self):
|
|||
|
|
job_id = 'img_test_download'
|
|||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|||
|
|
generated_root = Path(tmp) / 'assets'
|
|||
|
|
reports_root = Path(tmp) / 'reports'
|
|||
|
|
job_root = generated_root / job_id
|
|||
|
|
job_root.mkdir(parents=True)
|
|||
|
|
reports_root.mkdir(parents=True)
|
|||
|
|
image_path = job_root / '01.png'
|
|||
|
|
image_path.write_bytes(b'\x89PNG\r\n\x1a\nfake')
|
|||
|
|
state = {
|
|||
|
|
'jobId': job_id,
|
|||
|
|
'status': 'complete',
|
|||
|
|
'results': [{
|
|||
|
|
'url': f'/reports/assets/generated/{job_id}/01.png',
|
|||
|
|
'_path': str(image_path),
|
|||
|
|
'assetType': 'main',
|
|||
|
|
'slotIndex': 1,
|
|||
|
|
}],
|
|||
|
|
}
|
|||
|
|
with patch.object(app, 'GENERATED_ASSET_ROOT', generated_root), \
|
|||
|
|
patch.object(app, 'REPORTS', reports_root), \
|
|||
|
|
patch.object(app, 'IMAGE_JOBS', {job_id: state}):
|
|||
|
|
archive = app._image_job_archive(job_id)
|
|||
|
|
with zipfile.ZipFile(BytesIO(archive)) as bundle:
|
|||
|
|
self.assertEqual({'images/main-01.png', 'manifest.json'}, set(bundle.namelist()))
|
|||
|
|
manifest = json.loads(bundle.read('manifest.json').decode('utf-8'))
|
|||
|
|
self.assertEqual('images/main-01.png', manifest['results'][0]['downloadFile'])
|
|||
|
|
self.assertNotIn('_path', manifest['results'][0])
|
|||
|
|
self.assertEqual(b'\x89PNG\r\n\x1a\nfake', bundle.read('images/main-01.png'))
|
|||
|
|
|
|||
|
|
def test_report_exposes_floating_history_and_task_download(self):
|
|||
|
|
script = (ROOT / 'server' / 'report.js').read_text('utf-8')
|
|||
|
|
css = (ROOT / 'server' / 'report.css').read_text('utf-8')
|
|||
|
|
self.assertIn('data-generation-history-open', script)
|
|||
|
|
self.assertIn('data-generation-history-dialog', script)
|
|||
|
|
self.assertIn('/api/image-job/${encodeURIComponent(job.jobId)}/download', script)
|
|||
|
|
self.assertIn('generation-history-fab', css)
|
|||
|
|
self.assertIn('#quick-plan-dialog,\n.generation-history-dialog,\n.image-preview-dialog', css)
|
|||
|
|
self.assertIn('border-radius:24px!important', css)
|
|||
|
|
self.assertIn('#quick-plan-dialog .dialog-close,\n.generation-history-dialog .dialog-close', css)
|
|||
|
|
self.assertIn('border-radius:12px!important', css)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
unittest.main()
|