mirror of
https://github.com/fleetdm/fleet
synced 2026-04-21 13:37:30 +00:00
b21f54d648
For #26218 - Added signup_token authentication for Android enterprise callback and fixed API path to match API doc # Checklist for submitter - [x] Added/updated automated tests - [x] Manual QA for all new/changed functionality
125 lines
3.8 KiB
Go
125 lines
3.8 KiB
Go
// Automatically generated by mockimpl. DO NOT EDIT!
|
|
|
|
package mock
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/mdm/android"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
var _ android.Datastore = (*Datastore)(nil)
|
|
|
|
type CreateEnterpriseFunc func(ctx context.Context, userID uint) (uint, error)
|
|
|
|
type GetEnterpriseByIDFunc func(ctx context.Context, ID uint) (*android.EnterpriseDetails, error)
|
|
|
|
type GetEnterpriseBySignupTokenFunc func(ctx context.Context, signupToken string) (*android.EnterpriseDetails, error)
|
|
|
|
type GetEnterpriseFunc func(ctx context.Context) (*android.Enterprise, error)
|
|
|
|
type UpdateEnterpriseFunc func(ctx context.Context, enterprise *android.EnterpriseDetails) error
|
|
|
|
type DeleteAllEnterprisesFunc func(ctx context.Context) error
|
|
|
|
type DeleteOtherEnterprisesFunc func(ctx context.Context, ID uint) error
|
|
|
|
type CreateDeviceTxFunc func(ctx context.Context, tx sqlx.ExtContext, device *android.Device) (*android.Device, error)
|
|
|
|
type UpdateDeviceTxFunc func(ctx context.Context, tx sqlx.ExtContext, device *android.Device) error
|
|
|
|
type Datastore struct {
|
|
CreateEnterpriseFunc CreateEnterpriseFunc
|
|
CreateEnterpriseFuncInvoked bool
|
|
|
|
GetEnterpriseByIDFunc GetEnterpriseByIDFunc
|
|
GetEnterpriseByIDFuncInvoked bool
|
|
|
|
GetEnterpriseBySignupTokenFunc GetEnterpriseBySignupTokenFunc
|
|
GetEnterpriseBySignupTokenFuncInvoked bool
|
|
|
|
GetEnterpriseFunc GetEnterpriseFunc
|
|
GetEnterpriseFuncInvoked bool
|
|
|
|
UpdateEnterpriseFunc UpdateEnterpriseFunc
|
|
UpdateEnterpriseFuncInvoked bool
|
|
|
|
DeleteAllEnterprisesFunc DeleteAllEnterprisesFunc
|
|
DeleteAllEnterprisesFuncInvoked bool
|
|
|
|
DeleteOtherEnterprisesFunc DeleteOtherEnterprisesFunc
|
|
DeleteOtherEnterprisesFuncInvoked bool
|
|
|
|
CreateDeviceTxFunc CreateDeviceTxFunc
|
|
CreateDeviceTxFuncInvoked bool
|
|
|
|
UpdateDeviceTxFunc UpdateDeviceTxFunc
|
|
UpdateDeviceTxFuncInvoked bool
|
|
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func (ds *Datastore) CreateEnterprise(ctx context.Context, userID uint) (uint, error) {
|
|
ds.mu.Lock()
|
|
ds.CreateEnterpriseFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.CreateEnterpriseFunc(ctx, userID)
|
|
}
|
|
|
|
func (ds *Datastore) GetEnterpriseByID(ctx context.Context, ID uint) (*android.EnterpriseDetails, error) {
|
|
ds.mu.Lock()
|
|
ds.GetEnterpriseByIDFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.GetEnterpriseByIDFunc(ctx, ID)
|
|
}
|
|
|
|
func (ds *Datastore) GetEnterpriseBySignupToken(ctx context.Context, signupToken string) (*android.EnterpriseDetails, error) {
|
|
ds.mu.Lock()
|
|
ds.GetEnterpriseBySignupTokenFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.GetEnterpriseBySignupTokenFunc(ctx, signupToken)
|
|
}
|
|
|
|
func (ds *Datastore) GetEnterprise(ctx context.Context) (*android.Enterprise, error) {
|
|
ds.mu.Lock()
|
|
ds.GetEnterpriseFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.GetEnterpriseFunc(ctx)
|
|
}
|
|
|
|
func (ds *Datastore) UpdateEnterprise(ctx context.Context, enterprise *android.EnterpriseDetails) error {
|
|
ds.mu.Lock()
|
|
ds.UpdateEnterpriseFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.UpdateEnterpriseFunc(ctx, enterprise)
|
|
}
|
|
|
|
func (ds *Datastore) DeleteAllEnterprises(ctx context.Context) error {
|
|
ds.mu.Lock()
|
|
ds.DeleteAllEnterprisesFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.DeleteAllEnterprisesFunc(ctx)
|
|
}
|
|
|
|
func (ds *Datastore) DeleteOtherEnterprises(ctx context.Context, ID uint) error {
|
|
ds.mu.Lock()
|
|
ds.DeleteOtherEnterprisesFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.DeleteOtherEnterprisesFunc(ctx, ID)
|
|
}
|
|
|
|
func (ds *Datastore) CreateDeviceTx(ctx context.Context, tx sqlx.ExtContext, device *android.Device) (*android.Device, error) {
|
|
ds.mu.Lock()
|
|
ds.CreateDeviceTxFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.CreateDeviceTxFunc(ctx, tx, device)
|
|
}
|
|
|
|
func (ds *Datastore) UpdateDeviceTx(ctx context.Context, tx sqlx.ExtContext, device *android.Device) error {
|
|
ds.mu.Lock()
|
|
ds.UpdateDeviceTxFuncInvoked = true
|
|
ds.mu.Unlock()
|
|
return ds.UpdateDeviceTxFunc(ctx, tx, device)
|
|
}
|