cedar_test.go

package cedar

import (
	"testing"

	"github.com/mongodb/amboy/queue"
	"github.com/stretchr/testify/suite"
)

type ServiceCacheSuite struct {
	cache *envState
	suite.Suite
}

func TestServiceCacheSuite(t *testing.T) {
	suite.Run(t, new(ServiceCacheSuite))
}

func (s *ServiceCacheSuite) SetupTest() {
	s.cache = &envState{name: "cedar.testing"}
}

func (s *ServiceCacheSuite) TestDefaultCacheValues() {
	s.Nil(s.cache.localQueue)
	s.Nil(s.cache.remoteQueue)
	s.Nil(s.cache.remoteManager)
	s.Equal("cedar.testing", s.cache.name)
}

func (s *ServiceCacheSuite) TestLocalQueueNotSettableToNil() {
	s.Error(s.cache.SetLocalQueue(nil))
	s.Nil(s.cache.localQueue)

	q := queue.NewLocalLimitedSize(2, 1024)
	s.NotNil(q)
	s.NoError(s.cache.SetLocalQueue(q))
	s.NotNil(s.cache.localQueue)
	s.Equal(s.cache.localQueue, q)
	s.Error(s.cache.SetLocalQueue(nil))
	s.NotNil(s.cache.localQueue)
	s.Equal(s.cache.localQueue, q)
}

func (s *ServiceCacheSuite) TestRemoteQueueNotSettableToNil() {
	s.Error(s.cache.SetRemoteQueue(nil))
	s.Nil(s.cache.remoteQueue)

	q := queue.NewLocalLimitedSize(2, 1024)
	s.NotNil(q)
	s.NoError(s.cache.SetRemoteQueue(q))
	s.NotNil(s.cache.remoteQueue)
	s.Equal(s.cache.remoteQueue, q)
	s.Error(s.cache.SetRemoteQueue(nil))
	s.NotNil(s.cache.remoteQueue)
	s.Equal(s.cache.remoteQueue, q)

	s.Nil(s.cache.remoteManager)
}

func (s *ServiceCacheSuite) TestQueueGetterRetrievesQueue() {
	q := s.cache.GetLocalQueue()
	s.Nil(q)

	q = s.cache.GetRemoteQueue()
	s.Nil(q)

	q = queue.NewLocalLimitedSize(1, 1)
	s.NotNil(q)
	s.NoError(s.cache.SetLocalQueue(q))

	retrieved := s.cache.GetLocalQueue()
	s.NotNil(retrieved)
	s.Equal(retrieved, q)
}