53 lines
1.9 KiB
Go
53 lines
1.9 KiB
Go
package vault
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/vault/sdk/version"
|
|
)
|
|
|
|
// TestStoreMultipleVaultVersions writes multiple versions of 1.9.0 and verifies that only
|
|
// the original timestamp is stored.
|
|
func TestStoreMultipleVaultVersions(t *testing.T) {
|
|
c, _, _ := TestCoreUnsealed(t)
|
|
upgradeTimePlusEpsilon := time.Now()
|
|
wasStored, err := c.storeVersionTimestamp(context.Background(), version.Version, upgradeTimePlusEpsilon.Add(30*time.Hour))
|
|
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)
|
|
}
|
|
}
|
|
|
|
// TestGetOldestVersion verifies that FindOldestVersionTimestamp finds the oldest
|
|
// (in time) vault version stored.
|
|
func TestGetOldestVersion(t *testing.T) {
|
|
c, _, _ := TestCoreUnsealed(t)
|
|
upgradeTimePlusEpsilon := time.Now()
|
|
|
|
// 1.6.2 is stored before 1.6.1, so even though it is a higher number, it should be returned.
|
|
c.storeVersionTimestamp(context.Background(), "1.6.2", upgradeTimePlusEpsilon.Add(-4*time.Hour))
|
|
c.storeVersionTimestamp(context.Background(), "1.6.1", upgradeTimePlusEpsilon.Add(2*time.Hour))
|
|
c.loadVersionTimestamps(c.activeContext)
|
|
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)
|
|
}
|
|
}
|