fix: 404 error returned for unknown trees and countries

Greenstand · Jan 24, 2022 · c40f0e2 · c40f0e2
1 parent 9492aee
commit c40f0e2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/__tests__/e2e/trees.spec.ts b/__tests__/e2e/trees.spec.ts
@@ -13,6 +13,11 @@ describe("trees", () => {
     });
   })
 
+  it("Unknown tree", async () => {
+    const response = await supertest(app).get("/trees/1");
+    expect(response.status).toBe(404);
+  })
+
   it("trees?limit=1&offset=0", async () => {
     const response = await supertest(app).get("/trees?limit=1&offset=0");
     expect(response.status).toBe(200);

diff --git a/server/utils/HttpError.ts b/server/utils/HttpError.ts
@@ -12,6 +12,7 @@ export default class HttpError extends Error {
 
   constructor(code: number, message: string, toRollback: boolean = true) {
     super(message)
+    Object.setPrototypeOf(this, new.target.prototype); // restore prototype chain
     this.code = code
     // set rollback flag, so the transaction of db would rollback when catch this error
     // set default to true