updated tests

bcgov · Sep 8, 2023 · 32bf987 · 32bf987
1 parent 1487bb5
commit 32bf987
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/api/src/repositories/survey-block-repository.test.ts b/api/src/repositories/survey-block-repository.test.ts
@@ -5,7 +5,7 @@ import sinon from 'sinon';
 import sinonChai from 'sinon-chai';
 import { ApiExecuteSQLError } from '../errors/api-error';
 import { getMockDBConnection } from '../__mocks__/db';
-import { SurveyBlock, SurveyBlockRepository } from './survey-block-repository';
+import { PostSurveyBlock, SurveyBlockRepository } from './survey-block-repository';
 
 chai.use(sinonChai);
 
@@ -82,7 +82,7 @@ describe('SurveyBlockRepository', () => {
       });
 
       const repo = new SurveyBlockRepository(dbConnection);
-      const block: SurveyBlock = { survey_block_id: 1, survey_id: 1, name: 'Updated name', description: 'block' };
+      const block: PostSurveyBlock = { survey_block_id: 1, survey_id: 1, name: 'Updated name', description: 'block' };
       const response = await repo.updateSurveyBlock(block);
       expect(response.survey_block_id).to.be.eql(1);
       expect(response.name).to.be.eql('Updated name');
@@ -98,7 +98,7 @@ describe('SurveyBlockRepository', () => {
       });
 
       const repo = new SurveyBlockRepository(dbConnection);
-      const block: SurveyBlock = { survey_block_id: null, survey_id: 1, name: 'new', description: 'block' };
+      const block: PostSurveyBlock = { survey_block_id: null, survey_id: 1, name: 'new', description: 'block' };
       try {
         await repo.updateSurveyBlock(block);
         expect.fail();
@@ -131,7 +131,7 @@ describe('SurveyBlockRepository', () => {
       });
       const repo = new SurveyBlockRepository(dbConnection);
 
-      const block: SurveyBlock = { survey_block_id: null, survey_id: 1, name: 'new', description: 'block' };
+      const block: PostSurveyBlock = { survey_block_id: null, survey_id: 1, name: 'new', description: 'block' };
       const response = await repo.insertSurveyBlock(block);
 
       expect(response.name).to.be.eql('new');
@@ -148,7 +148,12 @@ describe('SurveyBlockRepository', () => {
       });
       const repo = new SurveyBlockRepository(dbConnection);
       try {
-        const block = ({ survey_block_id: null, survey_id: 1, name: null, description: null } as any) as SurveyBlock;
+        const block = ({
+          survey_block_id: null,
+          survey_id: 1,
+          name: null,
+          description: null
+        } as any) as PostSurveyBlock;
         await repo.insertSurveyBlock(block);
         expect.fail();
       } catch (error) {

diff --git a/api/src/services/survey-block-service.test.ts b/api/src/services/survey-block-service.test.ts
@@ -4,7 +4,7 @@ import { QueryResult } from 'pg';
 import sinon from 'sinon';
 import sinonChai from 'sinon-chai';
 import { ApiExecuteSQLError } from '../errors/api-error';
-import { SurveyBlock, SurveyBlockRepository } from '../repositories/survey-block-repository';
+import { PostSurveyBlock, SurveyBlockRepository } from '../repositories/survey-block-repository';
 import { getMockDBConnection } from '../__mocks__/db';
 import { SurveyBlockService } from './survey-block-service';
 
@@ -70,7 +70,7 @@ describe('SurveyBlockService', () => {
       const insertBlock = sinon.stub(SurveyBlockRepository.prototype, 'insertSurveyBlock').resolves();
       const updateBlock = sinon.stub(SurveyBlockRepository.prototype, 'updateSurveyBlock').resolves();
 
-      const blocks: SurveyBlock[] = [
+      const blocks: PostSurveyBlock[] = [
         { survey_block_id: null, survey_id: 1, name: 'Old Block', description: 'Updated' },
         { survey_block_id: null, survey_id: 1, name: 'New Block', description: 'block' }
       ];
@@ -114,7 +114,7 @@ describe('SurveyBlockService', () => {
       const insertBlock = sinon.stub(SurveyBlockRepository.prototype, 'insertSurveyBlock').resolves();
       const updateBlock = sinon.stub(SurveyBlockRepository.prototype, 'updateSurveyBlock').resolves();
 
-      const blocks: SurveyBlock[] = [
+      const blocks: PostSurveyBlock[] = [
         { survey_block_id: 10, survey_id: 1, name: 'Old Block', description: 'Updated' },
         { survey_block_id: null, survey_id: 1, name: 'New Block', description: 'block' }
       ];