mirror of
				https://github.com/NVIDIA/nvidia-container-toolkit
				synced 2025-06-26 18:18:24 +00:00 
			
		
		
		
	Add tests for executable locator
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
		
							parent
							
								
									55eb898186
								
							
						
					
					
						commit
						8da7e74408
					
				| @ -33,6 +33,10 @@ type executable struct { | |||||||
| func NewExecutableLocator(logger *log.Logger, root string) Locator { | func NewExecutableLocator(logger *log.Logger, root string) Locator { | ||||||
| 	paths := GetPaths(root) | 	paths := GetPaths(root) | ||||||
| 
 | 
 | ||||||
|  | 	return newExecutableLocator(logger, root, paths...) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func newExecutableLocator(logger *log.Logger, root string, paths ...string) *executable { | ||||||
| 	var prefixes []string | 	var prefixes []string | ||||||
| 	for _, dir := range paths { | 	for _, dir := range paths { | ||||||
| 		prefixes = append(prefixes, filepath.Join(root, dir)) | 		prefixes = append(prefixes, filepath.Join(root, dir)) | ||||||
| @ -44,6 +48,7 @@ func NewExecutableLocator(logger *log.Logger, root string) Locator { | |||||||
| 			filter:   assertExecutable, | 			filter:   assertExecutable, | ||||||
| 		}, | 		}, | ||||||
| 	} | 	} | ||||||
|  | 
 | ||||||
| 	return &l | 	return &l | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | |||||||
							
								
								
									
										77
									
								
								internal/lookup/executable_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										77
									
								
								internal/lookup/executable_test.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,77 @@ | |||||||
|  | /** | ||||||
|  | # Copyright (c) 2022, NVIDIA CORPORATION.  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 lookup | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||||
|  | 	"github.com/stretchr/testify/require" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | func TestExecutableLocator(t *testing.T) { | ||||||
|  | 	logger, _ := testlog.NewNullLogger() | ||||||
|  | 
 | ||||||
|  | 	testCases := []struct { | ||||||
|  | 		root             string | ||||||
|  | 		paths            []string | ||||||
|  | 		expectedPrefixes []string | ||||||
|  | 	}{ | ||||||
|  | 		{ | ||||||
|  | 			root:             "", | ||||||
|  | 			expectedPrefixes: nil, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			root:             "", | ||||||
|  | 			paths:            []string{"/"}, | ||||||
|  | 			expectedPrefixes: []string{"/"}, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			root:             "", | ||||||
|  | 			paths:            []string{"/", "/bin"}, | ||||||
|  | 			expectedPrefixes: []string{"/", "/bin"}, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			root:             "/", | ||||||
|  | 			expectedPrefixes: nil, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			root:             "/", | ||||||
|  | 			paths:            []string{"/"}, | ||||||
|  | 			expectedPrefixes: []string{"/"}, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			root:             "/", | ||||||
|  | 			paths:            []string{"/", "/bin"}, | ||||||
|  | 			expectedPrefixes: []string{"/", "/bin"}, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			root:             "/some/path", | ||||||
|  | 			paths:            []string{"/", "/bin"}, | ||||||
|  | 			expectedPrefixes: []string{"/some/path", "/some/path/bin"}, | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	for i, tc := range testCases { | ||||||
|  | 		t.Run(fmt.Sprintf("%d", i), func(t *testing.T) { | ||||||
|  | 			e := newExecutableLocator(logger, tc.root, tc.paths...) | ||||||
|  | 
 | ||||||
|  | 			require.EqualValues(t, tc.expectedPrefixes, e.prefixes) | ||||||
|  | 		}) | ||||||
|  | 	} | ||||||
|  | } | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user