finally got unittest right

Author: Bartek Ogryczak

pymysqlreplication/tests/test_basic.py

@@ -407,24 +407,24 @@ def test_log_pos_handles_disconnects(self):
         self.assertGreater(self.stream.log_pos, 0)
 
     def test_skip_to_timestamp(self):
+        self.stream.close()
         query = "CREATE TABLE test_1 (id INT NOT NULL AUTO_INCREMENT, data VARCHAR (50) NOT NULL, PRIMARY KEY (id))"
         self.execute(query)
         time.sleep(1)
         query = "SELECT UNIX_TIMESTAMP();"
         timestamp = self.execute(query).fetchone()[0]
-        query = "CREATE TABLE test_2 (id INT NOT NULL AUTO_INCREMENT, data VARCHAR (50) NOT NULL, PRIMARY KEY (id))"
-        self.execute(query)
+        query2 = "CREATE TABLE test_2 (id INT NOT NULL AUTO_INCREMENT, data VARCHAR (50) NOT NULL, PRIMARY KEY (id))"
+        self.execute(query2)
 
-        self.stream.close()
         self.stream = BinLogStreamReader(
             self.database,
             server_id=1024,
             skip_to_timestamp=timestamp,
             ignored_events=self.ignoredEvents(),
             )
-        self.assertIsInstance(self.stream.fetchone(), QueryEvent)
         event = self.stream.fetchone()
-        self.assertEqual(event.table, "test_2")
+        self.assertIsInstance(event, QueryEvent)
+        self.assertEqual(event.query, query2)
 
 
 class TestMultipleRowBinLogStreamReader(base.PyMySQLReplicationTestCase):