Fix types in tests

1 files changed, 5 insertions, 4 deletions

diff --git a/test/background/repositories/Mark.test.ts b/test/background/repositories/Mark.test.ts
index b1bbcf1..5cee5b6 100644
--- a/test/background/repositories/Mark.test.ts
+++ b/test/background/repositories/Mark.test.ts
@@ -1,4 +1,5 @@
-import MarkRepository from "background/repositories/MarkRepository";
+import MarkRepository from "../../../src/background/repositories/MarkRepository";
+import { expect } from "chai";
 
 describe("background/repositories/mark", () => {
   let repository: MarkRepository;
@@ -10,15 +11,15 @@ describe("background/repositories/mark", () => {
   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.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;
   });
 });