package runner

import (
	nomadApi "github.com/hashicorp/nomad/api"
	"github.com/stretchr/testify/suite"
	"testing"
)

func TestJobStoreTestSuite(t *testing.T) {
	suite.Run(t, new(JobStoreTestSuite))
}

type JobStoreTestSuite struct {
	suite.Suite
	jobStorage *localNomadJobStorage
	job        *NomadEnvironment
}

func (s *JobStoreTestSuite) SetupTest() {
	s.jobStorage = NewLocalNomadJobStorage()
	s.job = &NomadEnvironment{environmentID: defaultEnvironmentID}
}

func (s *JobStoreTestSuite) TestAddedJobCanBeRetrieved() {
	s.jobStorage.Add(s.job)
	retrievedJob, ok := s.jobStorage.Get(s.job.ID())
	s.True(ok, "A saved runner should be retrievable")
	s.Equal(s.job, retrievedJob)
}

func (s *JobStoreTestSuite) TestJobWithSameIdOverwritesOldOne() {
	otherJobWithSameID := &NomadEnvironment{environmentID: defaultEnvironmentID}
	otherJobWithSameID.templateJob = &nomadApi.Job{}
	s.NotEqual(s.job, otherJobWithSameID)

	s.jobStorage.Add(s.job)
	s.jobStorage.Add(otherJobWithSameID)
	retrievedJob, _ := s.jobStorage.Get(s.job.ID())
	s.NotEqual(s.job, retrievedJob)
	s.Equal(otherJobWithSameID, retrievedJob)
}

func (s *JobStoreTestSuite) TestDeletedJobIsNotAccessible() {
	s.jobStorage.Add(s.job)
	s.jobStorage.Delete(s.job.ID())
	retrievedRunner, ok := s.jobStorage.Get(s.job.ID())
	s.Nil(retrievedRunner)
	s.False(ok, "A deleted runner should not be accessible")
}

func (s *JobStoreTestSuite) TestLenOfEmptyPoolIsZero() {
	s.Equal(0, s.jobStorage.Length())
}

func (s *JobStoreTestSuite) TestLenChangesOnStoreContentChange() {
	s.Run("len increases when job is added", func() {
		s.jobStorage.Add(s.job)
		s.Equal(1, s.jobStorage.Length())
	})

	s.Run("len does not increase when job with same id is added", func() {
		s.jobStorage.Add(s.job)
		s.Equal(1, s.jobStorage.Length())
	})

	s.Run("len increases again when different job is added", func() {
		anotherJob := &NomadEnvironment{environmentID: anotherEnvironmentID}
		s.jobStorage.Add(anotherJob)
		s.Equal(2, s.jobStorage.Length())
	})

	s.Run("len decreases when job is deleted", func() {
		s.jobStorage.Delete(s.job.ID())
		s.Equal(1, s.jobStorage.Length())
	})
}