Skip to content
Snippets Groups Projects
Commit b5db535e authored by Richard T. Carback III's avatar Richard T. Carback III
Browse files

Fix ekv Memstore initializations

parent 4f554097
No related branches found
No related tags found
3 merge requests!510Release,!226WIP: Api2.0,!207WIP: Client Restructure
...@@ -16,7 +16,7 @@ import ( ...@@ -16,7 +16,7 @@ import (
// timestamp := time.Date(2009, 11, 17, 20, 34, 58, 651387237, time.UTC) // timestamp := time.Date(2009, 11, 17, 20, 34, 58, 651387237, time.UTC)
// idu, _ := generateFakeIdentity(rng, 15, timestamp) // idu, _ := generateFakeIdentity(rng, 15, timestamp)
// id := idu.Identity // id := idu.Identity
// kv := versioned.NewKV(make(ekv.Memstore)) // kv := versioned.NewKV(ekv.MakeMemstore())
// //
// id.End = time.Time{} // id.End = time.Time{}
// id.ExtraChecks = 0 // id.ExtraChecks = 0
......
...@@ -100,7 +100,7 @@ func Test_FileTransfer_Smoke(t *testing.T) { ...@@ -100,7 +100,7 @@ func Test_FileTransfer_Smoke(t *testing.T) {
tid, fileName, fileType, sender, size, preview} tid, fileName, fileType, sender, size, preview}
} }
myID1 := id.NewIdFromString("myID1", id.User, t) myID1 := id.NewIdFromString("myID1", id.User, t)
kv1 := versioned.NewKV(make(ekv.Memstore)) kv1 := versioned.NewKV(ekv.MakeMemstore())
endE2eChan1 := make(chan receive.Message, 3) endE2eChan1 := make(chan receive.Message, 3)
e2e1 := newMockE2e(myID1, e2eHandler) e2e1 := newMockE2e(myID1, e2eHandler)
e2e1.RegisterListener( e2e1.RegisterListener(
...@@ -125,7 +125,7 @@ func Test_FileTransfer_Smoke(t *testing.T) { ...@@ -125,7 +125,7 @@ func Test_FileTransfer_Smoke(t *testing.T) {
tid, fileName, fileType, sender, size, preview} tid, fileName, fileType, sender, size, preview}
} }
myID2 := id.NewIdFromString("myID2", id.User, t) myID2 := id.NewIdFromString("myID2", id.User, t)
kv2 := versioned.NewKV(make(ekv.Memstore)) kv2 := versioned.NewKV(ekv.MakeMemstore())
endE2eChan2 := make(chan receive.Message, 3) endE2eChan2 := make(chan receive.Message, 3)
e2e2 := newMockE2e(myID1, e2eHandler) e2e2 := newMockE2e(myID1, e2eHandler)
e2e2.RegisterListener( e2e2.RegisterListener(
......
...@@ -21,7 +21,7 @@ import ( ...@@ -21,7 +21,7 @@ import (
// Tests that NewManager returns a new Manager that matches the expected // Tests that NewManager returns a new Manager that matches the expected
// manager. // manager.
func TestNewManager(t *testing.T) { func TestNewManager(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
numFps := uint16(64) numFps := uint16(64)
fpv, _ := utility.NewStateVector(kv.Prefix(cypherManagerPrefix), fpv, _ := utility.NewStateVector(kv.Prefix(cypherManagerPrefix),
cypherManagerFpVectorKey, uint32(numFps)) cypherManagerFpVectorKey, uint32(numFps))
...@@ -134,7 +134,7 @@ func TestManager_Delete(t *testing.T) { ...@@ -134,7 +134,7 @@ func TestManager_Delete(t *testing.T) {
// Tests that a transfer key saved via saveKey can be loaded via loadKey. // Tests that a transfer key saved via saveKey can be loaded via loadKey.
func Test_saveKey_loadKey(t *testing.T) { func Test_saveKey_loadKey(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
key := &ftCrypto.TransferKey{42} key := &ftCrypto.TransferKey{42}
err := saveKey(key, kv) err := saveKey(key, kv)
...@@ -160,7 +160,7 @@ func newTestManager(numFps uint16, t *testing.T) (*Manager, *versioned.KV) { ...@@ -160,7 +160,7 @@ func newTestManager(numFps uint16, t *testing.T) (*Manager, *versioned.KV) {
t.Errorf("Failed to generate transfer key: %+v", err) t.Errorf("Failed to generate transfer key: %+v", err)
} }
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
m, err := NewManager(&key, numFps, kv) m, err := NewManager(&key, numFps, kv)
if err != nil { if err != nil {
t.Errorf("Failed to make new Manager: %+v", err) t.Errorf("Failed to make new Manager: %+v", err)
......
...@@ -23,7 +23,7 @@ import ( ...@@ -23,7 +23,7 @@ import (
// Tests that newReceivedTransfer returns a new ReceivedTransfer with the // Tests that newReceivedTransfer returns a new ReceivedTransfer with the
// expected values. // expected values.
func Test_newReceivedTransfer(t *testing.T) { func Test_newReceivedTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG()) tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG())
numFps := uint16(24) numFps := uint16(24)
...@@ -320,7 +320,7 @@ func TestReceivedTransfer_save(t *testing.T) { ...@@ -320,7 +320,7 @@ func TestReceivedTransfer_save(t *testing.T) {
// newTestReceivedTransfer creates a new ReceivedTransfer for testing. // newTestReceivedTransfer creates a new ReceivedTransfer for testing.
func newTestReceivedTransfer(numParts uint16, t *testing.T) ( func newTestReceivedTransfer(numParts uint16, t *testing.T) (
*ReceivedTransfer, *ftCrypto.TransferKey, uint16, *versioned.KV) { *ReceivedTransfer, *ftCrypto.TransferKey, uint16, *versioned.KV) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG()) tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG())
transferMAC := []byte("I am a transfer MAC") transferMAC := []byte("I am a transfer MAC")
...@@ -382,7 +382,7 @@ func TestReceivedTransfer_marshal_unmarshalReceivedTransfer(t *testing.T) { ...@@ -382,7 +382,7 @@ func TestReceivedTransfer_marshal_unmarshalReceivedTransfer(t *testing.T) {
// Tests that the part saved to storage via savePart can be loaded. // Tests that the part saved to storage via savePart can be loaded.
func Test_savePart_loadPart(t *testing.T) { func Test_savePart_loadPart(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
part := []byte("I am a part.") part := []byte("I am a part.")
partNum := 18 partNum := 18
......
...@@ -23,7 +23,7 @@ import ( ...@@ -23,7 +23,7 @@ import (
// Tests that NewOrLoadReceived returns a new Received when none exist in // Tests that NewOrLoadReceived returns a new Received when none exist in
// storage and that the list of incomplete transfers is nil. // storage and that the list of incomplete transfers is nil.
func TestNewOrLoadReceived_New(t *testing.T) { func TestNewOrLoadReceived_New(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
expected := &Received{ expected := &Received{
transfers: make(map[ftCrypto.TransferID]*ReceivedTransfer), transfers: make(map[ftCrypto.TransferID]*ReceivedTransfer),
kv: kv.Prefix(receivedTransfersStorePrefix), kv: kv.Prefix(receivedTransfersStorePrefix),
...@@ -48,7 +48,7 @@ func TestNewOrLoadReceived_New(t *testing.T) { ...@@ -48,7 +48,7 @@ func TestNewOrLoadReceived_New(t *testing.T) {
// Tests that NewOrLoadReceived returns a loaded Received when one exist in // Tests that NewOrLoadReceived returns a loaded Received when one exist in
// storage and that the list of incomplete transfers is correct. // storage and that the list of incomplete transfers is correct.
func TestNewOrLoadReceived_Load(t *testing.T) { func TestNewOrLoadReceived_Load(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
r, _, err := NewOrLoadReceived(kv) r, _, err := NewOrLoadReceived(kv)
if err != nil { if err != nil {
t.Errorf("Failed to make new Received: %+v", err) t.Errorf("Failed to make new Received: %+v", err)
...@@ -101,7 +101,7 @@ func TestNewOrLoadReceived_Load(t *testing.T) { ...@@ -101,7 +101,7 @@ func TestNewOrLoadReceived_Load(t *testing.T) {
// Tests that Received.AddTransfer makes a new transfer and adds it to the list. // Tests that Received.AddTransfer makes a new transfer and adds it to the list.
func TestReceived_AddTransfer(t *testing.T) { func TestReceived_AddTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
r, _, _ := NewOrLoadReceived(kv) r, _, _ := NewOrLoadReceived(kv)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
...@@ -137,7 +137,7 @@ func TestReceived_AddTransfer_TransferAlreadyExists(t *testing.T) { ...@@ -137,7 +137,7 @@ func TestReceived_AddTransfer_TransferAlreadyExists(t *testing.T) {
// Tests that Received.GetTransfer returns the expected transfer. // Tests that Received.GetTransfer returns the expected transfer.
func TestReceived_GetTransfer(t *testing.T) { func TestReceived_GetTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
r, _, _ := NewOrLoadReceived(kv) r, _, _ := NewOrLoadReceived(kv)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
...@@ -163,7 +163,7 @@ func TestReceived_GetTransfer(t *testing.T) { ...@@ -163,7 +163,7 @@ func TestReceived_GetTransfer(t *testing.T) {
// Tests that Sent.RemoveTransfer removes the transfer from the list. // Tests that Sent.RemoveTransfer removes the transfer from the list.
func TestReceived_RemoveTransfer(t *testing.T) { func TestReceived_RemoveTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
r, _, _ := NewOrLoadReceived(kv) r, _, _ := NewOrLoadReceived(kv)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
...@@ -201,7 +201,7 @@ func TestReceived_RemoveTransfer(t *testing.T) { ...@@ -201,7 +201,7 @@ func TestReceived_RemoveTransfer(t *testing.T) {
// Tests that Received.save saves the transfer ID list to storage by trying to // Tests that Received.save saves the transfer ID list to storage by trying to
// load it after a save. // load it after a save.
func TestReceived_save(t *testing.T) { func TestReceived_save(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
r, _, _ := NewOrLoadReceived(kv) r, _, _ := NewOrLoadReceived(kv)
r.transfers = map[ftCrypto.TransferID]*ReceivedTransfer{ r.transfers = map[ftCrypto.TransferID]*ReceivedTransfer{
ftCrypto.TransferID{0}: nil, ftCrypto.TransferID{1}: nil, ftCrypto.TransferID{0}: nil, ftCrypto.TransferID{1}: nil,
......
...@@ -27,7 +27,7 @@ import ( ...@@ -27,7 +27,7 @@ import (
// Tests that newSentTransfer returns a new SentTransfer with the expected // Tests that newSentTransfer returns a new SentTransfer with the expected
// values. // values.
func Test_newSentTransfer(t *testing.T) { func Test_newSentTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG()) tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG())
numFps := uint16(24) numFps := uint16(24)
...@@ -439,7 +439,7 @@ const numPrimeBytes = 512 ...@@ -439,7 +439,7 @@ const numPrimeBytes = 512
// newTestSentTransfer creates a new SentTransfer for testing. // newTestSentTransfer creates a new SentTransfer for testing.
func newTestSentTransfer(numParts uint16, t *testing.T) ( func newTestSentTransfer(numParts uint16, t *testing.T) (
*SentTransfer, [][]byte, *ftCrypto.TransferKey, uint16, *versioned.KV) { *SentTransfer, [][]byte, *ftCrypto.TransferKey, uint16, *versioned.KV) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
recipient := id.NewIdFromString("recipient", id.User, t) recipient := id.NewIdFromString("recipient", id.User, t)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG()) tid, _ := ftCrypto.NewTransferID(csprng.NewSystemRNG())
......
...@@ -24,7 +24,7 @@ import ( ...@@ -24,7 +24,7 @@ import (
// Tests that NewOrLoadSent returns a new Sent when none exist in storage and // Tests that NewOrLoadSent returns a new Sent when none exist in storage and
// that the list of unsent parts is nil. // that the list of unsent parts is nil.
func TestNewOrLoadSent_New(t *testing.T) { func TestNewOrLoadSent_New(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
expected := &Sent{ expected := &Sent{
transfers: make(map[ftCrypto.TransferID]*SentTransfer), transfers: make(map[ftCrypto.TransferID]*SentTransfer),
kv: kv.Prefix(sentTransfersStorePrefix), kv: kv.Prefix(sentTransfersStorePrefix),
...@@ -49,7 +49,7 @@ func TestNewOrLoadSent_New(t *testing.T) { ...@@ -49,7 +49,7 @@ func TestNewOrLoadSent_New(t *testing.T) {
// Tests that NewOrLoadSent returns a loaded Sent when one exist in storage and // Tests that NewOrLoadSent returns a loaded Sent when one exist in storage and
// that the list of unsent parts is correct. // that the list of unsent parts is correct.
func TestNewOrLoadSent_Load(t *testing.T) { func TestNewOrLoadSent_Load(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
s, _, err := NewOrLoadSent(kv) s, _, err := NewOrLoadSent(kv)
if err != nil { if err != nil {
t.Errorf("Failed to make new Sent: %+v", err) t.Errorf("Failed to make new Sent: %+v", err)
...@@ -118,7 +118,7 @@ func TestNewOrLoadSent_Load(t *testing.T) { ...@@ -118,7 +118,7 @@ func TestNewOrLoadSent_Load(t *testing.T) {
// Tests that Sent.AddTransfer makes a new transfer and adds it to the list. // Tests that Sent.AddTransfer makes a new transfer and adds it to the list.
func TestSent_AddTransfer(t *testing.T) { func TestSent_AddTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
s, _, _ := NewOrLoadSent(kv) s, _, _ := NewOrLoadSent(kv)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
...@@ -154,7 +154,7 @@ func TestSent_AddTransfer_TransferAlreadyExists(t *testing.T) { ...@@ -154,7 +154,7 @@ func TestSent_AddTransfer_TransferAlreadyExists(t *testing.T) {
// Tests that Sent.GetTransfer returns the expected transfer. // Tests that Sent.GetTransfer returns the expected transfer.
func TestSent_GetTransfer(t *testing.T) { func TestSent_GetTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
s, _, _ := NewOrLoadSent(kv) s, _, _ := NewOrLoadSent(kv)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
...@@ -180,7 +180,7 @@ func TestSent_GetTransfer(t *testing.T) { ...@@ -180,7 +180,7 @@ func TestSent_GetTransfer(t *testing.T) {
// Tests that Sent.RemoveTransfer removes the transfer from the list. // Tests that Sent.RemoveTransfer removes the transfer from the list.
func TestSent_RemoveTransfer(t *testing.T) { func TestSent_RemoveTransfer(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
s, _, _ := NewOrLoadSent(kv) s, _, _ := NewOrLoadSent(kv)
key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG()) key, _ := ftCrypto.NewTransferKey(csprng.NewSystemRNG())
...@@ -218,7 +218,7 @@ func TestSent_RemoveTransfer(t *testing.T) { ...@@ -218,7 +218,7 @@ func TestSent_RemoveTransfer(t *testing.T) {
// Tests that Sent.save saves the transfer ID list to storage by trying to load // Tests that Sent.save saves the transfer ID list to storage by trying to load
// it after a save. // it after a save.
func TestSent_save(t *testing.T) { func TestSent_save(t *testing.T) {
kv := versioned.NewKV(make(ekv.Memstore)) kv := versioned.NewKV(ekv.MakeMemstore())
s, _, _ := NewOrLoadSent(kv) s, _, _ := NewOrLoadSent(kv)
s.transfers = map[ftCrypto.TransferID]*SentTransfer{ s.transfers = map[ftCrypto.TransferID]*SentTransfer{
ftCrypto.TransferID{0}: nil, ftCrypto.TransferID{1}: nil, ftCrypto.TransferID{0}: nil, ftCrypto.TransferID{1}: nil,
......
...@@ -12,7 +12,7 @@ package utility ...@@ -12,7 +12,7 @@ package utility
func TestNewKnownRounds(t *testing.T) { func TestNewKnownRounds(t *testing.T) {
// Set up expected value // Set up expected value
size := 10 size := 10
rootKv := versioned.NewKV(make(ekv.Memstore)) rootKv := versioned.NewKV(ekv.MakeMemstore())
expectedKR := &KnownRounds{ expectedKR := &KnownRounds{
rounds: knownRounds.NewKnownRound(size), rounds: knownRounds.NewKnownRound(size),
kv: rootKv.Prefix(knownRoundsPrefix), kv: rootKv.Prefix(knownRoundsPrefix),
...@@ -37,7 +37,7 @@ func TestNewKnownRounds(t *testing.T) { ...@@ -37,7 +37,7 @@ func TestNewKnownRounds(t *testing.T) {
func TestLoadKnownRounds(t *testing.T) { func TestLoadKnownRounds(t *testing.T) {
// Set up expected value // Set up expected value
size := 10 size := 10
rootKv := versioned.NewKV(make(ekv.Memstore)) rootKv := versioned.NewKV(ekv.MakeMemstore())
expectedKR := &KnownRounds{ expectedKR := &KnownRounds{
rounds: knownRounds.NewKnownRound(size), rounds: knownRounds.NewKnownRound(size),
kv: rootKv.Prefix(knownRoundsPrefix), kv: rootKv.Prefix(knownRoundsPrefix),
...@@ -76,7 +76,7 @@ func TestKnownRounds_save(t *testing.T) { ...@@ -76,7 +76,7 @@ func TestKnownRounds_save(t *testing.T) {
size := 10 size := 10
expectedKR := &KnownRounds{ expectedKR := &KnownRounds{
rounds: knownRounds.NewKnownRound(size), rounds: knownRounds.NewKnownRound(size),
kv: versioned.NewKV(make(ekv.Memstore)), kv: versioned.NewKV(ekv.MakeMemstore()),
key: "testKey", key: "testKey",
} }
for i := 0; i < (size * 64); i++ { for i := 0; i < (size * 64); i++ {
...@@ -116,7 +116,7 @@ func TestKnownRounds_save(t *testing.T) { ...@@ -116,7 +116,7 @@ func TestKnownRounds_save(t *testing.T) {
// size := 10 // size := 10
// expectedKR := &KnownRounds{ // expectedKR := &KnownRounds{
// rounds: knownRounds.NewKnownRound(size), // rounds: knownRounds.NewKnownRound(size),
// kv: versioned.NewKV(make(ekv.Memstore)), // kv: versioned.NewKV(ekv.MakeMemstore()),
// key: "testKey", // key: "testKey",
// } // }
// for i := 0; i < (size * 64); i++ { // for i := 0; i < (size * 64); i++ {
...@@ -148,7 +148,7 @@ func TestKnownRounds_save(t *testing.T) { ...@@ -148,7 +148,7 @@ func TestKnownRounds_save(t *testing.T) {
func TestKnownRounds_Smoke(t *testing.T) { func TestKnownRounds_Smoke(t *testing.T) {
k := knownRounds.NewKnownRound(10) k := knownRounds.NewKnownRound(10)
kr, err := NewKnownRounds(versioned.NewKV(make(ekv.Memstore)), "testKey", k) kr, err := NewKnownRounds(versioned.NewKV(ekv.MakeMemstore()), "testKey", k)
if err != nil { if err != nil {
t.Fatalf("Failed to create new KnownRounds: %v", err) t.Fatalf("Failed to create new KnownRounds: %v", err)
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment