@@ -237,12 +237,13 @@ def test_log_pos(self):
237
237
238
238
self .assertIsInstance (self .stream .fetchone (), XidEvent )
239
239
240
- self .assertIsNone (self .stream .fetchone ())
241
240
242
241
def test_log_pos_handles_disconnects (self ):
242
+ self .stream .close ()
243
243
self .stream = BinLogStreamReader (
244
244
connection_settings = self .database ,
245
- resume_stream = True
245
+ resume_stream = False ,
246
+ only_events = [FormatDescriptionEvent , QueryEvent , TableMapEvent , WriteRowsEvent , XidEvent ]
246
247
)
247
248
248
249
query = "CREATE TABLE test (id INT PRIMARY KEY AUTO_INCREMENT, data VARCHAR (50) NOT NULL)"
@@ -251,25 +252,18 @@ def test_log_pos_handles_disconnects(self):
251
252
self .execute (query )
252
253
self .execute ("COMMIT" )
253
254
254
- self .assertIsInstance (self .stream .fetchone (), RotateEvent )
255
-
256
255
self .assertIsInstance (self .stream .fetchone (), FormatDescriptionEvent )
257
256
self .assertGreater (self .stream .log_pos , 0 )
258
-
259
257
self .assertIsInstance (self .stream .fetchone (), QueryEvent )
258
+
260
259
self .assertIsInstance (self .stream .fetchone (), QueryEvent )
261
260
self .assertIsInstance (self .stream .fetchone (), TableMapEvent )
262
261
self .assertIsInstance (self .stream .fetchone (), WriteRowsEvent )
263
- self .assertIsInstance (self .stream .fetchone (), XidEvent )
264
262
265
- self .assertIsNone (self .stream .fetchone ())
266
- self .assertIsInstance (self .stream .fetchone (), RotateEvent )
263
+ self .assertIsInstance (self .stream .fetchone (), XidEvent )
267
264
268
- self .assertIsInstance (self .stream .fetchone (), FormatDescriptionEvent )
269
265
self .assertGreater (self .stream .log_pos , 0 )
270
266
271
- #self.assertFalse(self.stream._BinLogStreamReader__connected_stream)
272
-
273
267
274
268
class TestMultipleRowBinLogStreamReader (base .PyMySQLReplicationTestCase ):
275
269
def test_insert_multiple_row_event (self ):
0 commit comments