github.com/twelsh-aw/go/src@v0.0.0-20230516233729-a56fe86a7c81/io/fs/readdir_test.go (about)

     1  // Copyright 2020 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package fs_test
     6  
     7  import (
     8  	. "io/fs"
     9  	"os"
    10  	"testing"
    11  	"testing/fstest"
    12  	"time"
    13  )
    14  
    15  type readDirOnly struct{ ReadDirFS }
    16  
    17  func (readDirOnly) Open(name string) (File, error) { return nil, ErrNotExist }
    18  
    19  func TestReadDir(t *testing.T) {
    20  	check := func(desc string, dirs []DirEntry, err error) {
    21  		t.Helper()
    22  		if err != nil || len(dirs) != 2 || dirs[0].Name() != "hello.txt" || dirs[1].Name() != "sub" {
    23  			var names []string
    24  			for _, d := range dirs {
    25  				names = append(names, d.Name())
    26  			}
    27  			t.Errorf("ReadDir(%s) = %v, %v, want %v, nil", desc, names, err, []string{"hello.txt", "sub"})
    28  		}
    29  	}
    30  
    31  	// Test that ReadDir uses the method when present.
    32  	dirs, err := ReadDir(readDirOnly{testFsys}, ".")
    33  	check("readDirOnly", dirs, err)
    34  
    35  	// Test that ReadDir uses Open when the method is not present.
    36  	dirs, err = ReadDir(openOnly{testFsys}, ".")
    37  	check("openOnly", dirs, err)
    38  
    39  	// Test that ReadDir on Sub of . works (sub_test checks non-trivial subs).
    40  	sub, err := Sub(testFsys, ".")
    41  	if err != nil {
    42  		t.Fatal(err)
    43  	}
    44  	dirs, err = ReadDir(sub, ".")
    45  	check("sub(.)", dirs, err)
    46  }
    47  
    48  func TestFileInfoToDirEntry(t *testing.T) {
    49  	testFs := fstest.MapFS{
    50  		"notadir.txt": {
    51  			Data:    []byte("hello, world"),
    52  			Mode:    0,
    53  			ModTime: time.Now(),
    54  			Sys:     &sysValue,
    55  		},
    56  		"adir": {
    57  			Data:    nil,
    58  			Mode:    os.ModeDir,
    59  			ModTime: time.Now(),
    60  			Sys:     &sysValue,
    61  		},
    62  	}
    63  
    64  	tests := []struct {
    65  		path     string
    66  		wantMode FileMode
    67  		wantDir  bool
    68  	}{
    69  		{path: "notadir.txt", wantMode: 0, wantDir: false},
    70  		{path: "adir", wantMode: os.ModeDir, wantDir: true},
    71  	}
    72  
    73  	for _, test := range tests {
    74  		test := test
    75  		t.Run(test.path, func(t *testing.T) {
    76  			fi, err := Stat(testFs, test.path)
    77  			if err != nil {
    78  				t.Fatal(err)
    79  			}
    80  
    81  			dirEntry := FileInfoToDirEntry(fi)
    82  			if g, w := dirEntry.Type(), test.wantMode; g != w {
    83  				t.Errorf("FileMode mismatch: got=%v, want=%v", g, w)
    84  			}
    85  			if g, w := dirEntry.Name(), test.path; g != w {
    86  				t.Errorf("Name mismatch: got=%v, want=%v", g, w)
    87  			}
    88  			if g, w := dirEntry.IsDir(), test.wantDir; g != w {
    89  				t.Errorf("IsDir mismatch: got=%v, want=%v", g, w)
    90  			}
    91  		})
    92  	}
    93  }