Signal-Android/libaxolotl/src/androidTest/java/org/whispersystems/test/groups/InMemorySenderKeyStore.java

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);
    }
  }
}