diff --git a/e2e/rekey/trigger_test.go b/e2e/rekey/trigger_test.go index 83cc5455e38db012442d4ad69e54f476d0d68e1f..a0feb8673988920a546ab7a31fca6c922c0658e4 100644 --- a/e2e/rekey/trigger_test.go +++ b/e2e/rekey/trigger_test.go @@ -67,15 +67,15 @@ func TestHandleTrigger(t *testing.T) { newBobSIDHPubKey.Export(newBobSIDHPubKeyBytes[1:]) // Maintain an ID for bob - bobID := id.NewIdFromBytes([]byte("test"), t) - myID := id.NewIdFromString("zezima", id.User, t) + bobID = id.NewIdFromBytes([]byte("test"), t) + aliceID = id.NewIdFromString("zezima", id.User, t) kv := versioned.NewKV(ekv.Memstore{}) - err := ratchet.New(kv, myID, alicePrivKey, grp) + err := ratchet.New(kv, aliceID, alicePrivKey, grp) if err != nil { t.Errorf("Failed to create ratchet: %+v", err) } - r, err := ratchet.Load(kv, myID, grp, mockCyHandler{}, mockServiceHandler{}, rng) + r, err = ratchet.Load(kv, aliceID, grp, mockCyHandler{}, mockServiceHandler{}, rng) if err != nil { t.Fatalf("Failed to load ratchet: %+v", err) } @@ -83,13 +83,13 @@ func TestHandleTrigger(t *testing.T) { // Add bob as a partner sendParams := session2.GetDefaultParams() receiveParams := session2.GetDefaultParams() - _, err = r.AddPartner(myID, bobID, bobPubKey, + _, err = r.AddPartner(aliceID, bobID, bobPubKey, alicePrivKey, bobSIDHPubKey, aliceSIDHPrivKey, sendParams, receiveParams) if err != nil { t.Errorf("Failed to add partner to ratchet: %+v", err) } - _, err = r.AddPartner(bobID, myID, alicePubKey, bobPrivKey, + _, err = r.AddPartner(bobID, aliceID, alicePubKey, bobPrivKey, aliceSIDHPubKey, bobSIDHPrivKey, sendParams, receiveParams) if err != nil { @@ -125,7 +125,7 @@ func TestHandleTrigger(t *testing.T) { // get Alice's manager for reception from Bob - receivedManager, err := r.GetPartner(bobID, myID) + receivedManager, err := r.GetPartner(bobID, aliceID) if err != nil { t.Errorf("Failed to get bob's manager: %v", err) } diff --git a/e2e/rekey/utils_test.go b/e2e/rekey/utils_test.go index 5707715f9c6dd535cd965bf42faab986a608bd76..4b27cef48f7084e67af55ccbd5d8d735bade8c51 100644 --- a/e2e/rekey/utils_test.go +++ b/e2e/rekey/utils_test.go @@ -16,6 +16,7 @@ import ( "gitlab.com/elixxir/client/network" "gitlab.com/elixxir/client/network/gateway" "gitlab.com/elixxir/client/network/historical" + "gitlab.com/elixxir/client/network/identity" "gitlab.com/elixxir/client/network/message" "gitlab.com/elixxir/client/stoppable" util "gitlab.com/elixxir/client/storage/utility" @@ -222,6 +223,11 @@ func (m mockServiceHandler) DeleteService(clientID *id.ID, toDelete message.Serv type mockNetManager struct{} +func (m *mockNetManager) GetIdentity(get *id.ID) (identity.TrackedID, error) { + //TODO implement me + panic("implement me") +} + func (m *mockNetManager) Follow(report network.ClientErrorReport) (stoppable.Stoppable, error) { return nil, nil } diff --git a/network/identity/tracker.go b/network/identity/tracker.go index c19144301c8dcb166dc8003d7d68de200889fe14..9f3eecebdd945de387addb382d686c0deedc82c7 100644 --- a/network/identity/tracker.go +++ b/network/identity/tracker.go @@ -50,6 +50,7 @@ type Tracker interface { AddIdentity(id *id.ID, validUntil time.Time, persistent bool) RemoveIdentity(id *id.ID) GetEphemeralIdentity(rng io.Reader, addressSize uint8) (receptionID.IdentityUse, error) + GetIdentity(get *id.ID) (TrackedID, error) } type manager struct {