From 0fe9e73429c2708491170c6e6dd0d3359141607c Mon Sep 17 00:00:00 2001
From: Tim Li <ltim@uber.com>
Date: Tue, 26 Aug 2025 12:29:01 -0700
Subject: [PATCH] Add CI job for unit test

---
 .github/workflows/ci_checks.yml | 22 +++++++++++++++++++++-
 pyproject.toml                  |  1 +
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci_checks.yml b/.github/workflows/ci_checks.yml
index fc150da..0340f55 100644
--- a/.github/workflows/ci_checks.yml
+++ b/.github/workflows/ci_checks.yml
@@ -19,7 +19,7 @@ jobs:
         uses: astral-sh/setup-uv@v1
       - name: Install dependencies
         run: |
-          uv sync
+          uv sync --extra dev
       - name: Ensure Proto Files changes are committed
         run: |
           uv run python scripts/generate_proto.py
@@ -34,3 +34,23 @@ jobs:
       - name: Run Ruff linter
         run: |
           uv tool run ruff check
+
+  test:
+    name: Unit Tests
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          submodules: true
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.13"
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v1
+      - name: Install dependencies
+        run: |
+          uv sync --extra dev
+      - name: Run unit tests
+        run: |
+          uv run python -m pytest tests/ -v
diff --git a/pyproject.toml b/pyproject.toml
index b3a1dba..8db2705 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -133,6 +133,7 @@ markers = [
     "slow: marks tests as slow (deselect with '-m \"not slow\"')",
     "integration: marks tests as integration tests",
     "unit: marks tests as unit tests",
+    "asyncio: marks tests as async tests",
 ]
 
 [tool.coverage.run]