Fix ReplaceExtension

ReplaceExtension had an unexpected behaviour when the file did not have
an extension. In certain cases, the final path would be severely
trimmed: out/.intermediates/my_file would become out/.new_extension.
Explicitly handle the case by appending the new extension.

Test: Run checkbuild on Android Soong
Change-Id: Ie27a98845894cfaee5af5e2a02d44168c40ed821

This is an imported pull request from
https://github.com/google/blueprint/pull/345

GitOrigin-RevId: f9166c0e6151499b4b1a23b89b0bc133203a1116
Change-Id: I63f0798177545792440b8a84b04f1090590f1642
This commit is contained in:
Thiébaud Weksteen 2021-02-15 18:54:26 +00:00 committed by Colin Cross
parent e980b25b95
commit 8ec690764e
3 changed files with 48 additions and 4 deletions

View file

@ -70,6 +70,7 @@ bootstrap_go_package {
testSrcs: [
"pathtools/fs_test.go",
"pathtools/glob_test.go",
"pathtools/lists_test.go",
],
}

View file

@ -38,10 +38,12 @@ func ReplaceExtensions(paths []string, extension string) []string {
return result
}
// ReplaceExtension changes the file extension. If the file does not have an
// extension, the new extension is appended.
func ReplaceExtension(path string, extension string) string {
dot := strings.LastIndex(path, ".")
if dot == -1 {
return path
oldExt := filepath.Ext(path)
if oldExt != "" {
path = strings.TrimSuffix(path, oldExt)
}
return path[:dot+1] + extension
return path + "." + extension
}

41
pathtools/lists_test.go Normal file
View file

@ -0,0 +1,41 @@
// Copyright 2021 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 pathtools
import (
"testing"
)
func TestLists_ReplaceExtension(t *testing.T) {
testCases := []struct {
from, ext, to string
}{
{"1.jpg", "png", "1.png"},
{"1", "png", "1.png"},
{"1.", "png", "1.png"},
{"2.so", "so.1", "2.so.1"},
{"/out/.test/1.png", "jpg", "/out/.test/1.jpg"},
{"/out/.test/1", "jpg", "/out/.test/1.jpg"},
}
for _, test := range testCases {
t.Run(test.from, func(t *testing.T) {
got := ReplaceExtension(test.from, test.ext)
if got != test.to {
t.Errorf("ReplaceExtension(%v, %v) = %v; want: %v", test.from, test.ext, got, test.to)
}
})
}
}