mirror of
				https://github.com/optim-enterprises-bv/kubernetes.git
				synced 2025-11-03 19:58:17 +00:00 
			
		
		
		
	refactor scheduler factory to use plugin architecture style like credentialprovider and cloudprovider for configuring priority functions and fit predicates
This commit is contained in:
		
							
								
								
									
										65
									
								
								plugin/pkg/scheduler/algorithmprovider/plugins_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										65
									
								
								plugin/pkg/scheduler/algorithmprovider/plugins_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,65 @@
 | 
			
		||||
/*
 | 
			
		||||
Copyright 2014 Google Inc. 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 algorithmprovider
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	"github.com/GoogleCloudPlatform/kubernetes/plugin/pkg/scheduler/factory"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	algorithmProviderNames = []string{
 | 
			
		||||
		factory.DefaultProvider,
 | 
			
		||||
	}
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func TestDefaultConfigExists(t *testing.T) {
 | 
			
		||||
	p, err := factory.GetAlgorithmProvider(factory.DefaultProvider)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Errorf("error retrivieving default provider: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	if p == nil {
 | 
			
		||||
		t.Error("algorithm provider config should not be nil")
 | 
			
		||||
	}
 | 
			
		||||
	if len(p.FitPredicateKeys) == 0 {
 | 
			
		||||
		t.Error("default algorithm provider shouldn't have 0 fit predicates")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestAlgorithmProviders(t *testing.T) {
 | 
			
		||||
	for _, pn := range algorithmProviderNames {
 | 
			
		||||
		p, err := factory.GetAlgorithmProvider(pn)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			t.Errorf("error retrivieving '%s' provider: %v", pn, err)
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		if len(p.PriorityFunctionKeys) == 0 {
 | 
			
		||||
			t.Error("%s algorithm provider shouldn't have 0 priority functions", pn)
 | 
			
		||||
		}
 | 
			
		||||
		for _, pf := range p.PriorityFunctionKeys.List() {
 | 
			
		||||
			if !factory.IsPriorityFunctionRegistered(pf) {
 | 
			
		||||
				t.Errorf("priority function %s is not registerd but is used in the %s algorithm provider", pf, pn)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		for _, fp := range p.FitPredicateKeys.List() {
 | 
			
		||||
			if !factory.IsFitPredicateRegistered(fp) {
 | 
			
		||||
				t.Errorf("fit predicate %s is not registerd but is used in the %s algorithm provider", fp, pn)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user