From b5f24aac849bcda2a59e306387a56d5430d1e937 Mon Sep 17 00:00:00 2001
From: Diego Lovison <diegolovison@gmail.com>
Date: Mon, 15 Jul 2024 16:25:17 -0300
Subject: [PATCH] test: Add script and GitHub Action for kfp-kubernetes-library

Signed-off-by: Diego Lovison <diegolovison@gmail.com>
---
 .github/workflows/kfp-kubernetes-test.yaml    | 37 +++++++++++++++++++
 test/presubmit-test-kfp-kubernetes-library.sh | 16 ++++++--
 2 files changed, 49 insertions(+), 4 deletions(-)
 create mode 100644 .github/workflows/kfp-kubernetes-test.yaml

diff --git a/.github/workflows/kfp-kubernetes-test.yaml b/.github/workflows/kfp-kubernetes-test.yaml
new file mode 100644
index 000000000000..f2bab6b71c01
--- /dev/null
+++ b/.github/workflows/kfp-kubernetes-test.yaml
@@ -0,0 +1,37 @@
+name: KFP K8s Platform Tests
+
+on:
+  push:
+    branches: [master]
+  pull_request:
+    paths:
+      - '.github/workflows/kfp-kubernetes-library.yaml'
+      - 'sdk/python/**'
+      - 'api/v2alpha1/**'
+      - 'kubernetes_platform/**'
+      - 'test/presubmit-test-kfp-kubernetes-library.sh'
+
+jobs:
+  kfp-kubernetes-library-test:
+    runs-on: ubuntu-24.04
+    strategy:
+      matrix:
+        python: [
+          { 'version': '3.8' },
+          { 'version': '3.9' },
+          { 'version': '3.10' },
+          { 'version': '3.11' },
+          { 'version': '3.12' }
+        ]    
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{matrix.python.version}}
+
+      - name: Run tests
+        # original file is using sudo
+        run: ./test/presubmit-test-kfp-kubernetes-library.sh
diff --git a/test/presubmit-test-kfp-kubernetes-library.sh b/test/presubmit-test-kfp-kubernetes-library.sh
index 30a688872ef1..3bf7eebcba87 100755
--- a/test/presubmit-test-kfp-kubernetes-library.sh
+++ b/test/presubmit-test-kfp-kubernetes-library.sh
@@ -15,11 +15,21 @@
 
 source_root=$(pwd)
 
+user_id=`id -u`
+if [$user_id -eq 0]; then
+  apt-get update && apt-get install -y protobuf-compiler
+  # rust needed for transitive deps in dev extras on Python:3.12
+  apt-get install rustc -y
+else
+  sudo apt-get update && sudo apt-get install -y protobuf-compiler
+  # rust needed for transitive deps in dev extras on Python:3.12
+  sudo apt-get install rustc -y
+fi
+
 pip install --upgrade pip
 pip install wheel
-
 pip install sdk/python
-apt-get update && apt-get install -y protobuf-compiler
+
 pushd api
 make clean python
 popd
@@ -31,7 +41,5 @@ pushd "$source_root/kubernetes_platform"
 make clean python
 popd
 
-# rust needed for transitive deps in dev extras on Python:3.12
-apt-get install rustc -y
 pip install -e "$source_root/kubernetes_platform/python[dev]"
 pytest "$source_root/kubernetes_platform/python/test" -n auto