upd migrate test

Author: joente

itest/lib/node.py

@@ -95,15 +95,15 @@ async def init(self):
         await self.expect(
             'Well done, you successfully initialized ThingsDB!!', timeout=5)
 
-    async def wait_join(self, secret: str):
+    async def wait_join(self, secret: str, timeout: int = 5):
         self.start(secret=secret)
         await self.expect(
-            'start listening for node connections', timeout=5)
+            'start listening for node connections', timeout=timeout)
 
-    async def run(self):
+    async def run(self, timeout: int = 5):
         self.start()
         await self.expect(
-            'start listening for node connections', timeout=5)
+            'start listening for node connections', timeout=timeout)
 
     async def init_and_run(self):
         await self.init()

itest/test_migrate.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 import asyncio
 import os
+import logging
 from lib import run_test
 from lib import default_test_setup
 from lib.testbase import TestBase
@@ -28,6 +29,14 @@ async def run(self):
             if not fn.endswith('.tar.gz'):
                 continue
 
+            logging.warning(f"""
+    {'#' * 80}
+
+    Next file: {fn}
+
+    {'#' * 80}
+""")
+
             fn = os.path.join(BACKUP_PATH, fn)
 
             collections = await query('collections_info();')
@@ -40,6 +49,25 @@ async def run(self):
             await asyncio.sleep(10)
             await client.authenticate('admin', 'pass')
 
+            logging.warning(f"""
+    {'#' * 80}
+
+    Imported file: {fn}, restart test...
+
+    {'#' * 80}
+""")
+
+            await self.node0.shutdown()
+            await self.node0.run(timeout=20)
+
+            logging.warning(f"""
+    {'#' * 80}
+
+    Finished file: {fn}
+
+    {'#' * 80}
+""")
+
         client.close()
         await client.wait_closed()