package protocol import ( "encoding/json" "testing" "github.com/stretchr/testify/suite" "github.com/status-im/status-go/protocol/common" ) type ChatTestSuite struct { suite.Suite } func TestChatSuite(t *testing.T) { suite.Run(t, new(ChatTestSuite)) } func (s *ChatTestSuite) TestValidateChat() { testCases := []struct { Name string Valid bool Chat Chat }{ { Name: "valid one to one chat", Valid: true, Chat: Chat{ ID: "0x0424a68f89ba5fcd5e0640c1e1f591d561fa4125ca4e2a43592bc4123eca10ce064e522c254bb83079ba404327f6eafc01ec90a1444331fe769d3f3a7f90b0dde1", Name: "", ChatType: ChatTypeOneToOne, }, }, { Name: "valid public chat", Valid: true, Chat: Chat{ ID: "status", Name: "status", ChatType: ChatTypePublic, }, }, { Name: "empty chatID", Valid: false, Chat: Chat{ ID: "", Name: "status", ChatType: ChatTypePublic, }, }, { Name: "invalid one to one chat, wrong public key", Valid: false, Chat: Chat{ ID: "0xnotvalid", Name: "", ChatType: ChatTypeOneToOne, }, }, } for _, tc := range testCases { s.Run(tc.Name, func() { err := tc.Chat.Validate() if tc.Valid { s.Require().NoError(err) } else { s.Require().Error(err) } }) } } func (s *ChatTestSuite) TestUpdateFromMessage() { // Base case, clock is higher message := &common.Message{} chat := &Chat{} message.Clock = 1 s.Require().NoError(chat.UpdateFromMessage(message, &testTimeSource{})) s.Require().NotNil(chat.LastMessage) s.Require().Equal(uint64(1), chat.LastClockValue) // Clock is lower and lastMessage is not nil message = &common.Message{} lastMessage := message chat = &Chat{LastClockValue: 2, LastMessage: lastMessage} message.Clock = 1 s.Require().NoError(chat.UpdateFromMessage(message, &testTimeSource{})) s.Require().Equal(lastMessage, chat.LastMessage) s.Require().Equal(uint64(2), chat.LastClockValue) // Clock is lower and lastMessage is nil message = &common.Message{} chat = &Chat{LastClockValue: 2} message.Clock = 1 s.Require().NoError(chat.UpdateFromMessage(message, &testTimeSource{})) s.Require().NotNil(chat.LastMessage) s.Require().Equal(uint64(2), chat.LastClockValue) // Clock is higher but lastMessage has lower clock message then the receiving one message = &common.Message{} chat = &Chat{LastClockValue: 2} message.Clock = 1 s.Require().NoError(chat.UpdateFromMessage(message, &testTimeSource{})) s.Require().NotNil(chat.LastMessage) s.Require().Equal(uint64(2), chat.LastClockValue) chat.LastClockValue = 4 message = &common.Message{} message.Clock = 3 s.Require().NoError(chat.UpdateFromMessage(message, &testTimeSource{})) s.Require().Equal(chat.LastMessage, message) s.Require().Equal(uint64(4), chat.LastClockValue) } func (s *ChatTestSuite) TestSerializeJSON() { message := &common.Message{} chat := &Chat{} message.Clock = 1 message.Text = "`some markdown text`" s.Require().NoError(message.PrepareContent("")) message.ParsedTextAst = nil chat.LastMessage = message encodedJSON, err := json.Marshal(chat) s.Require().NoError(err) decodedChat := &Chat{} s.Require().NoError(json.Unmarshal(encodedJSON, decodedChat)) s.Require().Equal(chat, decodedChat) }