classifier/controllers/export_test.go at 8f70168cfb568a33739c3fead1f0aca09b38276e · gianlucam76/classifier · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
Copyright 2022. projectsveltos.io. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package controllers

import (
	corev1 "k8s.io/api/core/v1"

	libsveltosv1beta1 "github.com/projectsveltos/libsveltos/api/v1beta1"
)

// GroupClassifierReportsByCluster is the test-facing wrapper around groupClassifierReportsByCluster.
// It accepts a plain cluster list (instead of the internal shard map) and returns a map from
// cluster ref to its reports so tests don't need to know about internal types.
func GroupClassifierReportsByCluster(
	reports []libsveltosv1beta1.ClassifierReport,
	clusterList []corev1.ObjectReference,
) map[corev1.ObjectReference][]*libsveltosv1beta1.ClassifierReport {

	groups := groupClassifierReportsByCluster(reports, buildShardClustersMap(clusterList))
	result := make(map[corev1.ObjectReference][]*libsveltosv1beta1.ClassifierReport, len(groups))
	for _, g := range groups {
		result[g.ref] = g.items
	}
	return result
}

var (
	DeployClassifierCRD                     = deployClassifierCRD
	DeployClassifierReportCRD               = deployClassifierReportCRD
	DeployHealthCheckCRD                    = deployHealthCheckCRD
	DeployHealthCheckReportCRD              = deployHealthCheckReportCRD
	DeployEventSourceCRD                    = deployEventSourceCRD
	DeployEventReportCRD                    = deployEventReportCRD
	DeployReloaderCRD                       = deployReloaderCRD
	DeployReloaderReportCRD                 = deployReloaderReportCRD
	DeployDebuggingConfigurationCRD         = deployDebuggingConfigurationCRD
	DeployClassifierInstance                = deployClassifierInstance
	DeploySveltosAgentInManagedCluster      = deploySveltosAgentInManagedCluster
	ClassifierHash                          = classifierHash
	DeployClassifierInCluster               = deployClassifierInCluster
	UndeployClassifierFromCluster           = undeployClassifierFromCluster
	RemoveClassifierReports                 = removeClassifierReports
	RemoveClusterClassifierReports          = removeClusterClassifierReports
	CollectClassifierReportsFromCluster     = collectClassifierReportsFromCluster
	DeploySveltosAgentInManagementCluster   = deploySveltosAgentInManagementCluster
	RemoveSveltosAgentFromManagementCluster = removeSveltosAgentFromManagementCluster
	GetSveltosAgentLabels                   = getSveltosAgentLabels
	GetSveltosAgentNamespace                = getSveltosAgentNamespace
	GetSveltosAgentPatches                  = getSveltosAgentPatches
	GetSveltosApplierPatches                = getSveltosApplierPatches

	CreateAccessRequest                        = createAccessRequest
	GetAccessRequestName                       = getAccessRequestName
	GetKubeconfigFromAccessRequest             = getKubeconfigFromAccessRequest
	UpdateSecretWithAccessManagementKubeconfig = updateSecretWithAccessManagementKubeconfig

	GetHandlersForFeature = getHandlersForFeature

	ProcessClassifier                      = (*ClassifierReconciler).processClassifier
	RemoveClassifier                       = (*ClassifierReconciler).removeClassifier
	RequeueClassifierForCluster            = (*ClassifierReconciler).requeueClassifierForCluster
	RequeueClassifierForClassifierReport   = (*ClassifierReconciler).requeueClassifierForClassifierReport
	UpdateMatchingClustersAndRegistrations = (*ClassifierReconciler).updateMatchingClustersAndRegistrations
	UpdateLabelsOnMatchingClusters         = (*ClassifierReconciler).updateLabelsOnMatchingClusters
	RegisterMatchingClusters               = (*ClassifierReconciler).registerMatchingClusters
	UndeployClassifier                     = (*ClassifierReconciler).undeployClassifier
	RemoveAllRegistrations                 = (*ClassifierReconciler).removeAllRegistrations
	ClassifyLabels                         = (*ClassifierReconciler).classifyLabels
	CleanUpNonMatchingClusters             = (*ClassifierReconciler).cleanUpNonMatchingClusters
)

var (
	CreatFeatureHandlerMaps = creatFeatureHandlerMaps
)

const (
	Controlplaneendpoint = controlplaneendpoint
)

// ManagementClusterClassifier exports for unit tests.
var (
	DoesMatchLabelFilters      = doesMatchLabelFilters
	RunClassificationLua       = runClassificationLua
	ClusterTypeFromKind        = clusterTypeFromKind
	MgmtClassifierAsClassifier = mgmtClassifierAsClassifier
	EnsureMgmtClassifierReport = ensureMgmtClassifierReport
	ClassifierLabelKeys        = classifierLabelKeys
	FetchResourcesForSelector  = fetchResourcesForSelector
	ListMgmtClassifierReports  = listMgmtClassifierReports
	GetMgmtClassifierReport    = getMgmtClassifierReport
	DeleteMgmtClassifierReport = deleteMgmtClassifierReport
	ApplyLabelsToCluster       = applyLabelsToCluster
	RemoveLabelsFromCluster    = removeLabelsFromCluster
)