106 lines
3.4 KiB
TypeScript
106 lines
3.4 KiB
TypeScript
import { IntegrationHarness } from "@/pkg/testutil/integration-harness";
|
|
import type { V1ApisCreateApiRequest, V1ApisCreateApiResponse } from "@/routes/v1_apis_createApi";
|
|
import type { V1ApisDeleteApiRequest, V1ApisDeleteApiResponse } from "@/routes/v1_apis_deleteApi";
|
|
import type { V1KeysCreateKeyRequest, V1KeysCreateKeyResponse } from "@/routes/v1_keys_createKey";
|
|
import { V1KeysGetKeyResponse } from "@/routes/v1_keys_getKey";
|
|
import type { V1KeysVerifyKeyRequest, V1KeysVerifyKeyResponse } from "@/routes/v1_keys_verifyKey";
|
|
import { expect, test } from "vitest";
|
|
|
|
test("remaining consistently counts down", async () => {
|
|
const h = await IntegrationHarness.init();
|
|
const { key: rootKey } = await h.createRootKey(["*"]);
|
|
|
|
const createApiResponse = await h.post<V1ApisCreateApiRequest, V1ApisCreateApiResponse>({
|
|
url: `${h.baseUrl}/v1/apis.createApi`,
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${rootKey}`,
|
|
},
|
|
body: {
|
|
name: "scenario-test-pls-delete",
|
|
},
|
|
});
|
|
expect(createApiResponse.status).toEqual(200);
|
|
expect(createApiResponse.body.apiId).toBeDefined();
|
|
expect(createApiResponse.headers).toHaveProperty("unkey-request-id");
|
|
|
|
const remaining = 100;
|
|
|
|
const createKeyResponse = await h.post<V1KeysCreateKeyRequest, V1KeysCreateKeyResponse>({
|
|
url: `${h.baseUrl}/v1/keys.createKey`,
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${rootKey}`,
|
|
},
|
|
body: {
|
|
apiId: createApiResponse.body.apiId,
|
|
byteLength: 32,
|
|
prefix: "test",
|
|
remaining,
|
|
enabled: true,
|
|
},
|
|
});
|
|
expect(createKeyResponse.status).toEqual(200);
|
|
|
|
for (let i = remaining - 1; i >= 0; i--) {
|
|
const valid = await h.post<V1KeysVerifyKeyRequest, V1KeysVerifyKeyResponse>({
|
|
url: `${h.baseUrl}/v1/keys.verifyKey`,
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
},
|
|
body: {
|
|
apiId: createApiResponse.body.apiId,
|
|
key: createKeyResponse.body.key,
|
|
},
|
|
});
|
|
|
|
expect(valid.status).toEqual(200);
|
|
expect(valid.body.valid).toBe(true);
|
|
expect(valid.body.remaining).toEqual(i);
|
|
}
|
|
|
|
const invalid = await h.post<V1KeysVerifyKeyRequest, V1KeysVerifyKeyResponse>({
|
|
url: `${h.baseUrl}/v1/keys.verifyKey`,
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
},
|
|
body: {
|
|
apiId: createApiResponse.body.apiId,
|
|
key: createKeyResponse.body.key,
|
|
},
|
|
});
|
|
expect(invalid.status).toEqual(200);
|
|
expect(invalid.body.valid).toBe(false);
|
|
expect(invalid.body.remaining).toEqual(0);
|
|
|
|
// wait until the updates can propagate from the durable object to the db
|
|
await new Promise((r) => setTimeout(r, 2000));
|
|
|
|
const key = await h.get<V1KeysGetKeyResponse>({
|
|
url: `${h.baseUrl}/v1/keys.getKey?keyId=${createKeyResponse.body.keyId}`,
|
|
|
|
headers: {
|
|
Authorization: `Bearer ${rootKey}`,
|
|
},
|
|
});
|
|
expect(key.status).toEqual(200);
|
|
expect(key.body.id).toEqual(createKeyResponse.body.keyId);
|
|
expect(key.body.remaining).toBeDefined();
|
|
expect(key.body.remaining).toEqual(0);
|
|
|
|
/**
|
|
* Teardown
|
|
*/
|
|
const deleteApi = await h.post<V1ApisDeleteApiRequest, V1ApisDeleteApiResponse>({
|
|
url: `${h.baseUrl}/v1/apis.deleteApi`,
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${rootKey}`,
|
|
},
|
|
body: {
|
|
apiId: createApiResponse.body.apiId,
|
|
},
|
|
});
|
|
expect(deleteApi.status).toEqual(200);
|
|
}, 60_000);
|