github.com/status-im/status-go@v1.1.0/protocol/messenger_config_test.go (about)

     1  package protocol
     2  
     3  import (
     4  	"github.com/multiformats/go-multiaddr"
     5  	"github.com/stretchr/testify/suite"
     6  
     7  	"github.com/status-im/status-go/appdatabase"
     8  	"github.com/status-im/status-go/params"
     9  	"github.com/status-im/status-go/protocol/communities"
    10  	"github.com/status-im/status-go/services/mailservers"
    11  	"github.com/status-im/status-go/t/helpers"
    12  )
    13  
    14  func WithTestStoreNode(s *suite.Suite, id string, address multiaddr.Multiaddr, fleet string, collectiblesServiceMock *CollectiblesServiceMock) Option {
    15  	return func(c *config) error {
    16  		sqldb, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
    17  		s.Require().NoError(err)
    18  
    19  		db := mailservers.NewDB(sqldb)
    20  		err = db.Add(mailservers.Mailserver{
    21  			ID:    id,
    22  			Name:  id,
    23  			Addr:  &address,
    24  			Fleet: fleet,
    25  		})
    26  		s.Require().NoError(err)
    27  
    28  		c.mailserversDatabase = db
    29  		c.clusterConfig = params.ClusterConfig{Fleet: fleet}
    30  		c.communityTokensService = collectiblesServiceMock
    31  
    32  		return nil
    33  	}
    34  }
    35  
    36  func WithAutoRequestHistoricMessages(enabled bool) Option {
    37  	return func(c *config) error {
    38  		c.codeControlFlags.AutoRequestHistoricMessages = enabled
    39  		return nil
    40  	}
    41  }
    42  
    43  func WithCuratedCommunitiesUpdateLoop(enabled bool) Option {
    44  	return func(c *config) error {
    45  		c.codeControlFlags.CuratedCommunitiesUpdateLoopEnabled = enabled
    46  		return nil
    47  	}
    48  }
    49  
    50  func WithCommunityManagerOptions(options []communities.ManagerOption) Option {
    51  	return func(c *config) error {
    52  		c.communityManagerOptions = options
    53  		return nil
    54  	}
    55  }