test: add workspaces test

Author: eoinsha

.projenrc.ts

@@ -6,12 +6,12 @@ const project = new awscdk.AwsCdkConstructLibrary({
   constructsVersion: '10.3.0',
   defaultReleaseBranch: 'main',
   jsiiVersion: '~5.5.0',
-  name: 'ft-python-lambda',
+  name: 'uv-python-lambda',
   projenrcTs: true,
-  repositoryUrl: 'git@github.com:fourtheorem/ft-python-lambda.git',
+  repositoryUrl: 'git@github.com:fourtheorem/uv-python-lambda.git',
   publishToPypi: {
-    distName: 'fourtheorem.ft-python-lambda',
-    module: 'fourtheorem.ft_python_lambda',
+    distName: 'fourtheorem.uv-python-lambda',
+    module: 'fourtheorem.uv_python_lambda',
   },
   // cdkVersion: '2.1.0',    /* CDK version to use. */
   // cdkDependencies: [],     /* CDK dependencies of this module. */

API.md

@@ -1,7 +1,47 @@
-# replace this
+# uv-python-lambda
 
+CDK Construct for Python Lambda Functions using [uv](https://docs.astral.sh/uv/)
 
-```
-error: failed to create directory `/.local/share/uv/python`
-```
-ensure uv.lock exists
+## Goals
+
+- ⚡️ Package and deploy Lambda Functions faster with `uv`'s speed
+- 📦 Support workspaces in a monorepo with [uv workspaces](https://docs.astral.sh/uv/concepts/workspaces/)
+
+## API
+
+See [API.md](API.md)
+
+## Example
+
+```python
+from fourtheorem.uv_python_lambda import PythonFunction
+from constructs import Construct
+
+# The root path should be relative to your CDK source file
+root_path = Path(__file__).parent.parent.parent
+
+
+class CdkStack(Stack):
+    def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
+        super().__init__(scope, construct_id, **kwargs)
+
+        fn = PythonFunction(
+          self,
+          "fn",
+          root_dir=str(root_path),
+          index="fetcher_lambda.py",
+          workspace_package="fetcher",  # Use a workspace package as the top-level Lambda entry point.
+          handler="handle_event",
+          bundling={
+              "asset_excludes": [
+                  ".venv/",
+                  "node_modules/",
+                  "cdk/",
+                  ".git/",
+                  ".idea/",
+                  "dist/",
+              ]
+          },
+          timeout=Duration.seconds(30),
+        )
+```

README.md

@@ -8,7 +8,7 @@ import { PythonFunction } from '../src';
 const resourcesPath = path.resolve(__dirname, 'resources');
 
 
-test('Create a function', async () => {
+test('Create a function from basic_app', async () => {
   const app = new App({});
   const stack = new Stack(app, 'test');
   new PythonFunction(stack, 'basic_app', {
@@ -30,11 +30,54 @@ test('Create a function', async () => {
     },
   });
   const functions = Object.values(template.findResources('AWS::Lambda::Function'));
-  expect(functions.length).toBe(1);
-  expect(functions[0].Properties.Handler).toBe('handler.lambda_handler');
-  expect(functions[0].Properties.Runtime).toBe('python3.12');
-  const [assetHash] = functions[0].Properties.Code.S3Key.split('.');
-  const assetPath = path.join(app.outdir, `asset.${assetHash}`);
-  const contents = await fs.readdir(assetPath);
+  expect(functions).toHaveLength(1);
+  const contents = await getFunctionAssetContents(functions[0], app);
   expect(contents).toContain('handler.py');
 });
+
+test('Create a function with workspaces_app', async () => {
+  const app = new App({});
+  const stack = new Stack(app, 'wstest');
+  new PythonFunction(stack, 'workspaces_app', {
+    rootDir: path.join(resourcesPath, 'workspaces_app'),
+    workspacePackage: 'app',
+    index: 'app_handler.py',
+    handler: 'handle_event',
+    runtime: Runtime.PYTHON_3_10,
+  });
+
+  const template = Template.fromStack(stack);
+  console.log(JSON.stringify(template.toJSON(), null, ' '));
+
+  template.hasResourceProperties('AWS::Lambda::Function', {
+    Handler: 'app_handler.handle_event',
+    Runtime: 'python3.10',
+    Code: {
+      S3Bucket: Match.anyValue(),
+      S3Key: Match.anyValue(),
+    },
+  });
+
+  const functions = Object.values(template.findResources('AWS::Lambda::Function'));
+  expect(functions).toHaveLength(1);
+  const contents = await getFunctionAssetContents(functions[0], app);
+  for (const entry of [
+    'app',
+    'common',
+    'pydantic',
+    'httpx',
+    '_common.pth',
+    'app_handler.py',
+  ]) {
+    expect(contents).toContain(entry);
+  }
+});
+
+// biome-ignore lint/suspicious/noExplicitAny: <explanation>
+async function getFunctionAssetContents(functionResource: any, app: App) {
+  const [assetHash] = functionResource.Properties.Code.S3Key.split('.');
+  const assetPath = path.join(app.outdir, `asset.${assetHash}`);
+  const contents = await fs.readdir(assetPath);
+  return contents;
+}
+

package.json

@@ -0,0 +1 @@
+3.10

test/function.test.ts

@@ -0,0 +1 @@
+3.10

test/resources/workspaces_app/.python-version

@@ -0,0 +1,10 @@
+from pydantic import BaseModel
+
+class Car(BaseModel):
+    brand: str
+    model: str
+    year: int
+
+def handle_event(event, context):
+    car = Car(brand="Toyota", model="Corolla", year=2020)
+    return car.model_dump()