iteration1

tests/unit/test_media_descriptor_change_set.py

@@ -212,6 +212,53 @@ class MediaDescriptorChangeSetTests(unittest.TestCase):
         self.assertIn("BPS=", metadata_tokens)
         self.assertIn("KEEP_ME=keep-me", metadata_tokens)
 
+    def test_cleanup_can_be_disabled_per_context(self):
+        context = {
+            "logger": get_ffx_logger(),
+            "config": StaticConfig(
+                {
+                    "metadata": {
+                        "remove": ["creation_time"],
+                        "streams": {
+                            "remove": ["BPS"],
+                        },
+                    }
+                }
+            ),
+            "apply_metadata_cleanup": False,
+        }
+
+        source_track = TrackDescriptor(
+            index=0,
+            source_index=0,
+            sub_index=0,
+            track_type=TrackType.AUDIO,
+            tags={"BPS": "keep-me"},
+        )
+        target_track = TrackDescriptor(
+            index=0,
+            source_index=0,
+            sub_index=0,
+            track_type=TrackType.AUDIO,
+            tags={"BPS": "keep-me"},
+        )
+
+        change_set = MediaDescriptorChangeSet(
+            context,
+            MediaDescriptor(
+                tags={"creation_time": "keep-me"},
+                track_descriptors=[target_track],
+            ),
+            MediaDescriptor(
+                tags={"creation_time": "keep-me"},
+                track_descriptors=[source_track],
+            ),
+        )
+
+        metadata_tokens = change_set.generateMetadataTokens()
+        self.assertNotIn("creation_time=", metadata_tokens)
+        self.assertNotIn("BPS=", metadata_tokens)
+
 
 if __name__ == "__main__":
     unittest.main()