| 100 | |
| 101 | |
| 102 | class VeoLiteGenerationHandlerTests(unittest.TestCase): |
| 103 | def test_tier_two_does_not_upgrade_lite_model_to_fake_ultra(self): |
| 104 | handler = GenerationHandler.__new__(GenerationHandler) |
| 105 | |
| 106 | model_key, message = handler._resolve_video_model_key_for_tier( |
| 107 | { |
| 108 | "model_key": "veo_3_1_t2v_lite", |
| 109 | "allow_tier_upgrade": False, |
| 110 | }, |
| 111 | "PAYGATE_TIER_TWO", |
| 112 | ) |
| 113 | |
| 114 | self.assertEqual(model_key, "veo_3_1_t2v_lite") |
| 115 | self.assertIsNone(message) |
| 116 | |
| 117 | def test_tier_two_still_upgrades_regular_model(self): |
| 118 | handler = GenerationHandler.__new__(GenerationHandler) |
| 119 | |
| 120 | model_key, message = handler._resolve_video_model_key_for_tier( |
| 121 | { |
| 122 | "model_key": "veo_3_1_t2v_fast", |
| 123 | }, |
| 124 | "PAYGATE_TIER_TWO", |
| 125 | ) |
| 126 | |
| 127 | self.assertEqual(model_key, "veo_3_1_t2v_fast_ultra") |
| 128 | self.assertIn("ultra", message) |
| 129 | |
| 130 | def test_quality_model_does_not_upgrade_to_fake_ultra(self): |
| 131 | handler = GenerationHandler.__new__(GenerationHandler) |
| 132 | |
| 133 | model_key, message = handler._resolve_video_model_key_for_tier( |
| 134 | { |
| 135 | "model_key": "veo_3_1_t2v", |
| 136 | }, |
| 137 | "PAYGATE_TIER_TWO", |
| 138 | ) |
| 139 | |
| 140 | self.assertEqual(model_key, "veo_3_1_t2v") |
| 141 | self.assertIsNone(message) |
| 142 | |
| 143 | def test_quality_4s_upsample_model_generates_then_upsamples(self): |
| 144 | cfg = MODEL_CONFIG["veo_3_1_t2v_4s_4k"] |
| 145 | |
| 146 | self.assertEqual(cfg["model_key"], "veo_3_1_t2v_quality_4s") |
| 147 | self.assertEqual(cfg["video_type"], "t2v") |
| 148 | self.assertEqual(cfg["upsample"]["model_key"], "veo_3_1_upsampler_4k") |
| 149 | self.assertEqual(cfg["upsample"]["resolution"], "VIDEO_RESOLUTION_4K") |
| 150 | |
| 151 | def test_quality_6s_i2v_1080p_model_generates_then_upsamples(self): |
| 152 | cfg = MODEL_CONFIG["veo_3_1_i2v_s_6s_1080p"] |
| 153 | |
| 154 | self.assertEqual(cfg["model_key"], "veo_3_1_i2v_s_quality_6s_fl") |
| 155 | self.assertEqual(cfg["video_type"], "i2v") |
| 156 | self.assertEqual(cfg["upsample"]["model_key"], "veo_3_1_upsampler_1080p") |
| 157 | self.assertEqual(cfg["upsample"]["resolution"], "VIDEO_RESOLUTION_1080P") |
| 158 | |
| 159 | def test_explicit_8s_aliases_reuse_default_upstream_keys(self): |
nothing calls this directly
no outgoing calls
no test coverage detected