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

test utils

Browse Source
This commit is contained in:
Greg DiCristofaro 2020-09-23 10:26:13 -04:00
parent cf7c808f65
commit 2e67ee6044
1 changed files with 57 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
Core/test/unit/src/org/sleuthkit/autopsy/testutils/TskMockUtils.java Normal file
View File

@ -0,0 +1,57 @@
/*
* To change this license header, choose License Headers in Project Properties.
* To change this template file, choose Tools | Templates
* and open the template in the editor.
*/
package org.sleuthkit.autopsy.testutils;
import java.util.ArrayList;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import org.sleuthkit.datamodel.BlackboardArtifact;
import org.sleuthkit.datamodel.BlackboardAttribute;
import org.sleuthkit.datamodel.DataSource;
import org.sleuthkit.datamodel.TskCoreException;
/**
*
* @author gregd
*/
public class TskMockUtils {
public static DataSource mockDataSource(long dataSourceId) {
DataSource dataSource = mock(DataSource.class);
when(dataSource.getName()).thenReturn("");
when(dataSource.getId()).thenReturn(dataSourceId);
return dataSource;
}
public static BlackboardArtifact mockArtifact(BlackboardArtifact.Type artifactType, long artifactId,
DataSource dataSource, BlackboardAttribute...attributes) throws TskCoreException {
BlackboardArtifact artifact = mock(BlackboardArtifact.class);
final Map<Integer, BlackboardAttribute> attributeTypes = Stream.of(attributes)
.collect(Collectors.toMap((attr) -> attr.getAttributeType().getTypeID(), Function.identity()));
when(artifact.getArtifactID()).thenReturn(artifactId);
when(artifact.getArtifactTypeID()).thenReturn(artifactType.getTypeID());
when(artifact.getAttribute(any())).thenAnswer((invocation) -> {
Object[] args = invocation.getArguments();
BlackboardArtifact.Type type = (BlackboardArtifact.Type) args[0];
return attributeTypes.get(type.getTypeID());
});
when(artifact.getAttributes()).thenReturn(new ArrayList<>(attributeTypes.values()));
when(artifact.getDataSource()).thenReturn(dataSource);
return artifact;
}
}