integrity_test.go (2091B)
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]any{"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 }