github.com/navikt/knorten@v0.0.0-20240419132333-1333f46ed8b6/pkg/events/teamclient.go (about)

     1  package events
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/navikt/knorten/pkg/database"
     7  	"github.com/navikt/knorten/pkg/database/gensql"
     8  	"github.com/navikt/knorten/pkg/logger"
     9  )
    10  
    11  type teamClient interface {
    12  	Create(ctx context.Context, team gensql.Team, log logger.Logger) bool
    13  	Update(ctx context.Context, team gensql.Team, log logger.Logger) bool
    14  	Delete(ctx context.Context, teamID string, log logger.Logger) bool
    15  }
    16  
    17  type teamMock struct {
    18  	EventCounts map[database.EventType]int
    19  }
    20  
    21  func newTeamMock() teamMock {
    22  	return teamMock{
    23  		EventCounts: map[database.EventType]int{},
    24  	}
    25  }
    26  
    27  func (tm teamMock) Create(ctx context.Context, team gensql.Team, log logger.Logger) bool {
    28  	tm.EventCounts[database.EventTypeCreateTeam]++
    29  	return false
    30  }
    31  
    32  func (tm teamMock) Update(ctx context.Context, team gensql.Team, log logger.Logger) bool {
    33  	tm.EventCounts[database.EventTypeUpdateTeam]++
    34  	return false
    35  }
    36  
    37  func (tm teamMock) Delete(ctx context.Context, teamID string, log logger.Logger) bool {
    38  	tm.EventCounts[database.EventTypeDeleteTeam]++
    39  	return false
    40  }