Skip to content

Commit 6a506eb

Browse files
committed
refactor: get none_sources as variable
1 parent ab1c5fb commit 6a506eb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pymysqlreplication/tests/test_basic.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -621,8 +621,9 @@ def test_get_none(self):
621621
write_rows_event = self.stream.fetchone()
622622
self.assertIsInstance(write_rows_event, WriteRowsEvent)
623623

624-
if write_rows_event.rows[0].get("none_sources"):
625-
self.assertEqual(write_rows_event.rows[0]["none_sources"]["col1"], "null")
624+
none_sources = write_rows_event.rows[0].get("none_sources")
625+
if none_sources:
626+
self.assertEqual(none_sources["col1"], "null")
626627

627628
def test_get_none_invalid(self):
628629
self.execute("SET SESSION SQL_MODE='ALLOW_INVALID_DATES'")

0 commit comments

Comments
 (0)