From 1352bc880df4cd25eff65843973fcc0dd21b6304 Mon Sep 17 00:00:00 2001 From: Joshua Wong Date: Wed, 26 Feb 2020 00:58:07 +0800 Subject: Add hugo.IsProduction function Fixes #6873 --- common/hugo/hugo.go | 4 ++++ common/hugo/hugo_test.go | 4 ++++ 2 files changed, 8 insertions(+) (limited to 'common') diff --git a/common/hugo/hugo.go b/common/hugo/hugo.go index 62d923bf0..037880836 100644 --- a/common/hugo/hugo.go +++ b/common/hugo/hugo.go @@ -54,6 +54,10 @@ func (i Info) Generator() template.HTML { return template.HTML(fmt.Sprintf(``, CurrentVersion.String())) } +func (i Info) IsProduction() bool { + return i.Environment == EnvironmentProduction +} + // NewInfo creates a new Hugo Info object. func NewInfo(environment string) Info { if environment == "" { diff --git a/common/hugo/hugo_test.go b/common/hugo/hugo_test.go index 5be575b62..8840a9e9e 100644 --- a/common/hugo/hugo_test.go +++ b/common/hugo/hugo_test.go @@ -31,5 +31,9 @@ func TestHugoInfo(t *testing.T) { c.Assert(hugoInfo.BuildDate, qt.Equals, buildDate) c.Assert(hugoInfo.Environment, qt.Equals, "production") c.Assert(string(hugoInfo.Generator()), qt.Contains, fmt.Sprintf("Hugo %s", hugoInfo.Version())) + c.Assert(hugoInfo.IsProduction(), qt.Equals, true) + + devHugoInfo := NewInfo("development") + c.Assert(devHugoInfo.IsProduction(), qt.Equals, false) } -- cgit v1.2.3