Move in memory data structures into project.

1 files changed, 23 insertions, 0 deletions

diff --git a/tests/src/test/java/org/whispersystems/libaxolotl/TestInMemoryIdentityKeyStore.java b/tests/src/test/java/org/whispersystems/libaxolotl/TestInMemoryIdentityKeyStore.java
new file mode 100644
index 00000000..10a206d3
--- /dev/null
+++ b/tests/src/test/java/org/whispersystems/libaxolotl/TestInMemoryIdentityKeyStore.java
@@ -0,0 +1,23 @@
+package org.whispersystems.libaxolotl;
+
+import org.whispersystems.libaxolotl.ecc.Curve;
+import org.whispersystems.libaxolotl.ecc.ECKeyPair;
+import org.whispersystems.libaxolotl.util.KeyHelper;
+
+public class TestInMemoryIdentityKeyStore extends org.whispersystems.libaxolotl.state.impl.InMemoryIdentityKeyStore {
+  public TestInMemoryIdentityKeyStore() {
+    super(generateIdentityKeyPair(), generateRegistrationId());
+  }
+
+  private static IdentityKeyPair generateIdentityKeyPair() {
+    ECKeyPair identityKeyPairKeys = Curve.generateKeyPair();
+
+    return new IdentityKeyPair(new IdentityKey(identityKeyPairKeys.getPublicKey()),
+                               identityKeyPairKeys.getPrivateKey());
+  }
+
+  private static int generateRegistrationId() {
+    return KeyHelper.generateRegistrationId(false);
+  }
+
+}