tests: add accessCounter test

Author: SharzyL

src/handlers/handleWrite.ts

@@ -126,6 +126,7 @@ export async function handlePostOrPut(
       throw new WorkerError(400, `password should not contain newline`)
     }
   }
+  const passwd = passwdFromForm || genRandStr(DEFAULT_PASSWD_LEN)
 
   // check if name is legal
   if (nameFromForm !== undefined && !NAME_REGEX.test(nameFromForm)) {
@@ -186,10 +187,6 @@ export async function handlePostOrPut(
       pasteName = genRandStr(isPrivate ? PRIVATE_PASTE_NAME_LEN : PASTE_NAME_LEN)
     }
 
-    const passwd = passwdFromForm || genRandStr(DEFAULT_PASSWD_LEN)
-    if (passwd.length === 0) {
-      throw new WorkerError(400, "Empty passwd is not allowed")
-    }
     await createPaste(env, pasteName, content, {
       expirationSeconds,
       now,

test/accessCounter.spec.ts

@@ -0,0 +1,36 @@
+import { expect, it, beforeEach, vi, afterEach } from "vitest"
+import { genRandomBlob, upload, workerFetch } from "./testUtils.js"
+import { createExecutionContext, env, waitOnExecutionContext } from "cloudflare:test"
+import { PasteMetadata } from "../src/storage/storage"
+
+beforeEach(() => {
+  vi.spyOn(Math, "random").mockReturnValue(0)
+})
+
+afterEach(() => {
+  vi.restoreAllMocks()
+})
+
+it("increase access counter", async () => {
+  const ctx = createExecutionContext()
+  const content = genRandomBlob(1024)
+  const name = "abc"
+  const url = (await upload(ctx, { c: content, n: name })).url
+
+  async function getCounter() {
+    const paste = await env.PB.getWithMetadata<PasteMetadata>("~" + name)
+    return paste?.metadata?.accessCounter
+  }
+
+  expect(await getCounter()).toStrictEqual(0)
+
+  await workerFetch(ctx, url)
+  await waitOnExecutionContext(ctx)
+
+  expect(await getCounter()).toStrictEqual(1)
+
+  await workerFetch(ctx, url)
+  await waitOnExecutionContext(ctx)
+
+  expect(await getCounter()).toStrictEqual(2)
+})

test/roles.spec.ts

@@ -4,17 +4,59 @@ import { createExecutionContext } from "cloudflare:test"
 import { MetaResponse } from "../src/shared"
 import { genRandStr } from "../src/common"
 
+const testMd: string = `
+# Header 1
+
+This is the content of \`test.md\`
+
+<script>
+alert("Script should be removed")
+</script>
+
+## Header 2
+
+| abc | defghi |
+| :-: | -----: |
+| bar |    baz |
+
+**Bold**, \`Monospace\`, _Italics_, ~~Strikethrough~~, [URL](https://github.com)
+
+- A
+- A1
+- A2
+- B
+
+![Panty](https://shz.al/~panty.jpg)
+
+1. first
+2. second
+
+\`\`\`js
+console.log("hello world")
+\`\`\`
+
+> Quotation
+
+$$
+\\int_{-\\infty}^{\\infty} e^{-x^2} = \\sqrt{\\pi}
+$$
+`
+
 test("markdown with role a", async () => {
-  const testMd = `# Hello` // TODO: use a stronger test file
   const ctx = createExecutionContext()
   const url = (await upload(ctx, { c: testMd })).url
 
   const revisitResponse = await workerFetch(ctx, addRole(url, "a"))
   expect(revisitResponse.status).toStrictEqual(200)
   expect(revisitResponse.headers.get("Content-Type")).toStrictEqual("text/html;charset=UTF-8")
   const responseHtml = await revisitResponse.text()
-  expect(responseHtml.indexOf("<title>Hello</title>")).toBeGreaterThan(-1)
-  expect(responseHtml.indexOf("<h1>Hello</h1>")).toBeGreaterThan(-1)
+  expect(responseHtml.indexOf("<title>Header 1</title>")).toBeGreaterThan(-1)
+  expect(responseHtml.indexOf('<code class="language-js">')).toBeGreaterThan(-1)
+
+  const bigMd = "1".repeat(1024 * 1024)
+  const bigUrl = (await upload(ctx, { c: bigMd })).url
+  const bigResp = await (await workerFetch(ctx, addRole(bigUrl, "a"))).text()
+  expect(bigResp.indexOf("Untitled")).toBeGreaterThan(-1)
 })
 
 test("meta with role m", async () => {