summaryrefslogtreecommitdiff
path: root/Homework/cs5260/consumer/t/S3WidgetDao.spec.ts
blob: 5eb4602b1c541b4bd33818083c5d994aa0103047 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import { Widget } from "@/shared/schema";
import { expect, test, mock } from "bun:test";
import { S3WidgetKeyGenerator } from "@/consumer/dao/keys";
import {
  DeleteObjectCommand,
  GetObjectCommand,
  PutObjectCommand,
  S3Client,
} from "@aws-sdk/client-s3";
import { S3WidgetDAO } from "@/consumer/dao";
import { VoidLogger } from "@/shared/t";

const testBucket = "test-bucket";
const testWidget = Widget.parse({
  id: "testWidgetId",
  owner: "simponic",
  label: "A Widget",
  description: "Nothing to see here!",
});

const testKeyGenerator = new S3WidgetKeyGenerator().withPrefix("test-widgets");

const mockedGetSend = mock(async (req: GetObjectCommand) => {
  if (req.input.Key == "test-widgets/simponic/testWidgetId") {
    return {
      Body: {
        transformToString: () => {
          return JSON.stringify(testWidget);
        },
      },
    };
  }
  return {};
});

const mockedPutSendSuccessful = mock(async (req: PutObjectCommand) => {
  if (req.input.Key == "test-widgets/simponic/testWidgetId") {
    return {
      ChecksumSHA256: "laksdjf-lkejfoaj",
    };
  }
});

const mockedDeleteSendSuccessful = mock(async (req: DeleteObjectCommand) => {
  if (req.input.Key == "test-widgets/simponic/testWidgetId") {
    return {
      $metadata: true,
    };
  }
});

test("when given widget then retrieves from s3", async () => {
  const mockedClient = { send: mockedGetSend };
  const logger = new VoidLogger();

  const dao = new S3WidgetDAO(
    testBucket,
    mockedClient as unknown as S3Client,
    testKeyGenerator,
    logger,
  );

  const resp = await dao.retrieve({ id: "testWidgetId", owner: "simponic" });

  expect(resp?.label).toEqual(testWidget.label);
  expect(resp?.id).toEqual(testWidget.id);
});

test("when cannot retrieve widget then returns undefined", async () => {
  const mockedClient = { send: mockedGetSend };
  const logger = new VoidLogger();

  const dao = new S3WidgetDAO(
    testBucket,
    mockedClient as unknown as S3Client,
    testKeyGenerator,
    logger,
  );

  const resp = await dao.retrieve({
    id: "thisWidgetDoesNotExist",
    owner: "simponic",
  });

  expect(resp).toBeUndefined();
});

test("when given widget then puts in s3 and returns", async () => {
  const mockedClient = { send: mockedPutSendSuccessful };
  const logger = new VoidLogger();
  const dao = new S3WidgetDAO(
    testBucket,
    mockedClient as unknown as S3Client,
    testKeyGenerator,
    logger,
  );

  const resp = await dao.save(testWidget);
  expect(resp).toEqual(testWidget);
});

test("when given widget then deletes in s3 and returns", async () => {
  const mockedClient = { send: mockedDeleteSendSuccessful };
  const logger = new VoidLogger();

  const dao = new S3WidgetDAO(
    testBucket,
    mockedClient as unknown as S3Client,
    testKeyGenerator,
    logger,
  );

  const resp = await dao.delete(testWidget);
  expect(resp).toEqual(testWidget);
});