From 52bad657d511ac25a06615bba7bc89ae586fce24 Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Wed, 21 Aug 2024 10:50:42 +0200
Subject: [PATCH] backend-common: add test for identity service shim

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../src/compat/legacy/legacy.test.ts          | 57 ++++++++++++++++++-
 1 file changed, 56 insertions(+), 1 deletion(-)

diff --git a/packages/backend-common/src/compat/legacy/legacy.test.ts b/packages/backend-common/src/compat/legacy/legacy.test.ts
index 08a645f660..d052b500b8 100644
--- a/packages/backend-common/src/compat/legacy/legacy.test.ts
+++ b/packages/backend-common/src/compat/legacy/legacy.test.ts
@@ -18,9 +18,14 @@ import {
   coreServices,
   createBackendPlugin,
 } from '@backstage/backend-plugin-api';
-import { mockServices, startTestBackend } from '@backstage/backend-test-utils';
+import {
+  mockCredentials,
+  mockServices,
+  startTestBackend,
+} from '@backstage/backend-test-utils';
 import { EventEmitter } from 'events';
 import { Router } from 'express';
+import request from 'supertest';
 import { createLegacyAuthAdapters } from '..';
 import { legacyPlugin } from './legacy';
 // eslint-disable-next-line @backstage/no-relative-monorepo-imports
@@ -115,4 +120,54 @@ describe('legacyPlugin', () => {
 
     await done;
   });
+
+  it('can auth users with the identity service shim', async () => {
+    const backend = await startTestBackend({
+      features: [
+        mockServices.rootConfig.factory({
+          data: {
+            backend: {
+              auth: {
+                keys: [
+                  {
+                    secret: 'test',
+                  },
+                ],
+              },
+            },
+          },
+        }),
+        legacyPlugin(
+          'test',
+          Promise.resolve({
+            async default({ identity }) {
+              const router = Router();
+
+              router.get('/', async (req, res) => {
+                const user = await identity.getIdentity({ request: req });
+                res.json(user);
+              });
+
+              return router;
+            },
+          }),
+        ),
+      ],
+    });
+
+    const res = await request(backend.server)
+      .get('/api/test')
+      .set('authorization', mockCredentials.user.header());
+
+    const mockUserRef = mockCredentials.user().principal.userEntityRef;
+    expect(res.status).toBe(200);
+    expect(res.body).toEqual({
+      token: mockCredentials.user.token(),
+      identity: {
+        type: 'user',
+        userEntityRef: mockUserRef,
+        ownershipEntityRefs: [mockUserRef],
+      },
+    });
+  });
 });