parent
93be31d43a
commit
ddde272fb0
|
@ -102,7 +102,7 @@ func (s *v3Manager) Save(ctx context.Context, cfg clientv3.Config, dbPath string
|
||||||
defer os.RemoveAll(partpath)
|
defer os.RemoveAll(partpath)
|
||||||
|
|
||||||
var f *os.File
|
var f *os.File
|
||||||
f, err = os.Create(partpath)
|
f, err = os.OpenFile(partpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, fileutil.PrivateFileMode)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("could not open %s (%v)", partpath, err)
|
return fmt.Errorf("could not open %s (%v)", partpath, err)
|
||||||
}
|
}
|
||||||
|
|
|
@ -26,6 +26,7 @@ import (
|
||||||
|
|
||||||
"github.com/coreos/etcd/clientv3"
|
"github.com/coreos/etcd/clientv3"
|
||||||
"github.com/coreos/etcd/embed"
|
"github.com/coreos/etcd/embed"
|
||||||
|
"github.com/coreos/etcd/pkg/fileutil"
|
||||||
"github.com/coreos/etcd/pkg/testutil"
|
"github.com/coreos/etcd/pkg/testutil"
|
||||||
|
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
|
@ -141,6 +142,25 @@ func TestSnapshotV3RestoreMulti(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestSnapshotFilePermissions ensures that the snapshot is saved with
|
||||||
|
// the correct file permissions.
|
||||||
|
func TestSnapshotFilePermissions(t *testing.T) {
|
||||||
|
expectedFileMode := os.FileMode(fileutil.PrivateFileMode)
|
||||||
|
kvs := []kv{{"foo1", "bar1"}, {"foo2", "bar2"}, {"foo3", "bar3"}}
|
||||||
|
dbPath := createSnapshotFile(t, kvs)
|
||||||
|
defer os.RemoveAll(dbPath)
|
||||||
|
|
||||||
|
dbInfo, err := os.Stat(dbPath)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to get test snapshot file status: %v", err)
|
||||||
|
}
|
||||||
|
actualFileMode := dbInfo.Mode()
|
||||||
|
|
||||||
|
if expectedFileMode != actualFileMode {
|
||||||
|
t.Fatalf("expected test snapshot file mode %s, got %s:", expectedFileMode, actualFileMode)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
type kv struct {
|
type kv struct {
|
||||||
k, v string
|
k, v string
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue