refactor: update for rebase-introduced migration and API change

Author: ormsbee

openedx_learning/apps/authoring/publishing/migrations/0005_draftchangelog.py

@@ -886,8 +886,10 @@ def test_parent_child_side_effects(self) -> None:
             container.pk,
             1,
             title="My Container",
-            publishable_entities_pks=[child_1.id, child_2.id],
-            entity_version_pks=None,
+            entity_rows=[
+                publishing_api.ContainerEntityRow(entity_pk=child_1.pk),
+                publishing_api.ContainerEntityRow(entity_pk=child_2.pk),
+            ],
             created=self.now,
             created_by=None,
         )
@@ -965,8 +967,10 @@ def test_bulk_parent_child_side_effects(self) -> None:
                 container.pk,
                 1,
                 title="My Container",
-                publishable_entities_pks=[child_1.id, child_2.id],
-                entity_version_pks=None,
+                entity_rows=[
+                    publishing_api.ContainerEntityRow(entity_pk=child_1.pk),
+                    publishing_api.ContainerEntityRow(entity_pk=child_2.pk),
+                ],
                 created=self.now,
                 created_by=None,
             )
@@ -1032,8 +1036,9 @@ def test_multiple_layers_of_containers(self):
             unit.pk,
             1,
             title="My Unit",
-            publishable_entities_pks=[component.pk],
-            entity_version_pks=None,
+            entity_rows=[
+                publishing_api.ContainerEntityRow(entity_pk=component.pk),
+            ],
             created=self.now,
             created_by=None,
         )
@@ -1044,8 +1049,9 @@ def test_multiple_layers_of_containers(self):
             subsection.pk,
             1,
             title="My Subsection",
-            publishable_entities_pks=[unit.pk],
-            entity_version_pks=None,
+            entity_rows=[
+                publishing_api.ContainerEntityRow(entity_pk=unit.pk),
+            ],
             created=self.now,
             created_by=None,
         )

openedx_learning/apps/authoring/publishing/migrations/0006_bootstrap_draftchangelog.py

@@ -173,7 +173,7 @@ def test_create_unit_queries(self):
         # The exact numbers here aren't too important - this is just to alert us if anything significant changes.
         with self.assertNumQueries(25):
             _empty_unit = self.create_unit_with_components([])
-        with self.assertNumQueries(28):
+        with self.assertNumQueries(29):
             # And try with a non-empty unit:
             self.create_unit_with_components([self.component_1, self.component_2_v1], key="u2")