Added #304 test coverage

moleculerjs · Jan 11, 2022 · 976e831 · 976e831
1 parent cc9cc07
commit 976e831
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/packages/moleculer-db-adapter-sequelize/test/unit/index.spec.js b/packages/moleculer-db-adapter-sequelize/test/unit/index.spec.js
@@ -5,6 +5,8 @@ const { ServiceBroker } = require("moleculer");
 jest.mock("sequelize");
 
 const model = {
+	tableName: "posts",
+	primaryKeyAttribute: 'id',
 	sync: jest.fn(() => Promise.resolve()),
 	findAll: jest.fn(() => Promise.resolve()),
 	count: jest.fn(() => Promise.resolve()),
@@ -40,6 +42,7 @@ function protectReject(err) {
 const fakeModel = {
 	name: "posts",
 	define: {
+		id: { type: "number", primaryKey: true },
 		a: 5
 	},
 	options: {
@@ -163,7 +166,10 @@ describe("Test SequelizeAdapter", () => {
 				adapter.model.findAll.mockClear();
 				adapter.createCursor(null, true);
 				expect(adapter.model.count).toHaveBeenCalledTimes(1);
-				expect(adapter.model.count).toHaveBeenCalledWith();
+				expect(adapter.model.count).toHaveBeenCalledWith({
+					distinct: true,
+					col: `${model.tableName}.${model.primaryKeyAttribute}`
+				});
 			});
 
 			it("call with query", () => {
@@ -179,7 +185,11 @@ describe("Test SequelizeAdapter", () => {
 				let query = {};
 				adapter.createCursor({ query }, true);
 				expect(adapter.model.count).toHaveBeenCalledTimes(1);
-				expect(adapter.model.count).toHaveBeenCalledWith({ where: query });
+				expect(adapter.model.count).toHaveBeenCalledWith({
+					distinct: true,
+					col: `${model.tableName}.${model.primaryKeyAttribute}`,
+					where: query
+				});
 			});
 
 			it("call with sort string", () => {