瀏覽代碼

refactor: Added data module find job filter by date types tests

Owen Diffey 2 年之前
父節點
當前提交
0858a4402c
共有 1 個文件被更改,包括 42 次插入1 次删除
  1. 42 1
      backend/src/modules/DataModule.spec.ts

+ 42 - 1
backend/src/modules/DataModule.spec.ts

@@ -10,7 +10,7 @@ import LogBook from "../LogBook";
 import ModuleManager from "../ModuleManager";
 import DataModule from "./DataModule";
 
-chai.should();
+const should = chai.should();
 chai.use(sinonChai);
 chai.use(chaiAsPromised);
 
@@ -335,6 +335,47 @@ describe("Data Module", function () {
 				"restrictedName"
 			]);
 		});
+
+		describe("filter by date types", function () {
+			it("Date", async function () {
+				const [document] = testData.abc;
+				const { createdAt } = document;
+				const resultDocument = await dataModule.find(jobContext, {
+					collection: "abc",
+					filter: { createdAt },
+					limit: 1,
+					useCache: false
+				});
+				should.exist(resultDocument);
+				resultDocument.createdAt.should.deep.equal(document.createdAt);
+			});
+
+			it("String", async function () {
+				const [document] = testData.abc;
+				const { createdAt } = document;
+				const resultDocument = await dataModule.find(jobContext, {
+					collection: "abc",
+					filter: { createdAt: createdAt.toString() },
+					limit: 1,
+					useCache: false
+				});
+				should.exist(resultDocument);
+				resultDocument.createdAt.should.deep.equal(document.createdAt);
+			});
+
+			it("Number", async function () {
+				const [document] = testData.abc;
+				const { createdAt } = document;
+				const resultDocument = await dataModule.find(jobContext, {
+					collection: "abc",
+					filter: { createdAt: createdAt.getTime() },
+					limit: 1,
+					useCache: false
+				});
+				should.exist(resultDocument);
+				resultDocument.createdAt.should.deep.equal(document.createdAt);
+			});
+		});
 	});
 
 	describe("normalize projection", function () {