xref: /aosp_15_r20/build/soong/rust/benchmark_test.go (revision 333d2b3687b3a337dbcca9d65000bca186795e39)
1 // Copyright 2021 The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //     http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 package rust
16 
17 import (
18 	"strings"
19 	"testing"
20 
21 	"android/soong/android"
22 )
23 
24 func TestRustBenchmark(t *testing.T) {
25 	ctx := testRust(t, `
26 		rust_benchmark_host {
27 			name: "my_bench",
28 			srcs: ["foo.rs"],
29 		}`)
30 
31 	testingModule := ctx.ModuleForTests("my_bench", "linux_glibc_x86_64")
32 	expectedOut := "my_bench/linux_glibc_x86_64/my_bench"
33 	outPath := testingModule.Output("my_bench").Output.String()
34 	if !strings.Contains(outPath, expectedOut) {
35 		t.Errorf("wrong output path: %v;  expected: %v", outPath, expectedOut)
36 	}
37 }
38 
39 func TestRustBenchmarkLinkage(t *testing.T) {
40 	ctx := testRust(t, `
41 		rust_benchmark {
42 			name: "my_bench",
43 			srcs: ["foo.rs"],
44 		}`)
45 
46 	testingModule := ctx.ModuleForTests("my_bench", "android_arm64_armv8-a").Module().(*Module)
47 
48 	if !android.InList("libcriterion.rlib-std", testingModule.Properties.AndroidMkRlibs) {
49 		t.Errorf("rlib-std variant for libcriterion not detected as a rustlib-defined rlib dependency for device rust_benchmark module")
50 	}
51 	if !android.InList("libstd", testingModule.Properties.AndroidMkRlibs) {
52 		t.Errorf("Device rust_benchmark module 'my_bench' does not link libstd as an rlib")
53 	}
54 }
55