github.com/kovansky/hugo@v0.92.3-0.20220224232819-63076e4ff19f/config/security/whitelist_test.go (about)

     1  // Copyright 2021 The Hugo Authors. All rights reserved.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  // http://www.apache.org/licenses/LICENSE-2.0
     7  //
     8  // Unless required by applicable law or agreed to in writing, software
     9  // distributed under the License is distributed on an "AS IS" BASIS,
    10  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package security
    15  
    16  import (
    17  	"testing"
    18  
    19  	qt "github.com/frankban/quicktest"
    20  )
    21  
    22  func TestWhitelist(t *testing.T) {
    23  	t.Parallel()
    24  	c := qt.New(t)
    25  
    26  	c.Run("none", func(c *qt.C) {
    27  		c.Assert(NewWhitelist("none", "foo").Accept("foo"), qt.IsFalse)
    28  		c.Assert(NewWhitelist().Accept("foo"), qt.IsFalse)
    29  		c.Assert(NewWhitelist("").Accept("foo"), qt.IsFalse)
    30  		c.Assert(NewWhitelist("  ", " ").Accept("foo"), qt.IsFalse)
    31  		c.Assert(Whitelist{}.Accept("foo"), qt.IsFalse)
    32  	})
    33  
    34  	c.Run("One", func(c *qt.C) {
    35  		w := NewWhitelist("^foo.*")
    36  		c.Assert(w.Accept("foo"), qt.IsTrue)
    37  		c.Assert(w.Accept("mfoo"), qt.IsFalse)
    38  	})
    39  
    40  	c.Run("Multiple", func(c *qt.C) {
    41  		w := NewWhitelist("^foo.*", "^bar.*")
    42  		c.Assert(w.Accept("foo"), qt.IsTrue)
    43  		c.Assert(w.Accept("bar"), qt.IsTrue)
    44  		c.Assert(w.Accept("mbar"), qt.IsFalse)
    45  	})
    46  
    47  }