fix: fail snapshot not always created

ffs-updatetest.py

@@ -139,9 +139,14 @@ with create_snapshot_context() as snapshot:
             logging.error("Not running the expected version, aborting")
             sys.exit(1)
         wait_for_gw_connection()
+        try:
             run_autoupdate()
+            wait_for_gw_connection()
             if not assert_release(args.after_version):
                 logging.error("Not running the expected version after update, aborting")
+                raise ValueError("Not running the expected version after update, aborting")
+        except:
+            logging.error("Something seems to have failed, saving state and exiting.")
             create_snapshot(prefix="test_fail_{}".format(i))
             sys.exit(1)
         restore_snapshot(snapshot)