Fix some small rebase bugs.

Author: GraylinKim

sc2reader/engine/plugins/context.py

@@ -67,9 +67,9 @@ def handleSelectionEvent(self, event, replay):
                 if unit_id in replay.objects:
                     unit = replay.objects[unit_id]
                     if not unit.is_type(unit_type):
-                        replay.datapack.change_type(unit, unit_type, self.frame)
+                        replay.datapack.change_type(unit, unit_type, event.frame)
                 else:
-                    unit = replay.datapack.create_unit(unit_id, unit_type, 0x00, self.frame)
+                    unit = replay.datapack.create_unit(unit_id, unit_type, 0x00, event.frame)
                     replay.objects[unit_id] = unit
 
             # If we have tracker events, the unit must already exist and must already

test_replays/test_all.py

@@ -286,8 +286,12 @@ def test_map_info(self):
 
     def test_plugins(self):
         from sc2reader.engine.plugins import ContextLoader, APMTracker, SelectionTracker
+        from sc2reader.factories.plugins.replay import toJSON
 
-        replay = sc2reader.load_replay(
+        from sc2reader.factories import SC2Factory
+        factory = SC2Factory()
+        factory.register_plugin('Replay', toJSON())
+        replay = factory.load_replay(
             "test_replays/2.0.5.25092/cn1.SC2Replay",
             engine=sc2reader.engine.GameEngine(plugins=[
                 ContextLoader(),