Properly implement session deep copy (#8602)

This commit is contained in:
Joram Wilander
2018-04-20 12:25:14 -04:00
committed by GitHub
parent 283f34b9c6
commit 853445dc2e
2 changed files with 45 additions and 2 deletions

View File

@@ -38,8 +38,21 @@ type Session struct {
}
func (me *Session) DeepCopy() *Session {
copy := *me
return &copy
copySession := *me
if me.Props != nil {
copySession.Props = CopyStringMap(me.Props)
}
if me.TeamMembers != nil {
copySession.TeamMembers = make([]*TeamMember, len(me.TeamMembers))
for index, tm := range me.TeamMembers {
copySession.TeamMembers[index] = new(TeamMember)
*copySession.TeamMembers[index] = *tm
}
}
return &copySession
}
func (me *Session) ToJson() string {

View File

@@ -7,8 +7,38 @@ import (
"strings"
"testing"
"time"
"github.com/stretchr/testify/assert"
)
func TestSessionDeepCopy(t *testing.T) {
sessionId := NewId()
userId := NewId()
mapKey := "key"
mapValue := "val"
session := &Session{Id: sessionId, Props: map[string]string{mapKey: mapValue}, TeamMembers: []*TeamMember{&TeamMember{UserId: userId, TeamId: "someteamId"}}}
copySession := session.DeepCopy()
copySession.Id = "changed"
copySession.Props[mapKey] = "changed"
copySession.TeamMembers[0].UserId = "changed"
assert.Equal(t, sessionId, session.Id)
assert.Equal(t, mapValue, session.Props[mapKey])
assert.Equal(t, userId, session.TeamMembers[0].UserId)
session = &Session{Id: sessionId}
copySession = session.DeepCopy()
assert.Equal(t, sessionId, session.Id)
session = &Session{TeamMembers: []*TeamMember{}}
copySession = session.DeepCopy()
assert.Equal(t, 0, len(copySession.TeamMembers))
}
func TestSessionJson(t *testing.T) {
session := Session{}
session.PreSave()