mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
9a07b4d089
2978 files. The fork is 2026-07-15 (rebrand: gitea.dev -> github.com/hanzoai/git), so 2026 is the year, and "Copyright 2026 Hanzo AI, Inc." was already the form used by the handful of files that had ours. Shape is now: // Copyright 2026 Hanzo AI, Inc. All rights reserved. // Copyright 2015 The Gogs Authors. All rights reserved. // Copyright 2019 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT Ours on top, theirs retained verbatim. Retained because MIT requires it, and separately because they are true — those authors did write that code. Not renamed to "The Git Authors": Git is a different project under a different license, and putting its name here would attribute this code to people who have nothing to do with it. Adding our line is the fix; overwriting theirs would have been both a licence violation and a false statement. gofmt still reports import-order drift in some of these files. It predates this change (verified against a stash) and is left alone rather than burying a one-line-per-file legal change under 3000 files of unrelated reordering. Build clean. Co-authored-by: Hanzo Dev <[email protected]>
92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package internal
|
|
|
|
import (
|
|
"bytes"
|
|
"html/template"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRenderInternalAttrs(t *testing.T) {
|
|
cases := []struct {
|
|
input, protected, recovered string
|
|
}{
|
|
{
|
|
input: `<div class="test">class="content"</div>`,
|
|
protected: `<div data-attr-class="sec:test">class="content"</div>`,
|
|
recovered: `<div class="test">class="content"</div>`,
|
|
},
|
|
{
|
|
input: "<div\nclass=\"test\" data-xxx></div>",
|
|
protected: `<div data-attr-class="sec:test" data-xxx></div>`,
|
|
recovered: `<div class="test" data-xxx></div>`,
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
var r RenderInternal
|
|
out := &bytes.Buffer{}
|
|
in := r.init("sec", out, "")
|
|
protected := r.ProtectSafeAttrs(template.HTML(c.input))
|
|
assert.EqualValues(t, c.protected, protected)
|
|
_, _ = io.WriteString(in, string(protected))
|
|
_ = in.Close()
|
|
assert.Equal(t, c.recovered, out.String())
|
|
}
|
|
|
|
var r1, r2 RenderInternal
|
|
protected := r1.ProtectSafeAttrs(`<div class="test"></div>`)
|
|
assert.EqualValues(t, `<div class="test"></div>`, protected, "non-initialized RenderInternal should not protect any attributes")
|
|
_ = r1.init("sec", nil, "")
|
|
protected = r1.ProtectSafeAttrs(`<div class="test"></div>`)
|
|
assert.EqualValues(t, `<div data-attr-class="sec:test"></div>`, protected)
|
|
assert.Equal(t, "data-attr-class", r1.SafeAttr("class"))
|
|
assert.Equal(t, "sec:val", r1.SafeValue("val"))
|
|
recovered, ok := r1.RecoverProtectedValue("sec:val")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "val", recovered)
|
|
recovered, ok = r1.RecoverProtectedValue("other:val")
|
|
assert.False(t, ok)
|
|
assert.Empty(t, recovered)
|
|
|
|
out2 := &bytes.Buffer{}
|
|
in2 := r2.init("sec-other", out2, "")
|
|
_, _ = io.WriteString(in2, string(protected))
|
|
_ = in2.Close()
|
|
assert.Equal(t, `<div data-attr-class="sec:test"></div>`, out2.String(), "different secureID should not recover the value")
|
|
}
|
|
|
|
func TestRenderInternalExtraHead(t *testing.T) {
|
|
t.Run("HeadExists", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
var r RenderInternal
|
|
in := r.init("sec", out, `<MY-TAG>`)
|
|
_, _ = io.WriteString(in, `<head>any</head>`)
|
|
_ = in.Close()
|
|
assert.Equal(t, `<head><MY-TAG>any</head>`, out.String())
|
|
})
|
|
|
|
t.Run("HeadNotExists", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
var r RenderInternal
|
|
in := r.init("sec", out, `<MY-TAG>`)
|
|
_, _ = io.WriteString(in, `<div></div>`)
|
|
_ = in.Close()
|
|
assert.Equal(t, `<MY-TAG><div></div>`, out.String())
|
|
})
|
|
|
|
t.Run("NotHTML", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
var r RenderInternal
|
|
in := r.init("sec", out, `<MY-TAG>`)
|
|
_, _ = io.WriteString(in, `<any>`)
|
|
_ = in.Close()
|
|
assert.Equal(t, `<any>`, out.String())
|
|
})
|
|
}
|