test: update test cases for badges

Author: trueberryless

test/e2e/badge.spec.ts

@@ -12,34 +12,80 @@ async function fetchBadge(page: { request: { get: (url: string) => Promise<any>
 }
 
 test.describe('badge API', () => {
-  test('unscoped package badge renders SVG', async ({ page, baseURL }) => {
-    const url = toLocalUrl(baseURL, '/api/registry/badge/nuxt')
-    const { response, body } = await fetchBadge(page, url)
-
-    expect(response.status()).toBe(200)
-    expect(response.headers()['content-type']).toContain('image/svg+xml')
-    expect(body).toContain('<svg')
-    expect(body).toContain('nuxt')
+  const badgeMap: Record<string, string> = {
+    'version': 'version',
+    'license': 'license',
+    'size': 'install size',
+    'downloads': 'downloads/mo',
+    'downloads-week': 'downloads/wk',
+    'vulnerabilities': 'vulns',
+    'dependencies': 'dependencies',
+    'updated': 'updated',
+    'engines': 'node',
+    'types': 'types',
+    'created': 'created',
+    'maintainers': 'maintainers',
+    'deprecated': 'status',
+  }
+
+  for (const [type, expectedLabel] of Object.entries(badgeMap)) {
+    test.describe(`${type} badge`, () => {
+      test('renders correct label', async ({ page, baseURL }) => {
+        const url = toLocalUrl(baseURL, `/api/registry/badge/${type}/nuxt`)
+        const { response, body } = await fetchBadge(page, url)
+
+        expect(response.status()).toBe(200)
+        expect(response.headers()['content-type']).toContain('image/svg+xml')
+        expect(body).toContain(expectedLabel)
+      })
+
+      test('scoped package renders successfully', async ({ page, baseURL }) => {
+        const url = toLocalUrl(baseURL, `/api/registry/badge/${type}/@nuxt/kit`)
+        const { response } = await fetchBadge(page, url)
+
+        expect(response.status()).toBe(200)
+      })
+
+      test('explicit version badge renders successfully', async ({ page, baseURL }) => {
+        const url = toLocalUrl(baseURL, `/api/registry/badge/${type}/nuxt/v/3.12.0`)
+        const { response, body } = await fetchBadge(page, url)
+
+        expect(response.status()).toBe(200)
+        if (type === 'version') {
+          expect(body).toContain('3.12.0')
+        }
+      })
+
+      test('respects name=true parameter', async ({ page, baseURL }) => {
+        const packageName = 'nuxt'
+        const url = toLocalUrl(baseURL, `/api/registry/badge/${type}/${packageName}?name=true`)
+        const { body } = await fetchBadge(page, url)
+
+        expect(body).toContain(packageName)
+        expect(body).not.toContain(expectedLabel)
+      })
+    })
+  }
+
+  test('custom color parameter is applied to SVG', async ({ page, baseURL }) => {
+    const customColor = 'ff69b4'
+    const url = toLocalUrl(baseURL, `/api/registry/badge/version/nuxt?color=${customColor}`)
+    const { body } = await fetchBadge(page, url)
+
+    expect(body).toContain(`fill="#${customColor}"`)
   })
 
-  test('scoped package badge renders SVG', async ({ page, baseURL }) => {
-    const url = toLocalUrl(baseURL, '/api/registry/badge/@nuxt/kit')
-    const { response, body } = await fetchBadge(page, url)
+  test('invalid badge type defaults to version strategy', async ({ page, baseURL }) => {
+    const url = toLocalUrl(baseURL, '/api/registry/badge/invalid-type/nuxt')
+    const { body } = await fetchBadge(page, url)
 
-    expect(response.status()).toBe(200)
-    expect(response.headers()['content-type']).toContain('image/svg+xml')
-    expect(body).toContain('<svg')
-    expect(body).toContain('@nuxt/kit')
+    expect(body).toContain('version')
   })
 
-  test('explicit version badge includes requested version', async ({ page, baseURL }) => {
-    const url = toLocalUrl(baseURL, '/api/registry/badge/nuxt/v/3.12.0')
-    const { response, body } = await fetchBadge(page, url)
+  test('missing package returns 404', async ({ page, baseURL }) => {
+    const url = toLocalUrl(baseURL, '/api/registry/badge/version/')
+    const { response } = await fetchBadge(page, url)
 
-    expect(response.status()).toBe(200)
-    expect(response.headers()['content-type']).toContain('image/svg+xml')
-    expect(body).toContain('<svg')
-    expect(body).toContain('nuxt')
-    expect(body).toContain('3.12.0')
+    expect(response.status()).toBe(404)
   })
 })