@@ -649,20 +649,23 @@ def test_get_none_invalid(self):
649
649
else :
650
650
self .assertEqual (event .event_type , UPDATE_ROWS_EVENT_V1 )
651
651
self .assertIsInstance (event , UpdateRowsEvent )
652
-
653
- if event .rows [0 ].get ("before_none_sources" ):
654
- self .assertEqual (event .rows [0 ]["before_none_sources" ]["col0" ], "null" )
655
- self .assertEqual (event .rows [0 ]["before_none_sources" ]["col1" ], "null" )
656
- self .assertEqual (event .rows [0 ]["before_none_sources" ]["col2" ], "out of datetime2 range" )
657
- self .assertEqual (event .rows [0 ]["before_none_sources" ]["col3" ], "null" )
658
- self .assertEqual (event .rows [0 ]["before_none_sources" ]["col4" ], "null" )
659
-
660
- if event .rows [0 ].get ("after_none_sources" ):
661
- self .assertEqual (event .rows [0 ]["after_none_sources" ]["col0" ], "null" )
662
- self .assertEqual (event .rows [0 ]["after_none_sources" ]["col1" ], "null" )
663
- self .assertEqual (event .rows [0 ]["after_none_sources" ]["col2" ], "null" )
664
- self .assertEqual (event .rows [0 ]["after_none_sources" ]["col3" ], "out of date range" )
665
- self .assertEqual (event .rows [0 ]["after_none_sources" ]["col4" ], "empty set" )
652
+
653
+ before_none_sources = event .rows [0 ].get ("before_none_sources" )
654
+ after_none_sources = event .rows [0 ].get ("after_none_sources" )
655
+
656
+ if before_none_sources :
657
+ self .assertEqual (before_none_sources ["col0" ], "null" )
658
+ self .assertEqual (before_none_sources ["col1" ], "null" )
659
+ self .assertEqual (before_none_sources ["col2" ], "out of datetime2 range" )
660
+ self .assertEqual (before_none_sources ["col3" ], "null" )
661
+ self .assertEqual (before_none_sources ["col4" ], "null" )
662
+
663
+ if after_none_sources :
664
+ self .assertEqual (after_none_sources ["col0" ], "null" )
665
+ self .assertEqual (after_none_sources ["col1" ], "null" )
666
+ self .assertEqual (after_none_sources ["col2" ], "null" )
667
+ self .assertEqual (after_none_sources ["col3" ], "out of date range" )
668
+ self .assertEqual (after_none_sources ["col4" ], "empty set" )
666
669
667
670
668
671
class TestMultipleRowBinLogStreamReader (base .PyMySQLReplicationTestCase ):
0 commit comments