diff --git a/test/java/foundation/pEp/jniadapter/test/utils/model/TestIdentity.java b/test/java/foundation/pEp/jniadapter/test/utils/model/TestIdentity.java new file mode 100644 index 0000000..7770c4f --- /dev/null +++ b/test/java/foundation/pEp/jniadapter/test/utils/model/TestIdentity.java @@ -0,0 +1,61 @@ +package foundation.pEp.jniadapter.test.utils.model; + +import foundation.pEp.jniadapter.Identity; + +import java.util.ArrayList; +import java.util.List; +import java.util.stream.Collectors; + +public class TestIdentity { + private Role role = null; + public Identity pEpIdent = null; + private List keys = new ArrayList<>(); + private TestKeyPair defaultKey = null; + private TestKeyPair defaultKeyPP = null; + + public TestIdentity(Role role) { + this.role = role; + pEpIdent = new Identity(); + pEpIdent.username = role.toString(); + pEpIdent.address = role + "@peptest.org"; + } + + public Role getRole() { + return role; + } + + public void addKey(TestKeyPair kp, boolean isDefault) { + keys.add(kp); + if (isDefault) { + if (kp.getType() == KeyType.NORMAL) { + defaultKey = kp; + } else { + defaultKeyPP = kp; + } + } + } + + public TestKeyPair getDefaultKey(boolean passphrase) { + if (!passphrase) { + return defaultKey; + } else { + return defaultKeyPP; + } + } + + public List getAllKeys() { + return keys; + } + + public List getNormalKeys() { + return keys.stream().filter(i -> { + return i.getType().equals(KeyType.NORMAL); + }).collect(Collectors.toList()); + } + + public List getPassphraseKeys() { + return keys.stream().filter(i -> { + return i.getType().equals(KeyType.PASSPHRASE); + }).collect(Collectors.toList()); + } +}