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

Unified Diff: common/dirwalk/tests/tools/walkdir/fileprocessor_verify.go

Issue 2054763004: luci-go/common/dirwalk: Code for walking a directory tree efficiently Base URL: https://github.com/luci/luci-go@master
Patch Set: Major rewrite of the code. Created 4 years, 1 month 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: common/dirwalk/tests/tools/walkdir/fileprocessor_verify.go
diff --git a/common/dirwalk/tests/tools/walkdir/fileprocessor_verify.go b/common/dirwalk/tests/tools/walkdir/fileprocessor_verify.go
new file mode 100644
index 0000000000000000000000000000000000000000..a21833e6e626e011da0141dc0af19b20668a2901
--- /dev/null
+++ b/common/dirwalk/tests/tools/walkdir/fileprocessor_verify.go
@@ -0,0 +1,67 @@
+// Copyright 2016 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 main
+
+import (
+ "io"
+ "log"
+ "os"
+ "strings"
+)
+
+// VerifyFileProcessor implements FileProcessor. It verifies that things are called in the right order.
+type VerifyFileProcessor struct {
+ BaseFileProcessor
+ rootDir string
+ finishedDirs []string
+}
+
+func (p *VerifyFileProcessor) check(path string) {
+ for _, d := range p.finishedDirs {
+ if strings.HasPrefix(path, d) {
+ log.Fatalf("Directory called before content\n Dir: %s\nContent: %s\n", path, d)
+ }
+ }
+}
+
+func (p *VerifyFileProcessor) checkFile(path string) {
+ stat, err := os.Stat(path)
+ if err != nil {
+ log.Fatalf("Error while stating small file\n%s", path)
+ }
+ if stat.IsDir() {
+ log.Fatalf("File function called with directory!\n%s", path)
+ }
+ p.check(path)
+}
+
+func (p *VerifyFileProcessor) Dir(path string) {
+ stat, err := os.Stat(path)
+ if err != nil {
+ log.Fatalf("Error while stating directory\n%s", path)
+ }
+ if !stat.IsDir() {
+ log.Fatalf("Dir() called with non-directory!\n%s", path)
+ }
+ p.check(path)
+ p.finishedDirs = append(p.finishedDirs, path)
+ p.BaseFileProcessor.Dir(path)
+}
+
+func (p *VerifyFileProcessor) SmallFile(path string, r io.ReadCloser) {
+ p.checkFile(path)
+ p.BaseFileProcessor.SmallFile(path, r)
+}
+
+func (p *VerifyFileProcessor) LargeFile(path string, r io.ReadCloser) {
+ p.checkFile(path)
+ p.BaseFileProcessor.LargeFile(path, r)
+}
+
+func (p *VerifyFileProcessor) Finished() {
+ if strings.Compare(p.finishedDirs[len(p.finishedDirs)-1], p.rootDir) != 0 {
+ log.Fatalf("Last directory should be the root\n%s\n%s", p.finishedDirs[len(p.finishedDirs)-1], p.rootDir)
+ }
+}

Powered by Google App Engine
This is Rietveld 408576698