From 6f10c1aedd974c95d77ce229c4d18629015ca602 Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Tue, 19 Apr 2022 13:31:20 +0200
Subject: [PATCH] backend-common: added tests for UrlReaderPredicateMux

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../src/reading/UrlReaderPredicateMux.test.ts | 87 +++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 packages/backend-common/src/reading/UrlReaderPredicateMux.test.ts

diff --git a/packages/backend-common/src/reading/UrlReaderPredicateMux.test.ts b/packages/backend-common/src/reading/UrlReaderPredicateMux.test.ts
new file mode 100644
index 0000000000..527f26a9ec
--- /dev/null
+++ b/packages/backend-common/src/reading/UrlReaderPredicateMux.test.ts
@@ -0,0 +1,87 @@
+/*
+ * Copyright 2020 The Backstage Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { getVoidLogger } from '../logging';
+import { UrlReaderPredicateMux } from './UrlReaderPredicateMux';
+
+describe('UrlReaderPredicateMux', () => {
+  it('forwards methods based on predicate', async () => {
+    const fooReader = {
+      read: jest.fn(),
+      readUrl: jest.fn(),
+      readTree: jest.fn(),
+      search: jest.fn(),
+    };
+    const barReader = {
+      read: jest.fn(),
+      readUrl: jest.fn(),
+      readTree: jest.fn(),
+      search: jest.fn(),
+    };
+
+    const mux = new UrlReaderPredicateMux(getVoidLogger());
+    mux.register({
+      predicate: url => url.hostname === 'foo',
+      reader: fooReader,
+    });
+    mux.register({
+      predicate: url => url.hostname === 'bar',
+      reader: barReader,
+    });
+
+    await mux.read('http://foo/1');
+    expect(fooReader.read).toHaveBeenCalledWith('http://foo/1');
+    await mux.readUrl('http://foo/2');
+    expect(fooReader.readUrl).toHaveBeenCalledWith('http://foo/2', undefined);
+    await mux.readTree('http://foo/3');
+    expect(fooReader.readTree).toHaveBeenCalledWith('http://foo/3', undefined);
+    await mux.search('http://foo/4');
+    expect(fooReader.search).toHaveBeenCalledWith('http://foo/4', undefined);
+
+    await mux.read('http://bar/1');
+    expect(barReader.read).toHaveBeenCalledWith('http://bar/1');
+    await mux.readUrl('http://bar/2');
+    expect(barReader.readUrl).toHaveBeenCalledWith('http://bar/2', undefined);
+    await mux.readTree('http://bar/3');
+    expect(barReader.readTree).toHaveBeenCalledWith('http://bar/3', undefined);
+    await mux.search('http://bar/4');
+    expect(barReader.search).toHaveBeenCalledWith('http://bar/4', undefined);
+  });
+
+  it('throws an error if no predicate matches', async () => {
+    const mux = new UrlReaderPredicateMux(getVoidLogger());
+
+    await expect(mux.readUrl('http://foo/1')).rejects.toThrowError(
+      /^Reading from 'http:\/\/foo\/1' is not allowed. You may/,
+    );
+
+    mux.register({
+      predicate: url => url.hostname === 'foo',
+      reader: {
+        read: jest.fn(),
+        readUrl: jest.fn(),
+        readTree: jest.fn(),
+        search: jest.fn(),
+      },
+    });
+
+    await expect(mux.readUrl('http://foo/1')).resolves.toBeUndefined();
+
+    await expect(mux.readUrl('http://bar/1')).rejects.toThrowError(
+      /^Reading from 'http:\/\/bar\/1' is not allowed. You may/,
+    );
+  });
+});