github.com/graemephi/kahugo@v0.62.3-0.20211121071557-d78c0423784d/resources/resource_transformers/integrity/integrity_test.go (about)

     1  // Copyright 2019 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 integrity
    15  
    16  import (
    17  	"html/template"
    18  	"testing"
    19  
    20  	"github.com/gohugoio/hugo/resources/resource"
    21  
    22  	qt "github.com/frankban/quicktest"
    23  	"github.com/gohugoio/hugo/resources/resource_transformers/htesting"
    24  )
    25  
    26  func TestHashFromAlgo(t *testing.T) {
    27  	for _, algo := range []struct {
    28  		name string
    29  		bits int
    30  	}{
    31  		{"md5", 128},
    32  		{"sha256", 256},
    33  		{"sha384", 384},
    34  		{"sha512", 512},
    35  		{"shaman", -1},
    36  	} {
    37  		t.Run(algo.name, func(t *testing.T) {
    38  			c := qt.New(t)
    39  			h, err := newHash(algo.name)
    40  			if algo.bits > 0 {
    41  				c.Assert(err, qt.IsNil)
    42  				c.Assert(h.Size(), qt.Equals, algo.bits/8)
    43  			} else {
    44  				c.Assert(err, qt.Not(qt.IsNil))
    45  				c.Assert(err.Error(), qt.Contains, "use either md5, sha256, sha384 or sha512")
    46  			}
    47  		})
    48  	}
    49  }
    50  
    51  func TestTransform(t *testing.T) {
    52  	c := qt.New(t)
    53  
    54  	spec, err := htesting.NewTestResourceSpec()
    55  	c.Assert(err, qt.IsNil)
    56  	client := New(spec)
    57  
    58  	r, err := htesting.NewResourceTransformerForSpec(spec, "hugo.txt", "Hugo Rocks!")
    59  	c.Assert(err, qt.IsNil)
    60  
    61  	transformed, err := client.Fingerprint(r, "")
    62  
    63  	c.Assert(err, qt.IsNil)
    64  	c.Assert(transformed.RelPermalink(), qt.Equals, "/hugo.a5ad1c6961214a55de53c1ce6e60d27b6b761f54851fa65e33066460dfa6a0db.txt")
    65  	c.Assert(transformed.Data(), qt.DeepEquals, map[string]interface{}{"Integrity": template.HTMLAttr("sha256-pa0caWEhSlXeU8HObmDSe2t2H1SFH6ZeMwZkYN+moNs=")})
    66  	content, err := transformed.(resource.ContentProvider).Content()
    67  	c.Assert(err, qt.IsNil)
    68  	c.Assert(content, qt.Equals, "Hugo Rocks!")
    69  }