mstoeck3/autopsy-flatpak
1
0
Fork 0
mirror of https://github.com/overcuriousity/autopsy-flatpak.git synced 2025-07-17 18:17:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Finished reference set tests

Browse Source
This commit is contained in:
Ann Priestman 2018-01-31 08:48:04 -05:00
parent e502fedd9b
commit 23c5e5be62
2 changed files with 284 additions and 208 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Core/src/org/sleuthkit/autopsy/centralrepository/datamodel/AbstractSqlEamDb.java
View File

@ -1541,6 +1541,9 @@ public abstract class AbstractSqlEamDb implements EamDb {
public EamOrganization getReferenceSetOrganization(int referenceSetID) throws EamDbException { public EamOrganization getReferenceSetOrganization(int referenceSetID) throws EamDbException {
EamGlobalSet globalSet = getReferenceSetByID(referenceSetID); EamGlobalSet globalSet = getReferenceSetByID(referenceSetID);
if(globalSet == null) {
throw new EamDbException("Reference set with ID " + referenceSetID + " not found");
}
return (getOrganizationByID(globalSet.getOrgID())); return (getOrganizationByID(globalSet.getOrgID()));
} }

489
Core/test/qa-functional/src/org/sleuthkit/autopsy/centralrepository/datamodel/CentralRepoDatamodelTest.java
View File

@ -71,16 +71,7 @@ public class CentralRepoDatamodelTest extends TestCase {
assertTrue("Unable to create test directory", testDirectory.toFile().exists()); assertTrue("Unable to create test directory", testDirectory.toFile().exists());
// Save the current central repo settings // Save the current central repo settings
propertiesMap = ModuleSettings.getConfigSettings(PROPERTIES_FILE); propertiesMap = ModuleSettings.getConfigSettings(PROPERTIES_FILE);
// Set up an Autopsy case for testing
try {
Case.createAsCurrentCase(Case.CaseType.SINGLE_USER_CASE, testDirectory.toString(), new CaseDetails("CentralRepoDatamodelTestCase"));
} catch (CaseActionException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
assertTrue("Failed to create test case", testDirectory.toFile().exists());
try { try {
dbSettingsSqlite.setDbName(CR_DB_NAME); dbSettingsSqlite.setDbName(CR_DB_NAME);
@ -150,14 +141,8 @@ public class CentralRepoDatamodelTest extends TestCase {
// Close and delete the test case and central repo db // Close and delete the test case and central repo db
try { try {
EamDb.getInstance().shutdownConnections(); EamDb.getInstance().shutdownConnections();
Case.closeCurrentCase(); Case.closeCurrentCase(); // There shouldn't be a case open, but it's fine to call this anyway
// This seems to help in allowing the Autopsy case to be deleted
try{
Thread.sleep(2000);
} catch (Exception ex){
}
FileUtils.deleteDirectory(testDirectory.toFile()); FileUtils.deleteDirectory(testDirectory.toFile());
} catch (EamDbException | CaseActionException | IOException ex) { } catch (EamDbException | CaseActionException | IOException ex) {
@ -169,7 +154,40 @@ public class CentralRepoDatamodelTest extends TestCase {
} }
/** /**
* Test method for the methods related to reference sets * Test method for the methods related to reference sets (does not include instance testing)
* newReferenceSet(EamGlobalSet eamGlobalSet) tests:
* - Test creating notable reference set
* - Test creating known reference set
* - Test creating duplicate reference set
* - Test creating almost duplicate reference set
* - Test with invalid org ID
* - Test with null name
* - Test with null version
* - Test with null known status
* - Test with null file type
* referenceSetIsValid(int referenceSetID, String referenceSetName, String version) tests:
* - Test on existing reference set
* - Test on invalid reference set
* - Test with null name
* - Test with null version
* referenceSetExists(String referenceSetName, String version) tests:
* - Test on existing reference set
* - Test on invalid reference set
* - Test with null name
* - Test with null version
* getReferenceSetByID(int globalSetID) tests:
* - Test with valid ID
* - Test with invalid ID
* getAllReferenceSets(CorrelationAttribute.Type correlationType) tests:
* - Test getting all file sets
* - Test getting all email sets
* - Test with null type parameter
* deleteReferenceSet(int referenceSetID) tests:
* - Test on valid reference set ID
* - Test on invalid reference set ID
* getReferenceSetOrganization(int referenceSetID) tests:
* - Test on valid reference set ID
* - Test on invalid reference set ID
*/ */
public void testReferenceSets() { public void testReferenceSets() {
String set1name = "referenceSet1"; String set1name = "referenceSet1";
@ -396,13 +414,44 @@ public class CentralRepoDatamodelTest extends TestCase {
assertTrue("setToDelete wasn't found in database", EamDb.getInstance().referenceSetIsValid(setToDeleteID, setToDelete.getSetName(), setToDelete.getVersion())); assertTrue("setToDelete wasn't found in database", EamDb.getInstance().referenceSetIsValid(setToDeleteID, setToDelete.getSetName(), setToDelete.getVersion()));
int currentCount = EamDb.getInstance().getAllReferenceSets(fileType).size(); int currentCount = EamDb.getInstance().getAllReferenceSets(fileType).size();
EamDb.getInstance().deleteReferenceSet(setToDeleteID);
assertFalse("Deleted reference set was found in database", EamDb.getInstance().referenceSetIsValid(setToDeleteID, setToDelete.getSetName(), setToDelete.getVersion()));
assertTrue("Unexpected number of reference sets in database after deletion", currentCount - 1 == EamDb.getInstance().getAllReferenceSets(fileType).size());
} catch (EamDbException ex){ } catch (EamDbException ex){
Exceptions.printStackTrace(ex); Exceptions.printStackTrace(ex);
Assert.fail(ex); Assert.fail(ex);
} }
// Test deleting a non-existent reference set
// The expectation is that nothing will happen
try {
int currentCount = EamDb.getInstance().getAllReferenceSets(fileType).size();
EamDb.getInstance().deleteReferenceSet(1234);
assertTrue("Number of reference sets changed after deleting non-existent set", currentCount == EamDb.getInstance().getAllReferenceSets(fileType).size());
} catch (EamDbException ex){
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting reference set organization for valid ID with org set
try {
EamOrganization org = EamDb.getInstance().getReferenceSetOrganization(set1id);
assertTrue("getReferenceSetOrganization returned null for valid set", org != null);
assertTrue("getReferenceSetOrganization returned the incorrect organization", org.getOrgID() == org1.getOrgID());
} catch (EamDbException ex){
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting reference set organization for non-existent reference set
try {
EamOrganization org = EamDb.getInstance().getReferenceSetOrganization(4567);
Assert.fail("getReferenceSetOrganization failed to throw exception for invalid reference set ID");
} catch (EamDbException ex){
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
} }
/** /**
@ -565,201 +614,225 @@ public class CentralRepoDatamodelTest extends TestCase {
final String caseAuuid = "caseA_uuid"; final String caseAuuid = "caseA_uuid";
CorrelationCase caseA; CorrelationCase caseA;
CorrelationCase caseB; CorrelationCase caseB;
// Test creating a case with valid name and uuid
try { try {
caseA = new CorrelationCase(caseAuuid, caseAname); // Set up an Autopsy case for testing
caseA = EamDb.getInstance().newCase(caseA); try {
assertTrue("Failed to create case", caseA != null); Case.createAsCurrentCase(Case.CaseType.SINGLE_USER_CASE, testDirectory.toString(), new CaseDetails("CentralRepoDatamodelTestCase"));
} catch (EamDbException ex) { } catch (CaseActionException ex) {
Exceptions.printStackTrace(ex); Exceptions.printStackTrace(ex);
Assert.fail(ex); Assert.fail(ex);
return; }
} assertTrue("Failed to create test case", testDirectory.toFile().exists());
// Test null uuid // Test creating a case with valid name and uuid
try { try {
CorrelationCase tempCase = new CorrelationCase(null, "nullUuidCase"); caseA = new CorrelationCase(caseAuuid, caseAname);
tempCase = EamDb.getInstance().newCase(tempCase); caseA = EamDb.getInstance().newCase(caseA);
Assert.fail("newCase did not throw expected exception from null uuid"); assertTrue("Failed to create case", caseA != null);
} catch (EamDbException ex) { } catch (EamDbException ex) {
// This is the expected behavior Exceptions.printStackTrace(ex);
} Assert.fail(ex);
return;
// Test null name
try {
CorrelationCase tempCase = new CorrelationCase("nullCaseUuid", null);
tempCase = EamDb.getInstance().newCase(tempCase);
Assert.fail("newCase did not throw expected exception from null name");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Test creating a case with an already used UUID
// This should just return the existing case object. Check that the total
// number of cases does not increase.
try {
int nCases = EamDb.getInstance().getCases().size();
CorrelationCase tempCase = new CorrelationCase(caseAuuid, "newCaseWithSameUUID");
tempCase = EamDb.getInstance().newCase(tempCase);
assertTrue("newCase returned null for existing UUID", tempCase != null);
assertTrue("newCase created a new case for an already existing UUID", nCases == EamDb.getInstance().getCases().size());
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test creating a case from an Autopsy case
// The case may already be in the database - the result is the same either way
try {
caseB = EamDb.getInstance().newCase(Case.getCurrentCase());
assertTrue("Failed to create correlation case from Autopsy case", caseB != null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
return;
}
// Test null Autopsy case
try {
Case nullCase = null;
CorrelationCase tempCase = EamDb.getInstance().newCase(nullCase);
Assert.fail("newCase did not throw expected exception from null case");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Test update case
// Will update the fields of an existing case object, save it, and then
// pull a new copy out of the database
try {
assertTrue(caseA != null);
String caseNumber = "12-34-56";
String creationDate = "01/12/2018";
String displayName = "Test Case";
String examinerEmail = "john@sample.com";
String examinerName = "John Doe";
String examinerPhone = "123-555-4567";
String notes = "Notes";
caseA.setCaseNumber(caseNumber);
caseA.setCreationDate(creationDate);
caseA.setDisplayName(displayName);
caseA.setExaminerEmail(examinerEmail);
caseA.setExaminerName(examinerName);
caseA.setExaminerPhone(examinerPhone);
caseA.setNotes(notes);
caseA.setOrg(org1);
EamDb.getInstance().updateCase(caseA);
// Retrievex a new copy of the case from the database to check that the
// fields were properly updated
CorrelationCase updatedCase = EamDb.getInstance().getCaseByUUID(caseA.getCaseUUID());
assertTrue("updateCase failed to update case number", caseNumber.equals(updatedCase.getCaseNumber()));
assertTrue("updateCase failed to update creation date", creationDate.equals(updatedCase.getCreationDate()));
assertTrue("updateCase failed to update display name", displayName.equals(updatedCase.getDisplayName()));
assertTrue("updateCase failed to update examiner email", examinerEmail.equals(updatedCase.getExaminerEmail()));
assertTrue("updateCase failed to update examiner name", examinerName.equals(updatedCase.getExaminerName()));
assertTrue("updateCase failed to update examiner phone number", examinerPhone.equals(updatedCase.getExaminerPhone()));
assertTrue("updateCase failed to update notes", notes.equals(updatedCase.getNotes()));
assertTrue("updateCase failed to update org (org is null)", updatedCase.getOrg() != null);
assertTrue("updateCase failed to update org (org ID is wrong)", org1.getOrgID() == updatedCase.getOrg().getOrgID());
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test update case with null case
try {
EamDb.getInstance().updateCase(null);
Assert.fail("updateCase did not throw expected exception from null case");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Test getting a case from an Autopsy case
try {
CorrelationCase tempCase = EamDb.getInstance().getCase(Case.getCurrentCase());
assertTrue("getCase returned null for current Autopsy case", tempCase != null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting a case by UUID
try {
CorrelationCase tempCase = EamDb.getInstance().getCaseByUUID(caseAuuid);
assertTrue("Failed to get case by UUID", tempCase != null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting a case with a non-existent UUID
try {
CorrelationCase tempCase = EamDb.getInstance().getCaseByUUID("badUUID");
assertTrue("getCaseByUUID returned non-null case for non-existent UUID", tempCase == null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting a case with null UUID
try {
CorrelationCase tempCase = EamDb.getInstance().getCaseByUUID(null);
assertTrue("getCaseByUUID returned non-null case for null UUID", tempCase == null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting the list of cases
// The test is to make sure the three cases we know are in the database are in the list
try {
List<CorrelationCase> caseList = EamDb.getInstance().getCases();
List<String> uuidList
= caseList.stream().map(c -> c.getCaseUUID()).collect(Collectors.toList());
assertTrue("getCases is missing data for existing cases", uuidList.contains(case1.getCaseUUID())
&& uuidList.contains(case2.getCaseUUID()) && (uuidList.contains(caseA.getCaseUUID()))
&& uuidList.contains(caseB.getCaseUUID()));
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test bulk case insert
try {
// Create a list of correlation cases. Make enough that the bulk threshold should be hit once.
List<CorrelationCase> cases = new ArrayList<>();
String bulkTestUuid = "bulkTestUUID_";
String bulkTestName = "bulkTestName_";
for (int i = 0; i < dbSettingsSqlite.getBulkThreshold() * 1.5; i++) {
String name = bulkTestUuid + String.valueOf(i);
String uuid = bulkTestName + String.valueOf(i);
cases.add(new CorrelationCase(uuid, name));
} }
// Get the current case count // Test null uuid
int nCases = EamDb.getInstance().getCases().size(); try {
CorrelationCase tempCase = new CorrelationCase(null, "nullUuidCase");
tempCase = EamDb.getInstance().newCase(tempCase);
Assert.fail("newCase did not throw expected exception from null uuid");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Insert the big list of cases // Test null name
EamDb.getInstance().bulkInsertCases(cases); try {
CorrelationCase tempCase = new CorrelationCase("nullCaseUuid", null);
tempCase = EamDb.getInstance().newCase(tempCase);
Assert.fail("newCase did not throw expected exception from null name");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Check that the case count is what is expected // Test creating a case with an already used UUID
assertTrue("bulkInsertCases did not insert the expected number of cases", nCases + cases.size() == EamDb.getInstance().getCases().size()); // This should just return the existing case object. Check that the total
} catch (EamDbException ex) { // number of cases does not increase.
Exceptions.printStackTrace(ex); try {
Assert.fail(ex); int nCases = EamDb.getInstance().getCases().size();
} CorrelationCase tempCase = new CorrelationCase(caseAuuid, "newCaseWithSameUUID");
tempCase = EamDb.getInstance().newCase(tempCase);
assertTrue("newCase returned null for existing UUID", tempCase != null);
assertTrue("newCase created a new case for an already existing UUID", nCases == EamDb.getInstance().getCases().size());
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test bulk case insert with null list // Test creating a case from an Autopsy case
try { // The case may already be in the database - the result is the same either way
EamDb.getInstance().bulkInsertCases(null); try {
Assert.fail("bulkInsertCases did not throw expected exception from null list"); caseB = EamDb.getInstance().newCase(Case.getCurrentCase());
} catch (EamDbException ex) { assertTrue("Failed to create correlation case from Autopsy case", caseB != null);
// This is the expected behavior } catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
return;
}
// Test null Autopsy case
try {
Case nullCase = null;
CorrelationCase tempCase = EamDb.getInstance().newCase(nullCase);
Assert.fail("newCase did not throw expected exception from null case");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Test update case
// Will update the fields of an existing case object, save it, and then
// pull a new copy out of the database
try {
assertTrue(caseA != null);
String caseNumber = "12-34-56";
String creationDate = "01/12/2018";
String displayName = "Test Case";
String examinerEmail = "john@sample.com";
String examinerName = "John Doe";
String examinerPhone = "123-555-4567";
String notes = "Notes";
caseA.setCaseNumber(caseNumber);
caseA.setCreationDate(creationDate);
caseA.setDisplayName(displayName);
caseA.setExaminerEmail(examinerEmail);
caseA.setExaminerName(examinerName);
caseA.setExaminerPhone(examinerPhone);
caseA.setNotes(notes);
caseA.setOrg(org1);
EamDb.getInstance().updateCase(caseA);
// Retrievex a new copy of the case from the database to check that the
// fields were properly updated
CorrelationCase updatedCase = EamDb.getInstance().getCaseByUUID(caseA.getCaseUUID());
assertTrue("updateCase failed to update case number", caseNumber.equals(updatedCase.getCaseNumber()));
assertTrue("updateCase failed to update creation date", creationDate.equals(updatedCase.getCreationDate()));
assertTrue("updateCase failed to update display name", displayName.equals(updatedCase.getDisplayName()));
assertTrue("updateCase failed to update examiner email", examinerEmail.equals(updatedCase.getExaminerEmail()));
assertTrue("updateCase failed to update examiner name", examinerName.equals(updatedCase.getExaminerName()));
assertTrue("updateCase failed to update examiner phone number", examinerPhone.equals(updatedCase.getExaminerPhone()));
assertTrue("updateCase failed to update notes", notes.equals(updatedCase.getNotes()));
assertTrue("updateCase failed to update org (org is null)", updatedCase.getOrg() != null);
assertTrue("updateCase failed to update org (org ID is wrong)", org1.getOrgID() == updatedCase.getOrg().getOrgID());
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test update case with null case
try {
EamDb.getInstance().updateCase(null);
Assert.fail("updateCase did not throw expected exception from null case");
} catch (EamDbException ex) {
// This is the expected behavior
}
// Test getting a case from an Autopsy case
try {
CorrelationCase tempCase = EamDb.getInstance().getCase(Case.getCurrentCase());
assertTrue("getCase returned null for current Autopsy case", tempCase != null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting a case by UUID
try {
CorrelationCase tempCase = EamDb.getInstance().getCaseByUUID(caseAuuid);
assertTrue("Failed to get case by UUID", tempCase != null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting a case with a non-existent UUID
try {
CorrelationCase tempCase = EamDb.getInstance().getCaseByUUID("badUUID");
assertTrue("getCaseByUUID returned non-null case for non-existent UUID", tempCase == null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting a case with null UUID
try {
CorrelationCase tempCase = EamDb.getInstance().getCaseByUUID(null);
assertTrue("getCaseByUUID returned non-null case for null UUID", tempCase == null);
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test getting the list of cases
// The test is to make sure the three cases we know are in the database are in the list
try {
List<CorrelationCase> caseList = EamDb.getInstance().getCases();
List<String> uuidList
= caseList.stream().map(c -> c.getCaseUUID()).collect(Collectors.toList());
assertTrue("getCases is missing data for existing cases", uuidList.contains(case1.getCaseUUID())
&& uuidList.contains(case2.getCaseUUID()) && (uuidList.contains(caseA.getCaseUUID()))
&& uuidList.contains(caseB.getCaseUUID()));
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test bulk case insert
try {
// Create a list of correlation cases. Make enough that the bulk threshold should be hit once.
List<CorrelationCase> cases = new ArrayList<>();
String bulkTestUuid = "bulkTestUUID_";
String bulkTestName = "bulkTestName_";
for (int i = 0; i < dbSettingsSqlite.getBulkThreshold() * 1.5; i++) {
String name = bulkTestUuid + String.valueOf(i);
String uuid = bulkTestName + String.valueOf(i);
cases.add(new CorrelationCase(uuid, name));
}
// Get the current case count
int nCases = EamDb.getInstance().getCases().size();
// Insert the big list of cases
EamDb.getInstance().bulkInsertCases(cases);
// Check that the case count is what is expected
assertTrue("bulkInsertCases did not insert the expected number of cases", nCases + cases.size() == EamDb.getInstance().getCases().size());
} catch (EamDbException ex) {
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
// Test bulk case insert with null list
try {
EamDb.getInstance().bulkInsertCases(null);
Assert.fail("bulkInsertCases did not throw expected exception from null list");
} catch (EamDbException ex) {
// This is the expected behavior
}
} finally {
try {
Case.closeCurrentCase();
// This seems to help in allowing the Autopsy case to be deleted
try{
Thread.sleep(2000);
} catch (Exception ex){
}
} catch (CaseActionException ex){
Exceptions.printStackTrace(ex);
Assert.fail(ex);
}
} }
} }