| // Copyright 2020 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 bp2build | 
 |  | 
 | import ( | 
 | 	"sort" | 
 | 	"testing" | 
 | ) | 
 |  | 
 | type bazelFilepath struct { | 
 | 	dir      string | 
 | 	basename string | 
 | } | 
 |  | 
 | func TestCreateBazelFiles_QueryView_AddsTopLevelFiles(t *testing.T) { | 
 | 	files := CreateBazelFiles(map[string]RuleShim{}, map[string]BazelTargets{}, QueryView) | 
 | 	expectedFilePaths := []bazelFilepath{ | 
 | 		{ | 
 | 			dir:      "", | 
 | 			basename: "BUILD.bazel", | 
 | 		}, | 
 | 		{ | 
 | 			dir:      "", | 
 | 			basename: "WORKSPACE", | 
 | 		}, | 
 | 		{ | 
 | 			dir:      bazelRulesSubDir, | 
 | 			basename: "BUILD.bazel", | 
 | 		}, | 
 | 		{ | 
 | 			dir:      bazelRulesSubDir, | 
 | 			basename: "providers.bzl", | 
 | 		}, | 
 | 		{ | 
 | 			dir:      bazelRulesSubDir, | 
 | 			basename: "soong_module.bzl", | 
 | 		}, | 
 | 	} | 
 |  | 
 | 	// Compare number of files | 
 | 	if a, e := len(files), len(expectedFilePaths); a != e { | 
 | 		t.Errorf("Expected %d files, got %d", e, a) | 
 | 	} | 
 |  | 
 | 	// Sort the files to be deterministic | 
 | 	sort.Slice(files, func(i, j int) bool { | 
 | 		if dir1, dir2 := files[i].Dir, files[j].Dir; dir1 == dir2 { | 
 | 			return files[i].Basename < files[j].Basename | 
 | 		} else { | 
 | 			return dir1 < dir2 | 
 | 		} | 
 | 	}) | 
 |  | 
 | 	// Compare the file contents | 
 | 	for i := range files { | 
 | 		actualFile, expectedFile := files[i], expectedFilePaths[i] | 
 |  | 
 | 		if actualFile.Dir != expectedFile.dir || actualFile.Basename != expectedFile.basename { | 
 | 			t.Errorf("Did not find expected file %s/%s", actualFile.Dir, actualFile.Basename) | 
 | 		} else if actualFile.Basename == "BUILD.bazel" || actualFile.Basename == "WORKSPACE" { | 
 | 			if actualFile.Contents != "" { | 
 | 				t.Errorf("Expected %s to have no content.", actualFile) | 
 | 			} | 
 | 		} else if actualFile.Contents == "" { | 
 | 			t.Errorf("Contents of %s unexpected empty.", actualFile) | 
 | 		} | 
 | 	} | 
 | } |