compressedfs - fs_test.go

     1  package compressedfs
     2  
     3  import (
     4  	"bytes"
     5  	"compress/gzip"
     6  	"fmt"
     7  	"io"
     8  	"io/fs"
     9  	"strings"
    10  	"testing"
    11  	"testing/fstest"
    12  )
    13  
    14  func TestFS(t *testing.T) {
    15  	var adata bytes.Buffer
    16  
    17  	g := gzip.NewWriter(&adata)
    18  
    19  	g.Write([]byte("Hello!"))
    20  
    21  	g.Close()
    22  
    23  	fs := New(fstest.MapFS{
    24  		"a.gz": &fstest.MapFile{
    25  			Data: adata.Bytes(),
    26  			Mode: 0755,
    27  		},
    28  	})
    29  
    30  	if err := fstest.TestFS(fs, "a"); err != nil {
    31  		t.Error(err)
    32  	}
    33  }
    34  
    35  func TestCompressedFS(t *testing.T) {
    36  	mfs := make(fstest.MapFS)
    37  
    38  	for file, data := range map[string]string{
    39  		"a.gz":   "Hello!",
    40  		"b/c":    "Beep",
    41  		"b/c.gz": "Boop!",
    42  		"d":      "",
    43  	} {
    44  		var buf bytes.Buffer
    45  
    46  		if strings.HasSuffix(file, ext) {
    47  			g := gzip.NewWriter(&buf)
    48  
    49  			io.WriteString(g, data)
    50  			g.Close()
    51  		} else {
    52  			buf.WriteString(data)
    53  		}
    54  
    55  		mfs[file] = &fstest.MapFile{
    56  			Data: buf.Bytes(),
    57  			Mode: 0755,
    58  		}
    59  	}
    60  
    61  	cfs := New(mfs)
    62  
    63  	entries, err := fs.ReadDir(cfs, ".")
    64  	if err != nil {
    65  		t.Errorf("unexpected error: %v", err)
    66  	}
    67  
    68  	var es []string
    69  
    70  	for _, entry := range entries {
    71  		info, _ := entry.Info()
    72  
    73  		es = append(es, fmt.Sprintf("%d: %s", info.Size(), entry.Name()))
    74  	}
    75  
    76  	expectation := "6: a\n0: b\n0: d"
    77  
    78  	if result := strings.Join(es, "\n"); result != expectation {
    79  		t.Errorf("expecting to read from dir: %s, got %s", expectation, result)
    80  	}
    81  
    82  	entries, err = fs.ReadDir(cfs, "b")
    83  	if err != nil {
    84  		t.Errorf("unexpected error: %v", err)
    85  	}
    86  
    87  	es = es[:0]
    88  
    89  	for _, entry := range entries {
    90  		info, _ := entry.Info()
    91  
    92  		es = append(es, fmt.Sprintf("%d: %s", info.Size(), entry.Name()))
    93  	}
    94  
    95  	expectation = "4: c\n29: c.gz"
    96  
    97  	if result := strings.Join(es, "\n"); result != expectation {
    98  		t.Errorf("expecting to read from dir: %s, got %s", expectation, result)
    99  	}
   100  
   101  	s, err := fs.Stat(cfs, "a")
   102  	if err != nil {
   103  		t.Errorf("unexpected error: %v", err)
   104  	} else if size := s.Size(); size != 6 {
   105  		t.Errorf("expecting to get %d bytes, got %d", 6, size)
   106  	}
   107  
   108  	data, err := fs.ReadFile(cfs, "a")
   109  	if err != nil {
   110  		t.Errorf("unexpected error: %v", err)
   111  	} else if string(data) != "Hello!" {
   112  		t.Errorf("expecting to read %q, got %q", "Hello!", data)
   113  	}
   114  
   115  	s, err = fs.Stat(cfs, "b/c")
   116  	if err != nil {
   117  		t.Errorf("unexpected error: %v", err)
   118  	} else if size := s.Size(); size != 4 {
   119  		t.Errorf("expecting to get %d bytes, got %d", 6, size)
   120  	}
   121  
   122  	data, err = fs.ReadFile(cfs, "b/c")
   123  	if err != nil {
   124  		t.Errorf("unexpected error: %v", err)
   125  	} else if string(data) != "Beep" {
   126  		t.Errorf("expecting to read %q, got %q", "Beep", data)
   127  	}
   128  }
   129