111 lines
2.7 KiB
Go
111 lines
2.7 KiB
Go
// 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/config"
|
|
|
|
"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"
|
|
)
|
|
|
|
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.NewFileMeta())
|
|
|
|
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() config.Provider {
|
|
v := config.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, nil, fs.Source)
|
|
}
|