diff options
author | Niall Sheridan <nsheridan@gmail.com> | 2017-04-07 19:16:58 +0100 |
---|---|---|
committer | Niall Sheridan <nsheridan@gmail.com> | 2017-04-10 11:13:24 +0100 |
commit | 8b63d26271a9250e262962c2998a36bae36f8d20 (patch) | |
tree | 601951b42117a661440f699049dac7c05a55978f /server/store/store_test.go | |
parent | ee93aa5df5ee713703d5ce0e5bb3ac85ad47e35e (diff) |
fix behaviour of SQLStore.List
Diffstat (limited to 'server/store/store_test.go')
-rw-r--r-- | server/store/store_test.go | 24 |
1 files changed, 21 insertions, 3 deletions
diff --git a/server/store/store_test.go b/server/store/store_test.go index e8bebaf..cd58ccd 100644 --- a/server/store/store_test.go +++ b/server/store/store_test.go @@ -48,15 +48,29 @@ func testStore(t *testing.T, db CertStorer) { KeyID: "a", Principals: []string{"b"}, CreatedAt: time.Now().UTC(), - Expires: time.Now().UTC().Add(1 * time.Minute), + Expires: time.Now().UTC().Add(-1 * time.Second), Raw: "AAAAAA", } if err := db.SetRecord(r); err != nil { t.Error(err) } - if _, err := db.List(true); err != nil { + + // includeExpired = false should return 0 results + recs, err := db.List(false) + if err != nil { t.Error(err) } + if len(recs) > 0 { + t.Errorf("Expected 0 results, got %d", len(recs)) + } + // includeExpired = false should return 1 result + recs, err = db.List(true) + if err != nil { + t.Error(err) + } + if recs[0].KeyID != r.KeyID { + t.Error("key mismatch") + } c, _, _, _, _ := ssh.ParseAuthorizedKey(testdata.Cert) cert := c.(*ssh.Certificate) @@ -66,9 +80,13 @@ func testStore(t *testing.T, db CertStorer) { t.Error(err) } - if _, err := db.Get("key"); err != nil { + ret, err := db.Get("key") + if err != nil { t.Error(err) } + if ret.KeyID != cert.KeyId { + t.Error("key mismatch") + } if err := db.Revoke("key"); err != nil { t.Error(err) } |