Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(577)

Unified Diff: vpython/filesystem/testfs/build.go

Issue 2703793003: vpython: Add filesystem operation / test packages. (Closed)
Patch Set: Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: vpython/filesystem/testfs/build.go
diff --git a/vpython/filesystem/testfs/build.go b/vpython/filesystem/testfs/build.go
new file mode 100644
index 0000000000000000000000000000000000000000..fbbe42127aefd002438daf025c7cbdeea7314773
--- /dev/null
+++ b/vpython/filesystem/testfs/build.go
@@ -0,0 +1,51 @@
+// Copyright 2017 The LUCI Authors. All rights reserved.
+// Use of this source code is governed under the Apache License, Version 2.0
+// that can be found in the LICENSE file.
+
+package testfs
+
+import (
+ "io/ioutil"
+ "path/filepath"
+ "sort"
+
+ "github.com/luci/luci-go/vpython/filesystem"
+)
+
+// BuildDir is a sentinel value that can be provided to Build as content to
+// instruct it to make a directory instead of a file.
+var BuildDir = "\x00"
iannucci 2017/02/21 08:25:08 why not layout := map[string]string { "some/dir
dnj 2017/02/21 18:07:49 Done.
+
+// Build constructs a filesystem hierarchy given a layout.
+//
+// The layouts keys should be ToSlash-style file paths. Its values should be the
+// content that is written at those paths. Intermediate directories will be
+// automatically created.
+func Build(base string, layout map[string]string) error {
+ keys := make([]string, 0, len(layout))
+ for k := range layout {
+ keys = append(keys, k)
+ }
+ sort.Strings(keys)
+
+ for _, path := range keys {
+ content := layout[path]
+ path = filepath.Join(base, filepath.FromSlash(path))
+
+ if content == BuildDir {
+ // Make a directory.
+ if err := filesystem.MakeDirs(path); err != nil {
+ return err
+ }
+ } else {
+ // Make a file.
+ if err := filesystem.MakeDirs(filepath.Dir(path)); err != nil {
+ return err
+ }
+ if err := ioutil.WriteFile(path, []byte(content), 0644); err != nil {
+ return err
+ }
+ }
+ }
+ return nil
+}

Powered by Google App Engine
This is Rietveld 408576698