diff options
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) } |