add transaction creator tests

Author: devksingh4

tests/unit/functions/auditLog.test.ts

@@ -1,5 +1,5 @@
 import { expect, test, describe, vi } from "vitest";
-import { createAuditLogEntry } from "../../../src/api/functions/auditLog";
+import { createAuditLogEntry, buildAuditLogTransactPut } from "../../../src/api/functions/auditLog";
 import { mockClient } from "aws-sdk-client-mock";
 import { DynamoDBClient, PutItemCommand } from "@aws-sdk/client-dynamodb";
 import { afterEach, beforeEach } from "node:test";
@@ -45,6 +45,110 @@ describe("Audit Log tests", () => {
   });
 });
 
+describe("Audit Log Transaction tests", () => {
+  test("Audit log transaction item is correctly created with all fields", () => {
+    // Setup mock date
+    const mockDate = new Date(2025, 3, 20, 12, 0, 0);
+    const mockTimestamp = mockDate.getTime();
+    vi.spyOn(Date, 'now').mockReturnValue(mockTimestamp);
+
+    const timestamp = Math.floor(mockTimestamp / 1000);
+    const expireAt = timestamp + Math.floor((365 * 24 * 60 * 60 * 1000) / 1000);
+
+    // Create test payload
+    const payload = {
+      module: Modules.IAM,
+      actor: '[email protected]',
+      target: '[email protected]',
+      requestId: 'abcdef',
+      message: "Created user"
+    };
+
+    // Expected marshalled item
+    const expectedItem = marshall({
+      ...payload,
+      createdAt: timestamp,
+      expireAt: expireAt
+    });
+
+    // Call the function being tested
+    const transactItem = buildAuditLogTransactPut({ entry: payload });
+
+    // Verify the result
+    expect(transactItem).toStrictEqual({
+      Put: {
+        TableName: genericConfig.AuditLogTable,
+        Item: expectedItem
+      }
+    });
+  });
+
+  test("Audit log transaction item with minimal fields is correctly created", () => {
+    // Setup mock date
+    const mockDate = new Date(2025, 3, 20, 12, 0, 0);
+    const mockTimestamp = mockDate.getTime();
+    vi.spyOn(Date, 'now').mockReturnValue(mockTimestamp);
+
+    const timestamp = Math.floor(mockTimestamp / 1000);
+    const expireAt = timestamp + Math.floor((365 * 24 * 60 * 60 * 1000) / 1000);
+
+    // Create test payload with only required fields
+    const payload = {
+      module: Modules.IAM,
+      actor: '[email protected]',
+      message: "Deleted resource"
+    };
+
+    // Expected marshalled item
+    const expectedItem = marshall({
+      ...payload,
+      createdAt: timestamp,
+      expireAt: expireAt
+    });
+
+    // Call the function being tested
+    const transactItem = buildAuditLogTransactPut({ entry: payload });
+
+    // Verify the result
+    expect(transactItem).toStrictEqual({
+      Put: {
+        TableName: genericConfig.AuditLogTable,
+        Item: expectedItem
+      }
+    });
+  });
+
+  test("Audit log transaction item correctly calculates expiration timestamp", () => {
+    // Setup mock date
+    const mockDate = new Date(2025, 3, 20, 12, 0, 0);
+    const mockTimestamp = mockDate.getTime();
+    vi.spyOn(Date, 'now').mockReturnValue(mockTimestamp);
+
+    const timestamp = Math.floor(mockTimestamp / 1000);
+    // Manually calculate the expected expiration
+    const retentionDays = 365;
+    const secondsInDay = 24 * 60 * 60;
+    const millisecondsInDay = secondsInDay * 1000;
+    const expectedExpireAt = timestamp + Math.floor((retentionDays * millisecondsInDay) / 1000);
+
+    // Create test payload
+    const payload = {
+      module: Modules.IAM,
+      actor: '[email protected]',
+      message: "Modified settings"
+    };
+
+    // Call the function being tested
+    const transactItem = buildAuditLogTransactPut({ entry: payload });
+
+    // Extract and verify the expiration timestamp
+    const marshalledItem = transactItem.Put.Item;
+    const unmarshalledItem = require('@aws-sdk/util-dynamodb').unmarshall(marshalledItem);
+
+    expect(unmarshalledItem.expireAt).toBe(expectedExpireAt);
+  });
+});
+
 beforeEach(() => {
   ddbMock.reset();
 });