fix test

Author: wxiaoguang

modules/assetfs/embed.go

@@ -27,6 +27,7 @@ type EmbeddedFile interface {
 
 type EmbeddedFileInfo interface {
 	fs.FileInfo
+	fs.DirEntry
 	GetGzipContent() ([]byte, bool)
 }
 
@@ -143,13 +144,17 @@ func (e *embeddedFS) getFileInfo(fullName string) (*embeddedFileInfo, error) {
 
 	fields := strings.Split(fullName, "/")
 	fi = e.meta().Root
+	found := true
 	for _, field := range fields {
 		for _, child := range fi.Children {
-			if child.BaseName == field {
+			if found = child.BaseName == field; found {
 				fi = child
 				break
 			}
 		}
+		if !found {
+			return nil, fs.ErrNotExist
+		}
 	}
 
 	e.filesMu.Lock()
@@ -213,7 +218,7 @@ func (f *EmbeddedCompressedFile) Read(p []byte) (n int, err error) {
 func (f *EmbeddedFileBase) Seek(offset int64, whence int) (int64, error) {
 	switch whence {
 	case io.SeekStart:
-		f.seekPos = 0 + offset
+		f.seekPos = offset
 	case io.SeekCurrent:
 		f.seekPos += offset
 	case io.SeekEnd:
@@ -234,8 +239,6 @@ func (f *EmbeddedCompressedFile) Close() error {
 	return f.decompressor.Close()
 }
 
-var _ fs.FileInfo = (*embeddedFileInfo)(nil)
-
 func (fi *embeddedFileInfo) Name() string {
 	return fi.BaseName
 }
@@ -252,8 +255,6 @@ func (fi *embeddedFileInfo) ModTime() time.Time {
 	return fi.fs.meta().MaxModTime
 }
 
-var _ fs.DirEntry = (*embeddedFileInfo)(nil)
-
 func (fi *embeddedFileInfo) IsDir() bool {
 	return fi.Children != nil
 }

modules/assetfs/embed_test.go

@@ -27,6 +27,9 @@ func TestEmbed(t *testing.T) {
 	require.NoError(t, err)
 	efs := NewEmbeddedFS(data)
 
+	_, err = fs.ReadFile(efs, "not exist")
+	assert.ErrorIs(t, err, fs.ErrNotExist)
+
 	content, err := fs.ReadFile(efs, "a.txt")
 	require.NoError(t, err)
 	assert.Equal(t, "a", string(content))

modules/zstd/zstd_test.go

@@ -16,7 +16,7 @@ import (
 )
 
 func TestWriterReader(t *testing.T) {
-	testData := prepareTestData(t, 20_000_000)
+	testData := prepareTestData(t, 1_000_000)
 
 	result := bytes.NewBuffer(nil)
 
@@ -64,7 +64,7 @@ func TestWriterReader(t *testing.T) {
 }
 
 func TestSeekableWriterReader(t *testing.T) {
-	testData := prepareTestData(t, 20_000_000)
+	testData := prepareTestData(t, 2_000_000)
 
 	result := bytes.NewBuffer(nil)
 
@@ -109,15 +109,15 @@ func TestSeekableWriterReader(t *testing.T) {
 		reader, err := NewSeekableReader(assertReader)
 		require.NoError(t, err)
 
-		_, err = reader.Seek(10_000_000, io.SeekStart)
+		_, err = reader.Seek(1_000_000, io.SeekStart)
 		require.NoError(t, err)
 
 		data := make([]byte, 1000)
 		_, err = io.ReadFull(reader, data)
 		require.NoError(t, err)
 		require.NoError(t, reader.Close())
 
-		assert.Equal(t, testData[10_000_000:10_000_000+1000], data)
+		assert.Equal(t, testData[1_000_000:1_000_000+1000], data)
 
 		// Should seek 3 times,
 		// the first two times are for getting the index,