106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package vault
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/vault/sdk/version"
|
|
)
|
|
|
|
// TestVersionStore_StoreMultipleVaultVersions writes multiple versions of 1.9.0 and verifies that only
|
|
// the original timestamp is stored.
|
|
func TestVersionStore_StoreMultipleVaultVersions(t *testing.T) {
|
|
c, _, _ := TestCoreUnsealed(t)
|
|
upgradeTimePlusEpsilon := time.Now().UTC()
|
|
wasStored, err := c.storeVersionTimestamp(context.Background(), version.Version, upgradeTimePlusEpsilon.Add(30*time.Hour), false)
|
|
if err != nil || wasStored {
|
|
t.Fatalf("vault version was re-stored: %v, err is: %s", wasStored, err.Error())
|
|
}
|
|
upgradeTime, ok := c.versionTimestamps[version.Version]
|
|
if !ok {
|
|
t.Fatalf("no %s version timestamp found", version.Version)
|
|
}
|
|
if upgradeTime.After(upgradeTimePlusEpsilon) {
|
|
t.Fatalf("upgrade time for %s is incorrect: got %+v, expected less than %+v", version.Version, upgradeTime, upgradeTimePlusEpsilon)
|
|
}
|
|
}
|
|
|
|
// TestVersionStore_GetOldestVersion verifies that FindOldestVersionTimestamp finds the oldest
|
|
// (in time) vault version stored.
|
|
func TestVersionStore_GetOldestVersion(t *testing.T) {
|
|
c, _, _ := TestCoreUnsealed(t)
|
|
upgradeTimePlusEpsilon := time.Now().UTC()
|
|
|
|
// 1.6.2 is stored before 1.6.1, so even though it is a higher number, it should be returned.
|
|
versionEntries := []struct {
|
|
version string
|
|
ts time.Time
|
|
}{
|
|
{"1.6.2", upgradeTimePlusEpsilon.Add(-4 * time.Hour)},
|
|
{"1.6.1", upgradeTimePlusEpsilon.Add(2 * time.Hour)},
|
|
}
|
|
|
|
for _, entry := range versionEntries {
|
|
_, err := c.storeVersionTimestamp(context.Background(), entry.version, entry.ts, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to write version entry %#v, err: %s", entry, err.Error())
|
|
}
|
|
}
|
|
|
|
err := c.loadVersionTimestamps(c.activeContext)
|
|
if err != nil {
|
|
t.Fatalf("failed to populate version history cache, err: %s", err.Error())
|
|
}
|
|
|
|
if len(c.versionTimestamps) != 3 {
|
|
t.Fatalf("expected 3 entries in timestamps map after refresh, found: %d", len(c.versionTimestamps))
|
|
}
|
|
v, tm, err := c.FindOldestVersionTimestamp()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if v != "1.6.2" {
|
|
t.Fatalf("expected 1.6.2, found: %s", v)
|
|
}
|
|
if tm.Before(upgradeTimePlusEpsilon.Add(-6*time.Hour)) || tm.After(upgradeTimePlusEpsilon.Add(-2*time.Hour)) {
|
|
t.Fatalf("incorrect upgrade time logged: %v", tm)
|
|
}
|
|
}
|
|
|
|
func TestVersionStore_SelfHealUTC(t *testing.T) {
|
|
c, _, _ := TestCoreUnsealed(t)
|
|
estLoc, err := time.LoadLocation("EST")
|
|
if err != nil {
|
|
t.Fatalf("failed to load location, err: %s", err.Error())
|
|
}
|
|
|
|
nowEST := time.Now().In(estLoc)
|
|
|
|
versionEntries := []struct {
|
|
version string
|
|
ts time.Time
|
|
}{
|
|
{"1.9.0", nowEST.Add(24 * time.Hour)},
|
|
{"1.9.1", nowEST.Add(48 * time.Hour)},
|
|
}
|
|
|
|
for _, entry := range versionEntries {
|
|
_, err := c.storeVersionTimestamp(context.Background(), entry.version, entry.ts, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to write version entry %#v, err: %s", entry, err.Error())
|
|
}
|
|
}
|
|
|
|
err = c.loadVersionTimestamps(c.activeContext)
|
|
if err != nil {
|
|
t.Fatalf("failed to load version timestamps, err: %s", err.Error())
|
|
}
|
|
|
|
for versionStr, ts := range c.versionTimestamps {
|
|
if ts.Location() != time.UTC {
|
|
t.Fatalf("failed to convert %s timestamp %s to UTC", versionStr, ts)
|
|
}
|
|
}
|
|
}
|