Added test env, local execution tests

Author: 3coins

.github/workflows/rust.yml

@@ -16,7 +16,50 @@ jobs:
 
     steps:
     - uses: actions/checkout@v4
+
+    - name: Set up Rust cache
+      uses: actions/cache@v4
+      with:
+        path: |
+          ~/.cargo/bin/
+          ~/.cargo/registry/index/
+          ~/.cargo/registry/cache/
+          ~/.cargo/git/db/
+          target/
+        key: ${{ runner.os }}-cargo-${{ hashFiles('**/Cargo.lock') }}
+        restore-keys: |
+          ${{ runner.os }}-cargo-
+
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.11'
+
+    - name: Install uv
+      uses: astral-sh/setup-uv@v5
+      with:
+        enable-cache: true
+
+    - name: Cache Python test venv
+      uses: actions/cache@v4
+      with:
+        path: tests/.test-venv
+        key: ${{ runner.os }}-test-venv-${{ hashFiles('requirements.txt') }}
+        restore-keys: |
+          ${{ runner.os }}-test-venv-
+
+    - name: Setup test environment
+      run: ./tests/setup_test_env.sh
+      continue-on-error: true
+
     - name: Build
       run: cargo build --verbose
-    - name: Run tests
-      run: cargo test --verbose
+
+    - name: Run local mode tests
+      run: cargo test --test integration_local_mode --verbose
+
+    - name: Run execution tests
+      run: cargo test --test integration_execution --verbose
+
+    - name: Run all other tests
+      run: cargo test --lib --bins --verbose

.gitignore

@@ -27,3 +27,4 @@ target
 
 .ipynb_checkpoints/
 .jupyter/
+tests/.test-venv

requirements.txt

@@ -1,9 +1,12 @@
-# Python dependencies for nb-cli local execution
+# Python dependencies for nb-cli local execution and testing
 # Install with: pip install -r requirements.txt
 
 # Core dependencies for local notebook execution
 nbclient>=0.8.0
 nbformat>=5.9.0
 
+# Required for execution tests
+ipykernel>=6.0.0
+
 # Note: These are only required for local mode execution
 # Remote mode (connecting to Jupyter Server) doesn't need these

tests/README.md

@@ -6,30 +6,84 @@ This directory contains integration tests for the nb-cli tool's local mode opera
 
 ```
 tests/
-├── integration_local_mode.rs  # Main integration test suite
+├── integration_local_mode.rs   # Core CLI operations (no execution)
+├── integration_execution.rs    # Execution tests (requires Python)
+├── test_helpers.rs             # Test utilities and venv setup
+├── setup_test_env.sh           # Script to setup test environment
 ├── fixtures/                   # Test notebook fixtures
-│   ├── empty.ipynb            # Empty notebook with no cells
-│   ├── basic.ipynb            # Basic notebook with one empty cell
-│   ├── with_code.ipynb        # Notebook with code cells
-│   ├── mixed_cells.ipynb      # Notebook with mixed cell types
-│   └── with_outputs.ipynb     # Notebook with cell outputs
+│   ├── empty.ipynb             # Empty notebook with no cells
+│   ├── basic.ipynb             # Basic notebook with one empty cell
+│   ├── with_code.ipynb         # Notebook with code cells
+│   ├── mixed_cells.ipynb       # Notebook with mixed cell types
+│   ├── with_outputs.ipynb      # Notebook with cell outputs
+│   ├── for_execution.ipynb     # Simple execution test notebook
+│   └── with_error.ipynb        # Notebook with error cell
+├── .test-venv/                 # Test virtual environment (auto-created)
 └── README.md                   # This file
 ```
 
 ## Running Tests
 
-Run all integration tests:
+### Setup (One Time)
+
+For execution tests, you need to setup the Python environment:
+
+```bash
+# Option 1: Use the setup script (requires uv)
+./tests/setup_test_env.sh
+
+# Option 2: Manual setup
+uv venv tests/.test-venv
+tests/.test-venv/bin/pip install nbclient nbformat ipykernel
+```
+
+**Note**: If you don't have `uv` installed:
+```bash
+curl -LsSf https://astral.sh/uv/install.sh | sh
+```
+
+### Run All Tests
+
+Run all integration tests (both local and execution):
+```bash
+cargo test
+```
+
+### Run Specific Test Suites
+
+Run only local mode tests (no Python required):
 ```bash
 cargo test --test integration_local_mode
 ```
 
-Run specific tests:
+Run only execution tests (requires Python venv):
+```bash
+cargo test --test integration_execution
+```
+
+### Run Specific Tests
+
 ```bash
 cargo test --test integration_local_mode test_create_empty_notebook
-cargo test --test integration_local_mode test_add_cell
+cargo test --test integration_execution test_execute_single_cell
 ```
 
-Run tests with output:
+### Run with Output
+
 ```bash
 cargo test --test integration_local_mode -- --nocapture
+cargo test --test integration_execution -- --nocapture
+```
+
+### Execution Tests Behavior
+
+Execution tests will automatically:
+- ✅ **Skip gracefully** if Python environment is not set up
+- ✅ **Auto-create venv** with `uv` if available
+- ✅ **Install dependencies** (nbclient, nbformat, ipykernel)
+- ✅ **Use isolated environment** (tests/.test-venv)
+
+If execution tests are skipped, you'll see:
+```
+⚠️  Skipping test: execution environment not available
 ```

tests/fixtures/for_execution.ipynb

@@ -0,0 +1,43 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cell-1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "x = 42"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cell-2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "y = x + 10"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cell-3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "print(f'Result: {y}')"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

tests/fixtures/with_error.ipynb

@@ -0,0 +1,33 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cell-1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "valid_code = 123"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cell-2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "undefined_variable"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}