chore(tests): add basic unit test

Author: eoinsha

README.md

@@ -1 +1,7 @@
-# replace this
+# replace this
+
+
+```
+error: failed to create directory `/.local/share/uv/python`
+```
+ensure uv.lock exists

src/bundling.ts

@@ -146,9 +146,9 @@ export class Bundling {
     commands.push(...[
       `rsync -rLv ${excludeArgs.join(' ')} ${options.inputDir}/ ${options.outputDir}`,
       `cd ${options.outputDir}`, // uv pip install needs to be run from here for editable deps to relative paths to be resolved
-      `VIRTUAL_ENV=/tmp/venv uv sync ${uvCommonArgs} ${uvPackageArgs} --compile-bytecode --no-dev --frozen --no-editable --link-mode=copy`,
-      `VIRTUAL_ENV=/tmp/venv uv export ${uvCommonArgs} ${uvPackageArgs} --no-dev --frozen --no-editable > ${reqsFile}`,
-      `uv pip install -r ${reqsFile} --target ${options.outputDir} --reinstall --compile-bytecode --link-mode=copy --editable $(grep -e "^\./" ${reqsFile})`,
+      `uv sync ${uvCommonArgs} ${uvPackageArgs} --python-preference=only-system --compile-bytecode --no-dev --frozen --no-editable --link-mode=copy`,
+      `uv export ${uvCommonArgs} ${uvPackageArgs} --no-dev --frozen --no-editable > ${reqsFile}`,
+      `uv pip install -r ${reqsFile} --target ${options.outputDir} --reinstall --compile-bytecode --link-mode=copy`, // --editable $(grep -e "^\./" ${reqsFile})`,
       `rm -rf ${options.outputDir}/.venv`,
     ]);
     commands.push(

src/function.ts

@@ -93,15 +93,17 @@ export class PythonFunction extends Function {
     });
 
     const assetPath = ((this.node.defaultChild) as CfnFunction).getMetadata('aws:asset:path');
-    const codePath = path.join(process.env.CDK_OUTDIR as string, assetPath);
+    if (assetPath) { // TODO - remove - we always need one
+      const codePath = path.join(process.env.CDK_OUTDIR as string, assetPath);
 
-    const pythonPaths = getPthFilePaths(codePath);
+      const pythonPaths = getPthFilePaths(codePath);
 
-    if (pythonPaths.length > 0) {
-      let pythonPathValue = environment.PYTHONPATH;
-      const addedPaths = pythonPaths.join(':');
-      pythonPathValue = pythonPathValue ? `${pythonPathValue}:${addedPaths}` : addedPaths;
-      this.addEnvironment('PYTHONPATH', pythonPathValue);
+      if (pythonPaths.length > 0) {
+        let pythonPathValue = environment.PYTHONPATH;
+        const addedPaths = pythonPaths.join(':');
+        pythonPathValue = pythonPathValue ? `${pythonPathValue}:${addedPaths}` : addedPaths;
+        this.addEnvironment('PYTHONPATH', pythonPathValue);
+      }
     }
   }
 }

test/function.test.ts

@@ -0,0 +1,40 @@
+import * as fs from 'node:fs/promises';
+import * as path from 'node:path';
+import { App, Stack } from 'aws-cdk-lib';
+import { Match, Template } from 'aws-cdk-lib/assertions';
+import { Runtime } from 'aws-cdk-lib/aws-lambda';
+import { PythonFunction } from '../src';
+
+const resourcesPath = path.resolve(__dirname, 'resources');
+
+
+test('Create a function', async () => {
+  const app = new App({});
+  const stack = new Stack(app, 'test');
+  new PythonFunction(stack, 'basic_app', {
+    rootDir: path.join(resourcesPath, 'basic_app'),
+    index: 'handler.py',
+    handler: 'lambda_handler',
+    runtime: Runtime.PYTHON_3_12,
+  });
+
+  const template = Template.fromStack(stack);
+  console.log(JSON.stringify(template.toJSON(), null, ' '));
+
+  template.hasResourceProperties('AWS::Lambda::Function', {
+    Handler: 'handler.lambda_handler',
+    Runtime: 'python3.12',
+    Code: {
+      S3Bucket: Match.anyValue(),
+      S3Key: Match.anyValue(),
+    },
+  });
+  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(contents).toContain('handler.py');
+});

test/hello.test.ts

@@ -0,0 +1 @@
+3.12

test/resources/basic_app/.python-version

@@ -0,0 +1,5 @@
+def lambda_handler(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }

test/resources/basic_app/README.md

@@ -0,0 +1,7 @@
+[project]
+name = "basic-app"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = []