Fix the fix for tests

This commit is contained in:
Martin Angers 2024-04-08 11:34:31 -04:00
parent f9e1bc2e97
commit f5129bfa1a
4 changed files with 8 additions and 8 deletions

View file

@ -182,7 +182,7 @@ func TestApplyTeamSpecs(t *testing.T) {
return map[string]uint{fleet.BuiltinMacOS14PlusLabelName: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}
@ -589,7 +589,7 @@ func TestApplyAppConfig(t *testing.T) {
return map[string]uint{fleet.BuiltinMacOS14PlusLabelName: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}
@ -1173,7 +1173,7 @@ func TestApplyAsGitOps(t *testing.T) {
require.ElementsMatch(t, labels, []string{fleet.BuiltinMacOS14PlusLabelName})
return map[string]uint{fleet.BuiltinMacOS14PlusLabelName: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}

View file

@ -2224,7 +2224,7 @@ func TestGetTeamsYAMLAndApply(t *testing.T) {
require.ElementsMatch(t, labels, []string{fleet.BuiltinMacOS14PlusLabelName})
return map[string]uint{fleet.BuiltinMacOS14PlusLabelName: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}

View file

@ -179,7 +179,7 @@ func TestBasicTeamGitOps(t *testing.T) {
require.ElementsMatch(t, labels, []string{fleet.BuiltinMacOS14PlusLabelName})
return map[string]uint{fleet.BuiltinMacOS14PlusLabelName: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}
@ -437,7 +437,7 @@ func TestFullTeamGitOps(t *testing.T) {
require.ElementsMatch(t, labels, []string{fleet.BuiltinMacOS14PlusLabelName})
return map[string]uint{fleet.BuiltinMacOS14PlusLabelName: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}

View file

@ -135,7 +135,7 @@ func TestGetOrCreatePreassignTeam(t *testing.T) {
ds.GetMDMAppleSetupAssistantFuncInvoked = false
ds.SetOrUpdateMDMAppleSetupAssistantFuncInvoked = false
ds.LabelIDsByNameFuncInvoked = false
ds.NewMDMAppleDeclarationFuncInvoked = false
ds.SetOrUpdateMDMAppleDeclarationFuncInvoked = false
ds.BulkSetPendingMDMHostProfilesFuncInvoked = false
}
setupDS := func(t *testing.T) {
@ -192,7 +192,7 @@ func TestGetOrCreatePreassignTeam(t *testing.T) {
require.ElementsMatch(t, names, []string{fleet.BuiltinMacOS14PlusLabelName})
return map[string]uint{names[0]: 1}, nil
}
ds.NewMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
ds.SetOrUpdateMDMAppleDeclarationFunc = func(ctx context.Context, declaration *fleet.MDMAppleDeclaration) (*fleet.MDMAppleDeclaration, error) {
declaration.DeclarationUUID = uuid.NewString()
return declaration, nil
}