Skip to content

Commit

Permalink
Merge pull request #671 from nategbrown9/add-assessment-phase
Browse files Browse the repository at this point in the history
Create assessment sets phase to DRAFT if null
  • Loading branch information
DwayneJengSage authored Oct 11, 2023
2 parents bda95f2 + e5ef435 commit e23516f
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -412,6 +412,9 @@ private Assessment createAssessmentInternal(String appId, Assessment assessment)
assessment.setModifiedOn(timestamp);
assessment.setDeleted(false);
assessment.setOriginGuid(null);
if (assessment.getPhase() == null) {
assessment.setPhase(AssessmentPhase.DRAFT);
}
sanitizeAssessment(assessment);

String osName = assessment.getOsName();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,36 @@ public void createAssessment() {
assertEquals(config.getModifiedOn(), CREATED_ON);
assertNotNull(config.getConfig());
}

@Test
public void createAssessmentNoPhase() {
when(mockOrganizationService.getOrganizationOpt(TEST_APP_ID, TEST_OWNER_ID))
.thenReturn(Optional.of(mockOrganization));
when(mockDao.getAssessmentRevisions(any(), any(), any(), anyInt(), anyInt(), anyBoolean()))
.thenReturn(EMPTY_LIST);

Assessment assessment = AssessmentTest.createAssessment();
assessment.setGuid(null);
assessment.setDeleted(true); // can't do this, it's reset
assessment.setPhase(null);

service.createAssessment(TEST_APP_ID, assessment);

verify(mockDao).createAssessment(eq(TEST_APP_ID), eq(assessment), configCaptor.capture());

assertEquals(assessment.getGuid(), GUID);
assertEquals(assessment.getOwnerId(), TEST_OWNER_ID);
// Same timestamp on create
assertEquals(assessment.getCreatedOn(), CREATED_ON);
assertEquals(assessment.getModifiedOn(), CREATED_ON);
assertFalse(assessment.isDeleted());
assertEquals(AssessmentPhase.DRAFT, assessment.getPhase());

AssessmentConfig config = configCaptor.getValue();
assertEquals(config.getCreatedOn(), CREATED_ON);
assertEquals(config.getModifiedOn(), CREATED_ON);
assertNotNull(config.getConfig());
}

@Test
public void createAssessmentAdjustsOsNameAlias() {
Expand Down

0 comments on commit e23516f

Please sign in to comment.