From 60800e155612bea797eed93c67046a23d26054cc Mon Sep 17 00:00:00 2001
From: Moxie Marlinspike <moxie@thoughtcrime.org>
Date: Mon, 24 Nov 2014 12:54:30 -0800
Subject: Break out into separate repo.

---
 .../test/groups/InMemorySenderKeyStore.java        | 33 ++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 src/androidTest/java/org/whispersystems/test/groups/InMemorySenderKeyStore.java

(limited to 'src/androidTest/java/org/whispersystems/test/groups/InMemorySenderKeyStore.java')

diff --git a/src/androidTest/java/org/whispersystems/test/groups/InMemorySenderKeyStore.java b/src/androidTest/java/org/whispersystems/test/groups/InMemorySenderKeyStore.java
new file mode 100644
index 00000000..450d802d
--- /dev/null
+++ b/src/androidTest/java/org/whispersystems/test/groups/InMemorySenderKeyStore.java
@@ -0,0 +1,33 @@
+package org.whispersystems.test.groups;
+
+import org.whispersystems.libaxolotl.groups.state.SenderKeyRecord;
+import org.whispersystems.libaxolotl.groups.state.SenderKeyStore;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+public class InMemorySenderKeyStore implements SenderKeyStore {
+
+  private final Map<String, SenderKeyRecord> store = new HashMap<>();
+
+  @Override
+  public void storeSenderKey(String senderKeyId, SenderKeyRecord record) {
+    store.put(senderKeyId, record);
+  }
+
+  @Override
+  public SenderKeyRecord loadSenderKey(String senderKeyId) {
+    try {
+      SenderKeyRecord record = store.get(senderKeyId);
+
+      if (record == null) {
+        return new SenderKeyRecord();
+      } else {
+        return new SenderKeyRecord(record.serialize());
+      }
+    } catch (IOException e) {
+      throw new AssertionError(e);
+    }
+  }
+}
-- 
cgit v1.2.3