diff --git a/internal/vault/vault.go b/internal/vault/vault.go index b535317..2243dc7 100644 --- a/internal/vault/vault.go +++ b/internal/vault/vault.go @@ -227,27 +227,23 @@ func (v *Vault) NumSecrets() (int, error) { return 0, fmt.Errorf("failed to read secrets directory: %w", err) } - // Count only directories that contain at least one version file + // Count only directories that have a "current" version pointer file count := 0 for _, entry := range entries { if !entry.IsDir() { continue } - // Check if this secret directory contains any version files + // A valid secret has a "current" file pointing to the active version secretDir := filepath.Join(secretsDir, entry.Name()) - versionFiles, err := afero.ReadDir(v.fs, secretDir) + currentFile := filepath.Join(secretDir, "current") + exists, err := afero.Exists(v.fs, currentFile) if err != nil { continue // Skip directories we can't read } - // Look for at least one version file (excluding "current" symlink) - for _, vFile := range versionFiles { - if !vFile.IsDir() && vFile.Name() != "current" { - count++ - - break // Found at least one version, count this secret - } + if exists { + count++ } } diff --git a/internal/vault/vault_test.go b/internal/vault/vault_test.go index bed6752..a69bbdf 100644 --- a/internal/vault/vault_test.go +++ b/internal/vault/vault_test.go @@ -162,6 +162,24 @@ func TestVaultOperations(t *testing.T) { } }) + // Test NumSecrets + t.Run("NumSecrets", func(t *testing.T) { + vlt, err := GetCurrentVault(fs, stateDir) + if err != nil { + t.Fatalf("Failed to get current vault: %v", err) + } + + numSecrets, err := vlt.NumSecrets() + if err != nil { + t.Fatalf("Failed to count secrets: %v", err) + } + + // We added one secret in SecretOperations + if numSecrets != 1 { + t.Errorf("Expected 1 secret, got %d", numSecrets) + } + }) + // Test unlocker operations t.Run("UnlockerOperations", func(t *testing.T) { vlt, err := GetCurrentVault(fs, stateDir)