ci(*:skip) Add flower-simulation e2e tests

.github/workflows/e2e.yml

@@ -139,11 +139,11 @@ jobs:
         if: ${{ matrix.dataset }}
         run: python -c "${{ matrix.dataset }}"
       - name: Run edge client test
-        if: ${{ matrix.directory != 'bare-client-auth' }}
+        if: ${{ matrix.directory != 'bare-client-auth'}}
         run: ./../test_legacy.sh "${{ matrix.directory }}"
       - name: Run virtual client test
-        if: ${{ matrix.directory != 'bare-client-auth' }}
-        run: python simulation.py
+        if: ${{ matrix.directory != 'bare-client-auth' && matrix.directory != 'bare-https'}}
+        run: ./../test_simulation.sh "${{ matrix.directory }}"
       - name: Run simulation engine test
         if: ${{ matrix.directory == 'pytorch' || matrix.directory == 'tensorflow'}}
         run: python simulation_next.py

e2e/framework-pandas/client.py

@@ -1,3 +1,4 @@
+from pathlib import Path
 from typing import Dict, List, Tuple
 
 import numpy as np

e2e/test_simulation.sh

@@ -0,0 +1,35 @@
+#!/bin/bash
+set -e
+
+if [ -z "$1" ]; then
+  echo "Usage: $0 <framework>"
+  exit 1
+fi
+
+framework=$1
+
+case "$framework" in
+  framework-pandas)
+    server_app="server:app"
+    client_app="client:app"
+    app_dir="./"
+    ;;
+  *)
+    server_app="server:app"
+    client_app="${framework}.client:app"
+    app_dir="./.."
+    ;;
+esac
+
+echo flower-simulation --server-app $server_app --client-app $client_app --num-supernodes 2 --app-dir $app_dir
+
+timeout 2m flower-simulation --server-app $server_app --client-app $client_app --num-supernodes 2 --app-dir $app_dir &
+pid=$!
+
+wait $pid
+res=$?
+
+if [[ "$res" = "0" ]];
+  then echo "Training worked correctly";
+  else echo "Training had an issue" && exit 1;
+fi