@@ -82,7 +82,7 @@ func TestBasicSessionStore(t *testing.T) {
8282 // Ensure that we can retrieve each session by both its local pub key
8383 // and by its ID.
8484 for _ , s := range []* Session {s1 , s2 , s3 } {
85- session , err := db .GetSession (ctx , s .LocalPublicKey )
85+ session , err := db .GetSessionByLocalPub (ctx , s .LocalPublicKey )
8686 require .NoError (t , err )
8787 assertEqualSessions (t , s , session )
8888
@@ -92,7 +92,7 @@ func TestBasicSessionStore(t *testing.T) {
9292 }
9393
9494 // Fetch session 1 and assert that it currently has no remote pub key.
95- session1 , err := db .GetSession (ctx , s1 .LocalPublicKey )
95+ session1 , err := db .GetSessionByLocalPub (ctx , s1 .LocalPublicKey )
9696 require .NoError (t , err )
9797 require .Nil (t , session1 .RemotePublicKey )
9898
@@ -107,7 +107,7 @@ func TestBasicSessionStore(t *testing.T) {
107107 require .NoError (t , err )
108108
109109 // Assert that the session now does have the remote pub key.
110- session1 , err = db .GetSession (ctx , s1 .LocalPublicKey )
110+ session1 , err = db .GetSessionByLocalPub (ctx , s1 .LocalPublicKey )
111111 require .NoError (t , err )
112112 require .True (t , remotePub .IsEqual (session1 .RemotePublicKey ))
113113
@@ -116,7 +116,7 @@ func TestBasicSessionStore(t *testing.T) {
116116
117117 // Now revoke the session and assert that the state is revoked.
118118 require .NoError (t , db .ShiftState (ctx , s1 .ID , StateRevoked ))
119- s1 , err = db .GetSession (ctx , s1 .LocalPublicKey )
119+ s1 , err = db .GetSessionByLocalPub (ctx , s1 .LocalPublicKey )
120120 require .NoError (t , err )
121121 require .Equal (t , s1 .State , StateRevoked )
122122
@@ -299,7 +299,7 @@ func TestStateShift(t *testing.T) {
299299
300300 // Check that the session is in the StateCreated state. Also check that
301301 // the "RevokedAt" time has not yet been set.
302- s1 , err := db .GetSession (ctx , s1 .LocalPublicKey )
302+ s1 , err := db .GetSessionByLocalPub (ctx , s1 .LocalPublicKey )
303303 require .NoError (t , err )
304304 require .Equal (t , StateCreated , s1 .State )
305305 require .Equal (t , time.Time {}, s1 .RevokedAt )
@@ -310,7 +310,7 @@ func TestStateShift(t *testing.T) {
310310
311311 // This should have worked. Since it is now in a terminal state, the
312312 // "RevokedAt" time should be set.
313- s1 , err = db .GetSession (ctx , s1 .LocalPublicKey )
313+ s1 , err = db .GetSessionByLocalPub (ctx , s1 .LocalPublicKey )
314314 require .NoError (t , err )
315315 require .Equal (t , StateRevoked , s1 .State )
316316 require .True (t , clock .Now ().Equal (s1 .RevokedAt ))
0 commit comments