1 files changed, 9 insertions, 8 deletions
diff --git a/test/background/repositories/Mark.test.ts b/test/background/repositories/Mark.test.ts
index ed1a68e..5cee5b6 100644
--- a/test/background/repositories/Mark.test.ts
+++ b/test/background/repositories/Mark.test.ts
@@ -1,24 +1,25 @@
-import MarkRepository from 'background/repositories/MarkRepository';
+import MarkRepository from "../../../src/background/repositories/MarkRepository";
+import { expect } from "chai";
 
-describe('background/repositories/mark', () => {
+describe("background/repositories/mark", () => {
   let repository: MarkRepository;
 
   beforeEach(() => {
     repository = new MarkRepository();
   });
 
-  it('get and set', async() => {
-    const mark = { tabId: 1, url: 'http://example.com', x: 10, y: 30 };
+  it("get and set", async () => {
+    const mark = { tabId: 1, url: "http://example.com", x: 10, y: 30 };
 
-    repository.setMark('A', mark);
+    await repository.setMark("A", mark);
 
-    let got = await repository.getMark('A');
+    let got = (await repository.getMark("A"))!!;
     expect(got.tabId).to.equal(1);
-    expect(got.url).to.equal('http://example.com');
+    expect(got.url).to.equal("http://example.com");
     expect(got.x).to.equal(10);
     expect(got.y).to.equal(30);
 
-    got = await repository.getMark('B');
+    got = (await repository.getMark("B"))!!;
     expect(got).to.be.undefined;
   });
 });