Skip to content

Commit

Permalink
Merge 9fc0fc3 into backport/gh-19720-empty-server-metadata/officially…
Browse files Browse the repository at this point in the history
…-game-platypus
  • Loading branch information
hc-github-team-consul-core authored Dec 19, 2023
2 parents 6b970f1 + 9fc0fc3 commit f68a412
Show file tree
Hide file tree
Showing 2 changed files with 34 additions and 1 deletion.
2 changes: 1 addition & 1 deletion agent/consul/server_metadata.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ func (md *ServerMetadata) IsLastSeenStale(d time.Duration) bool {
// OpenServerMetadata is a helper function for opening the server metadata file
// with the correct permissions.
func OpenServerMetadata(filename string) (io.WriteCloser, error) {
return os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
return os.OpenFile(filename, os.O_WRONLY|os.O_CREATE, 0600)
}

type ServerMetadataReadFunc func(filename string) (*ServerMetadata, error)
Expand Down
33 changes: 33 additions & 0 deletions agent/consul/server_metadata_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ package consul
import (
"bytes"
"errors"
"os"
"testing"
"time"

Expand Down Expand Up @@ -66,3 +67,35 @@ func TestWriteServerMetadata(t *testing.T) {
assert.True(t, b.Len() > 0)
})
}

func TestWriteServerMetadata_MultipleTimes(t *testing.T) {
file, err := os.CreateTemp("", "server_metadata.json")
assert.NoError(t, err)
defer os.Remove(file.Name())

// prepare some data in server_metadata.json
_, err = OpenServerMetadata(file.Name())
assert.NoError(t, err)
err = WriteServerMetadata(file)
assert.NoError(t, err)
stat, err := file.Stat()
assert.NoError(t, err)

t.Run("reopen not truncate file", func(t *testing.T) {
_, err = OpenServerMetadata(file.Name())
assert.NoError(t, err)

// file size unchanged
stat2, err := file.Stat()
assert.NoError(t, err)
assert.Equal(t, stat.Size(), stat2.Size())
})

t.Run("write updates the file", func(t *testing.T) {
err = WriteServerMetadata(file)
assert.NoError(t, err)
stat2, err := file.Stat()
assert.NoError(t, err)
assert.Equal(t, stat2.ModTime(), stat2.ModTime())
})
}

0 comments on commit f68a412

Please sign in to comment.