test: fix imports

Makefile

@@ -22,3 +22,21 @@ eslint:
 	@echo -e "$(TARGET_COLOR)Running eslint $$(npx eslint --version)$(NO_COLOR)"
 	@npx eslint .; \
 	echo "Passed"
+
+validate-package:
+	@echo -e "$(TARGET_COLOR)Checking package content$(NO_COLOR)"
+	@npm publish --dry-run 2>&1 | tee publish_output.txt
+	@\
+	FILES_TO_CHECK="lambda/code.zip lib/index.d.ts lib/index.js lib/types.d.ts lib/types.js"; \
+	MISSING_FILES=""; \
+	for file in $$FILES_TO_CHECK; do \
+		if ! grep -q $$file publish_output.txt; then \
+			MISSING_FILES="$$MISSING_FILES $$file"; \
+		fi; \
+	done; \
+	if [ -n "$$MISSING_FILES" ]; then \
+		echo "❌ The following files are NOT included in the package:$$MISSING_FILES"; \
+		rm publish_output.txt; \
+		exit 1; \
+	fi
+	@rm publish_output.txt

test/Makefile

@@ -5,7 +5,6 @@ clean:
 
 install: package
 	@echo Installing dependencies...
-	@cd .. && make install
 	@npm install ../cdk-athena-test.tgz --no-save --prefer-offline --cache ../.npm
 	@npm list
 
@@ -33,6 +32,6 @@ package:
 	@echo Building npm tgz...
 	@ \
 	cd .. && \
-	make build && \
+	make install build && \
 	npm pack && \
 	mv cdk-athena-*.tgz cdk-athena-test.tgz

test/lib/index.ts

@@ -8,8 +8,7 @@ import {
   Tags,
 } from 'aws-cdk-lib';
 import { Construct } from 'constructs';
-
-import { EncryptionOption, LogLevel, NamedQuery, WorkGroup } from '../../lib';
+import { EncryptionOption, LogLevel, NamedQuery, WorkGroup } from 'cdk-athena';
 
 export class TestStack extends Stack {
   constructor(scope: Construct, id: string, props?: StackProps) {