feat: add test for ping controller

.gitignore

@@ -1,2 +1,3 @@
 .env
-node_modules
+node_modules
+coverage

src/controllers/auth-middleware.ts

@@ -0,0 +1,36 @@
+import { IHttpServerComponent } from "@well-known-components/interfaces";
+import { Context, StatusCode } from "../types";
+
+async function withAuthTokenValidation(
+  context: IHttpServerComponent.DefaultContext<Context<string>>,
+  next: () => Promise<IHttpServerComponent.IResponse>
+): Promise<IHttpServerComponent.IResponse> {
+  // Validate Authorization header
+  const authHeader = context.request.headers.get("Authorization");
+  if (!authHeader || !authHeader.startsWith("Bearer ")) {
+    return {
+      status: StatusCode.UNAUTHORIZED,
+      body: {
+        ok: false,
+        message: "Missing or invalid Authorization header",
+      },
+    };
+  }
+
+  const token = authHeader.slice(7); // Extract token after 'Bearer '
+  const expectedToken = process.env.AUTH_TOKEN;
+
+  if (token !== expectedToken) {
+    return {
+      status: StatusCode.UNAUTHORIZED,
+      body: {
+        ok: false,
+        message: "Invalid authorization token",
+      },
+    };
+  }
+
+  return next();
+}
+
+export { withAuthTokenValidation };

src/controllers/handlers/ping-handler.ts

@@ -7,7 +7,7 @@ export async function pingHandler(context: Pick<HandlerContextWithPath<'metrics'
     components: { metrics }
   } = context
 
-  metrics.increment('wkc_logger_logs_total', {
+  metrics.increment('test_ping_counter', {
     pathname: url.pathname
   })
 

src/ports/squids/component.ts

@@ -39,7 +39,7 @@ export async function createSubsquidComponent({
       const servicesResponse = await client.send(listServicesCommand)
 
       const serviceArns = servicesResponse.serviceArns || []
-      const squidServices = serviceArns.filter(arn => arn.includes('squid-server'))
+      const squidServices = serviceArns.filter(arn => arn.includes('-squid-server'))
 
       // Step 2: Get tasks for each service and fetch task IPs
       const results: Squid[] = []

src/types.ts

@@ -6,8 +6,8 @@ import type {
   IBaseComponent,
   IMetricsComponent
 } from '@well-known-components/interfaces'
-import { metricDeclarations } from '@well-known-components/logger/dist/metrics'
 import { IPgComponent } from '@well-known-components/pg-component'
+import { metricDeclarations } from './metrics'
 import { ISquidComponent } from './ports/squids/types'
 
 export type GlobalContext = {

test/unit/ping-controller.spec.ts

@@ -0,0 +1,30 @@
+import { createTestMetricsComponent } from '@well-known-components/metrics'
+import { pingHandler } from '../../src/controllers/handlers/ping-handler'
+import { metricDeclarations } from '../../src/metrics'
+
+describe('ping-controller-unit', () => {
+  it('must return the pathname of a URL', async () => {
+    const url = new URL('https://github.com/well-known-components')
+    const metrics = createTestMetricsComponent(metricDeclarations)
+    expect((await metrics.getValue('test_ping_counter')).values).toEqual([])
+    expect(await pingHandler({ url, components: { metrics } })).toEqual({ body: url.pathname })
+    expect((await metrics.getValue('test_ping_counter')).values).toEqual([{ labels: { pathname: '/well-known-components' }, value: 1 }])
+  })
+
+  it('metrics should create a brand new registry', async () => {
+    const url = new URL('https://github.com/well-known-components')
+    const metrics = createTestMetricsComponent(metricDeclarations)
+    expect((await metrics.getValue('test_ping_counter')).values).toEqual([])
+    expect(await pingHandler({ url, components: { metrics } })).toEqual({ body: url.pathname })
+    expect((await metrics.getValue('test_ping_counter')).values).toEqual([{ labels: { pathname: '/well-known-components' }, value: 1 }])
+  })
+
+  it('calling twice should increment twice the metrics', async () => {
+    const url = new URL('https://github.com/well-known-components')
+    const metrics = createTestMetricsComponent(metricDeclarations)
+    expect((await metrics.getValue('test_ping_counter')).values).toEqual([])
+    expect(await pingHandler({ url, components: { metrics } })).toEqual({ body: url.pathname })
+    expect(await pingHandler({ url, components: { metrics } })).toEqual({ body: url.pathname })
+    expect((await metrics.getValue('test_ping_counter')).values).toEqual([{ labels: { pathname: '/well-known-components' }, value: 2 }])
+  })
+})