github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/gc/access_tokens_collector_test.go (about)

     1  package gc_test
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/pf-qiu/concourse/v6/atc/db/dbfakes"
     7  	"github.com/pf-qiu/concourse/v6/atc/gc"
     8  	. "github.com/onsi/ginkgo"
     9  	. "github.com/onsi/gomega"
    10  	"gopkg.in/square/go-jose.v2/jwt"
    11  )
    12  
    13  var _ = Describe("AccessTokensCollector", func() {
    14  	var collector GcCollector
    15  	var fakeLifecycle *dbfakes.FakeAccessTokenLifecycle
    16  
    17  	BeforeEach(func() {
    18  		fakeLifecycle = new(dbfakes.FakeAccessTokenLifecycle)
    19  
    20  		collector = gc.NewAccessTokensCollector(fakeLifecycle, jwt.DefaultLeeway)
    21  	})
    22  
    23  	Describe("Run", func() {
    24  		It("tells the access token lifecycle to remove expired access tokens", func() {
    25  			err := collector.Run(context.TODO())
    26  			Expect(err).NotTo(HaveOccurred())
    27  
    28  			Expect(fakeLifecycle.RemoveExpiredAccessTokensCallCount()).To(Equal(1))
    29  			leeway := fakeLifecycle.RemoveExpiredAccessTokensArgsForCall(0)
    30  			Expect(leeway).To(Equal(jwt.DefaultLeeway))
    31  		})
    32  	})
    33  })