From fdaf4868fbe00742840d0c63611e74a448b72e41 Mon Sep 17 00:00:00 2001
From: Samruddhi Khandale <skhandale@microsoft.com>
Date: Thu, 8 Sep 2022 22:08:41 +0000
Subject: [PATCH] fix test

---
 test/python/install_jupyterlab_os_provided.sh | 11 +++++++++++
 test/python/install_os_provided_python.sh     |  1 -
 test/python/scenarios.json                    | 16 +++++++++++++---
 3 files changed, 24 insertions(+), 4 deletions(-)
 create mode 100644 test/python/install_jupyterlab_os_provided.sh

diff --git a/test/python/install_jupyterlab_os_provided.sh b/test/python/install_jupyterlab_os_provided.sh
new file mode 100644
index 0000000..cdf6705
--- /dev/null
+++ b/test/python/install_jupyterlab_os_provided.sh
@@ -0,0 +1,11 @@
+#!/bin/bash
+
+set -e
+
+# Optional: Import test library
+source dev-container-features-test-lib
+
+check "jupyter lab version" jupyter lab --version
+
+# Report result
+reportResults
diff --git a/test/python/install_os_provided_python.sh b/test/python/install_os_provided_python.sh
index 3c4a8f6..50a617e 100644
--- a/test/python/install_os_provided_python.sh
+++ b/test/python/install_os_provided_python.sh
@@ -9,7 +9,6 @@ check "python3 is installed" python3 --version
 check "python is installed" python --version
 check "pip is installed" pip --version
 check "pip is installed" pip3 --version
-check "jupyter lab version" jupyter lab --version
 
 check "node is installed" node --version
 
diff --git a/test/python/scenarios.json b/test/python/scenarios.json
index 561d8f6..0caaeee 100644
--- a/test/python/scenarios.json
+++ b/test/python/scenarios.json
@@ -36,15 +36,25 @@
             }
         }
     },
-    "install_os_provided_python": {
-        "image": "mcr.microsoft.com/devcontainers/base:0-bullseye",
+    "install_jupyterlab_os_provided": {
+        "image": "mcr.microsoft.com/vscode/devcontainers/base:focal",
+        "remoteUser": "vscode",
         "features": {
-            "node": "latest",
+            "common-utils": {
+                "username": "vscode"
+            },
             "python": {
                 "version": "os-provided",
                 "installJupyterlab": true,
                 "configureJupyterlabAllowOrigin": "*"
             }
         }
+    },
+    "install_os_provided_python": {
+        "image": "mcr.microsoft.com/devcontainers/base:0-bullseye",
+        "features": {
+            "node": "latest",
+            "python": "os-provided"
+        }
     }
 }