Prefetch completion items and store them to cache

1 files changed, 74 insertions, 0 deletions

diff --git a/test/background/completion/impl/PrefetchAndCache.test.ts b/test/background/completion/impl/PrefetchAndCache.test.ts
new file mode 100644
index 0000000..23e3879
--- /dev/null
+++ b/test/background/completion/impl/PrefetchAndCache.test.ts
@@ -0,0 +1,74 @@
+import PrefetchAndCache, {shortKey} from "../../../../src/background/completion/impl/PrefetchAndCache";
+import { expect } from 'chai';
+
+class MockRepository {
+  public history: string[] = [];
+
+  constructor(
+    private items: string[],
+  ) {
+  }
+
+  get(query: string): Promise<string[]> {
+    this.history.push(query);
+    if (query.length === 0) {
+      return Promise.resolve(this.items);
+    } else {
+      return Promise.resolve(this.items.filter(item => item.includes(query)));
+    }
+  }
+}
+const filter = (items: string[], query: string) => query.length === 0 ? items : items.filter(item => item.includes(query));
+
+describe('shortKey', () => {
+  it('returns query excluding the last word', () => {
+      const query = "hello\t world    good bye";
+      const shorten = shortKey(query);
+      expect(shorten).to.equal("hello world good")
+  });
+
+  it('returns half-length of the query', () => {
+    const query = "the-query-with-super-long-word";
+    const shorten = shortKey(query);
+    expect(shorten).to.equal("the-query-with-")
+  });
+
+  it('returns shorten URL', () => {
+    let query = "https://example.com/path/to/resource?q=hello";
+    let shorten = shortKey(query);
+    expect(shorten).to.equal("https://example.com/path/to/");
+
+    query = "https://example.com/path/to/resource/#id1";
+    shorten = shortKey(query);
+    expect(shorten).to.equal("https://example.com/path/to/");
+
+    query = "https://www.google.c";
+    shorten = shortKey(query);
+    expect(shorten).to.equal("https://ww");
+  })
+});
+
+describe('PrefetchAndCache', () => {
+  describe('get', () => {
+    it('returns cached request', async() => {
+      const repo = new MockRepository(["apple", "apple pie", "apple juice", "banana", "banana pudding", "cherry"]);
+      const sut = new PrefetchAndCache(repo.get.bind(repo), filter);
+
+      expect(await sut.get("apple pie")).deep.equal(["apple pie"]);
+      expect(await sut.get("apple ")).deep.equal(["apple", "apple pie", "apple juice"]);
+      expect(await sut.get("apple")).deep.equal(["apple", "apple pie", "apple juice"]);
+      expect(await sut.get("appl")).deep.equal(["apple", "apple pie", "apple juice"]);
+      expect(repo.history).to.deep.equal(["apple", 'ap']);
+
+      expect(await sut.get("banana")).deep.equal(["banana", "banana pudding"]);
+      expect(repo.history).to.deep.equal(["apple", "ap", "ban"]);
+      expect(await sut.get("banana p")).deep.equal(["banana pudding"]);
+      expect(repo.history).to.deep.equal(["apple", "ap", "ban", "banana"]);
+      expect(await sut.get("ba")).deep.equal(["banana", "banana pudding"]);
+      expect(repo.history).to.deep.equal(["apple", "ap", "ban", "banana", "b"]);
+
+      expect(await sut.get("")).to.have.lengthOf(6);
+      expect(repo.history).to.deep.equal(["apple", "ap", "ban", "banana", "b", ""]);
+    });
+  });
+});
\ No newline at end of file