add controversy and imported to untrusted list

This commit is contained in:
Umputun
2021-09-01 14:21:52 -05:00
parent 72bdf1b176
commit 91441d1160
2 changed files with 16 additions and 10 deletions
+2
View File
@@ -86,9 +86,11 @@ func (c *Comment) PrepareUntrusted() {
c.Votes = make(map[string]bool)
c.VotedIPs = make(map[string]VotedIPInfo)
c.Score = 0
c.Controversy = 0
c.Edit = nil
c.Pin = false
c.Deleted = false
c.Imported = false
}
// SetDeleted clears comment info, reset to deleted state. hard flag will clear all user info as well
+14 -10
View File
@@ -87,16 +87,18 @@ func TestComment_Sanitize(t *testing.T) {
func TestComment_PrepareUntrusted(t *testing.T) {
comment := Comment{
Text: `blah`,
User: User{ID: "username"},
ParentID: "p123",
ID: "123",
Locator: Locator{SiteID: "site", URL: "url"},
Score: 10,
Pin: true,
Deleted: true,
Timestamp: time.Date(2018, 1, 1, 9, 30, 0, 0, time.Local),
Votes: map[string]bool{"uu": true},
Text: `blah`,
User: User{ID: "username"},
ParentID: "p123",
ID: "123",
Locator: Locator{SiteID: "site", URL: "url"},
Score: 10,
Pin: true,
Deleted: true,
Timestamp: time.Date(2018, 1, 1, 9, 30, 0, 0, time.Local),
Votes: map[string]bool{"uu": true},
Controversy: 123,
Imported: true,
}
comment.PrepareUntrusted()
@@ -110,6 +112,8 @@ func TestComment_PrepareUntrusted(t *testing.T) {
assert.Equal(t, make(map[string]bool), comment.Votes)
assert.Equal(t, make(map[string]VotedIPInfo), comment.VotedIPs)
assert.Equal(t, User{ID: "username"}, comment.User)
assert.Equal(t, 0., comment.Controversy)
assert.Equal(t, false, comment.Imported)
}