assets/embed_gzip_test.go

// Copyright 2021 The Prometheus Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package assets

import (
	"embed"
	"io/ioutil"
	"strings"
	"testing"
)

//go:embed testdata
var EmbedFS embed.FS

var testFS = New(EmbedFS)

func TestFS(t *testing.T) {
	cases := []struct {
		name            string
		path            string
		expectedSize    int64
		expectedContent string
	}{
		{
			name:            "uncompressed file",
			path:            "testdata/uncompressed",
			expectedSize:    4,
			expectedContent: "foo\n",
		},
		{
			name:            "compressed file",
			path:            "testdata/compressed",
			expectedSize:    4,
			expectedContent: "foo\n",
		},
		{
			name:            "both, open uncompressed",
			path:            "testdata/both",
			expectedSize:    4,
			expectedContent: "foo\n",
		},
		{
			name:         "both, open compressed",
			path:         "testdata/both.gz",
			expectedSize: 29,
			// we don't check content for a explicitly compressed file
			expectedContent: "",
		},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			f, err := testFS.Open(c.path)
			if err != nil {
				t.Fatal(err)
			}

			stat, err := f.Stat()
			if err != nil {
				t.Fatal(err)
			}

			size := stat.Size()
			if size != c.expectedSize {
				t.Fatalf("size is wrong, expected %d, got %d", c.expectedSize, size)
			}

			if strings.HasSuffix(c.path, ".gz") {
				// don't read the comressed content
				return
			}

			content, err := ioutil.ReadAll(f)
			if err != nil {
				t.Fatal(err)
			}
			if string(content) != c.expectedContent {
				t.Fatalf("content is wrong, expected %s, got %s", c.expectedContent, string(content))
			}
		})
	}
}