197 lines
4.1 KiB
Go
197 lines
4.1 KiB
Go
// Copyright 2018 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 filecache
|
|
|
|
import (
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/spf13/afero"
|
|
|
|
"github.com/gohugoio/hugo/config"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDecodeConfig(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
assert := require.New(t)
|
|
|
|
configStr := `
|
|
resourceDir = "myresources"
|
|
contentDir = "content"
|
|
dataDir = "data"
|
|
i18nDir = "i18n"
|
|
layoutDir = "layouts"
|
|
assetDir = "assets"
|
|
archetypeDir = "archetypes"
|
|
|
|
[caches]
|
|
[caches.getJSON]
|
|
maxAge = "10m"
|
|
dir = "/path/to/c1"
|
|
[caches.getCSV]
|
|
maxAge = "11h"
|
|
dir = "/path/to/c2"
|
|
[caches.images]
|
|
dir = "/path/to/c3"
|
|
|
|
`
|
|
|
|
cfg, err := config.FromConfigString(configStr, "toml")
|
|
assert.NoError(err)
|
|
fs := afero.NewMemMapFs()
|
|
decoded, err := DecodeConfig(fs, cfg)
|
|
assert.NoError(err)
|
|
|
|
assert.Equal(5, len(decoded))
|
|
|
|
c2 := decoded["getcsv"]
|
|
assert.Equal("11h0m0s", c2.MaxAge.String())
|
|
assert.Equal(filepath.FromSlash("/path/to/c2/filecache/getcsv"), c2.Dir)
|
|
|
|
c3 := decoded["images"]
|
|
assert.Equal(time.Duration(-1), c3.MaxAge)
|
|
assert.Equal(filepath.FromSlash("/path/to/c3/filecache/images"), c3.Dir)
|
|
|
|
}
|
|
|
|
func TestDecodeConfigIgnoreCache(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
assert := require.New(t)
|
|
|
|
configStr := `
|
|
resourceDir = "myresources"
|
|
contentDir = "content"
|
|
dataDir = "data"
|
|
i18nDir = "i18n"
|
|
layoutDir = "layouts"
|
|
assetDir = "assets"
|
|
archeTypedir = "archetypes"
|
|
|
|
ignoreCache = true
|
|
[caches]
|
|
[caches.getJSON]
|
|
maxAge = 1234
|
|
dir = "/path/to/c1"
|
|
[caches.getCSV]
|
|
maxAge = 3456
|
|
dir = "/path/to/c2"
|
|
[caches.images]
|
|
dir = "/path/to/c3"
|
|
|
|
`
|
|
|
|
cfg, err := config.FromConfigString(configStr, "toml")
|
|
assert.NoError(err)
|
|
fs := afero.NewMemMapFs()
|
|
decoded, err := DecodeConfig(fs, cfg)
|
|
assert.NoError(err)
|
|
|
|
assert.Equal(5, len(decoded))
|
|
|
|
for _, v := range decoded {
|
|
assert.Equal(time.Duration(0), v.MaxAge)
|
|
}
|
|
|
|
}
|
|
|
|
func TestDecodeConfigDefault(t *testing.T) {
|
|
assert := require.New(t)
|
|
cfg := newTestConfig()
|
|
|
|
if runtime.GOOS == "windows" {
|
|
cfg.Set("resourceDir", "c:\\cache\\resources")
|
|
cfg.Set("cacheDir", "c:\\cache\\thecache")
|
|
|
|
} else {
|
|
cfg.Set("resourceDir", "/cache/resources")
|
|
cfg.Set("cacheDir", "/cache/thecache")
|
|
}
|
|
|
|
fs := afero.NewMemMapFs()
|
|
|
|
decoded, err := DecodeConfig(fs, cfg)
|
|
|
|
assert.NoError(err)
|
|
|
|
assert.Equal(5, len(decoded))
|
|
|
|
imgConfig := decoded[cacheKeyImages]
|
|
jsonConfig := decoded[cacheKeyGetJSON]
|
|
|
|
if runtime.GOOS == "windows" {
|
|
assert.Equal(filepath.FromSlash("_gen/images"), imgConfig.Dir)
|
|
} else {
|
|
assert.Equal("_gen/images", imgConfig.Dir)
|
|
assert.Equal("/cache/thecache/hugoproject/filecache/getjson", jsonConfig.Dir)
|
|
}
|
|
|
|
assert.True(imgConfig.isResourceDir)
|
|
assert.False(jsonConfig.isResourceDir)
|
|
}
|
|
|
|
func TestDecodeConfigInvalidDir(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
assert := require.New(t)
|
|
|
|
configStr := `
|
|
resourceDir = "myresources"
|
|
contentDir = "content"
|
|
dataDir = "data"
|
|
i18nDir = "i18n"
|
|
layoutDir = "layouts"
|
|
assetDir = "assets"
|
|
archeTypedir = "archetypes"
|
|
|
|
[caches]
|
|
[caches.getJSON]
|
|
maxAge = "10m"
|
|
dir = "/"
|
|
|
|
`
|
|
if runtime.GOOS == "windows" {
|
|
configStr = strings.Replace(configStr, "/", "c:\\\\", 1)
|
|
}
|
|
|
|
cfg, err := config.FromConfigString(configStr, "toml")
|
|
assert.NoError(err)
|
|
fs := afero.NewMemMapFs()
|
|
|
|
_, err = DecodeConfig(fs, cfg)
|
|
assert.Error(err)
|
|
|
|
}
|
|
|
|
func newTestConfig() *viper.Viper {
|
|
cfg := viper.New()
|
|
cfg.Set("workingDir", filepath.FromSlash("/my/cool/hugoproject"))
|
|
cfg.Set("contentDir", "content")
|
|
cfg.Set("dataDir", "data")
|
|
cfg.Set("resourceDir", "resources")
|
|
cfg.Set("i18nDir", "i18n")
|
|
cfg.Set("layoutDir", "layouts")
|
|
cfg.Set("archetypeDir", "archetypes")
|
|
cfg.Set("assetDir", "assets")
|
|
|
|
return cfg
|
|
}
|