diff options
Diffstat (limited to 'source/filesystem_test.go')
-rw-r--r-- | source/filesystem_test.go | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/source/filesystem_test.go b/source/filesystem_test.go new file mode 100644 index 000000000..ec7a305dc --- /dev/null +++ b/source/filesystem_test.go @@ -0,0 +1,111 @@ +// Copyright 2015 The Hugo Authors. All rights reserved. +// +// 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 source + +import ( + "fmt" + "path/filepath" + "runtime" + "testing" + + "github.com/gohugoio/hugo/modules" + + "github.com/gohugoio/hugo/langs" + + "github.com/spf13/afero" + + qt "github.com/frankban/quicktest" + "github.com/gohugoio/hugo/helpers" + "github.com/gohugoio/hugo/hugofs" + + "github.com/spf13/viper" +) + +func TestEmptySourceFilesystem(t *testing.T) { + c := qt.New(t) + ss := newTestSourceSpec() + src := ss.NewFilesystem("") + files, err := src.Files() + c.Assert(err, qt.IsNil) + if len(files) != 0 { + t.Errorf("new filesystem should contain 0 files.") + } +} + +func TestUnicodeNorm(t *testing.T) { + if runtime.GOOS != "darwin" { + // Normalization code is only for Mac OS, since it is not necessary for other OSes. + return + } + + c := qt.New(t) + + paths := []struct { + NFC string + NFD string + }{ + {NFC: "å", NFD: "\x61\xcc\x8a"}, + {NFC: "é", NFD: "\x65\xcc\x81"}, + } + + ss := newTestSourceSpec() + fi := hugofs.NewFileMetaInfo(nil, hugofs.FileMeta{}) + + for i, path := range paths { + base := fmt.Sprintf("base%d", i) + c.Assert(afero.WriteFile(ss.Fs.Source, filepath.Join(base, path.NFD), []byte("some data"), 0777), qt.IsNil) + src := ss.NewFilesystem(base) + _ = src.add(path.NFD, fi) + files, err := src.Files() + c.Assert(err, qt.IsNil) + f := files[0] + if f.BaseFileName() != path.NFC { + t.Fatalf("file %q name in NFD form should be normalized (%s)", f.BaseFileName(), path.NFC) + } + } + +} + +func newTestConfig() *viper.Viper { + v := viper.New() + v.Set("contentDir", "content") + v.Set("dataDir", "data") + v.Set("i18nDir", "i18n") + v.Set("layoutDir", "layouts") + v.Set("archetypeDir", "archetypes") + v.Set("resourceDir", "resources") + v.Set("publishDir", "public") + v.Set("assetDir", "assets") + _, err := langs.LoadLanguageSettings(v, nil) + if err != nil { + panic(err) + } + mod, err := modules.CreateProjectModule(v) + if err != nil { + panic(err) + } + v.Set("allModules", modules.Modules{mod}) + + return v +} + +func newTestSourceSpec() *SourceSpec { + v := newTestConfig() + fs := hugofs.NewFrom(hugofs.NewBaseFileDecorator(afero.NewMemMapFs()), v) + ps, err := helpers.NewPathSpec(fs, v, nil) + if err != nil { + panic(err) + } + return NewSourceSpec(ps, fs.Source) +} |