function.test.w

bring cloud;
bring http;
bring expect;
bring util;
bring dynamodb;
bring sns;
bring ses;
bring fs;
bring "./lib.w" as python;

class CustomLiftable impl python.ILiftable {
  pub liftData(): Json {
    return {
      "info": "CustomData",
    };
  }
}

let table = new dynamodb.Table(
  attributes: [
    {
      name: "id",
      type: "S",
    },
  ],
  hashKey: "id",
) as "table1";
let emailClient = new ses.EmailService({});
let mobileClient = new sns.MobileNotifications();
let bucket = new cloud.Bucket();
bucket.addObject("test.txt", "Hello, world!");

let func = new cloud.Function(new python.InflightFunctionHandler(
  path: fs.join(@dirname, "./test-assets"),
  handler: "main.handler",
  lift: {
    "bucket": {
      obj: bucket,
      allow: ["get", "put"],
    },
    "table": {
      obj: table,
      allow: ["get", "put"],
    },
    "sms": {
      obj: mobileClient,
      allow: ["publish"],
    },
    "email": {
      obj: emailClient,
      allow: ["sendEmail"],
    },
    "custom": {
      obj: new CustomLiftable(),
      allow: [],
    }
  },
), { env: { "FOO": "bar" } });

new std.Test(inflight () => {
  table.put(Item: { id: "test", body: "dynamoDbValue" });

  let res = func.invoke("function1");
  log("res: {res ?? "null"}");
  expect.equal(Json.parse(res!).get("body"), "Hello!");
  expect.equal(bucket.get("test.txt"), "Hello, world!function1bardynamoDbValueCustomData");
}, timeout: 3m) as "invokes the function";