package protocol

import (
	"errors"
	"sync"
	"testing"
	"time"

	"github.com/stretchr/testify/suite"

	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/protocol/tt"
)

type MessengerCommunityForMobileTestingTestSuite struct {
	MessengerBaseTestSuite
}

func TestMessengerCommunityForMobileTesting(t *testing.T) {
	suite.Run(t, new(MessengerCommunityForMobileTestingTestSuite))
}

func (s *MessengerCommunityForMobileTestingTestSuite) SetupTest() {
	s.MessengerBaseTestSuite.SetupTest()
}

func (s *MessengerCommunityForMobileTestingTestSuite) TearDownTest() {
	s.MessengerBaseTestSuite.TearDownTest()
}

func (s *MessengerCommunityForMobileTestingTestSuite) TestCreateClosedCommunity() {
	var wg sync.WaitGroup
	wg.Add(1)
	// simulate invoking `HandleCommunityDescription`
	go func() {
		err := tt.RetryWithBackOff(func() error {
			r, err := s.m.RetrieveAll()
			s.Require().NoError(err)
			if len(r.Communities()) > 0 {
				return nil
			}
			return errors.New("not receive enough messages relate to community")
		})
		wg.Done()
		s.Require().NoError(err)
	}()

	wg.Add(1)
	var communityID types.HexBytes
	// simulate frontend(mobile) invoking `CreateClosedCommunity`
	go func() {
		response, err := s.m.CreateClosedCommunity()
		s.Require().NoError(err)
		s.Require().Len(response.Communities(), 1)
		s.Require().Len(response.Communities()[0].Categories(), 2)
		s.Require().Len(response.Chats(), 4)
		s.Require().Len(response.Communities()[0].Description().Chats, 4)
		communityID = response.Communities()[0].ID()
		wg.Done()
	}()

	timeout := time.After(20 * time.Second)
	done := make(chan bool)
	go func() {
		wg.Wait()
		close(done)
	}()

	select {
	case <-timeout:
		s.Fail("TestCreateClosedCommunity timed out")
	case <-done:
		// validate concurrent updating community result
		lastCommunity, err := s.m.GetCommunityByID(communityID)
		s.Require().NoError(err)
		s.Require().Len(lastCommunity.Categories(), 2)
		s.Require().Len(lastCommunity.Description().Chats, 4)
	}
}