|  | // Copyright 2018 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 android | 
|  |  | 
|  | import ( | 
|  | "testing" | 
|  | ) | 
|  |  | 
|  | func testVtsConfig(test *testing.T, bpFileContents string) *TestContext { | 
|  | config := TestArchConfig(buildDir, nil) | 
|  |  | 
|  | ctx := NewTestArchContext() | 
|  | ctx.RegisterModuleType("vts_config", ModuleFactoryAdaptor(VtsConfigFactory)) | 
|  | ctx.Register() | 
|  | mockFiles := map[string][]byte{ | 
|  | "Android.bp": []byte(bpFileContents), | 
|  | } | 
|  | ctx.MockFileSystem(mockFiles) | 
|  | _, errs := ctx.ParseFileList(".", []string{"Android.bp"}) | 
|  | FailIfErrored(test, errs) | 
|  | _, errs = ctx.PrepareBuildActions(config) | 
|  | FailIfErrored(test, errs) | 
|  | return ctx | 
|  | } | 
|  |  | 
|  | func TestVtsConfig(t *testing.T) { | 
|  | ctx := testVtsConfig(t, ` | 
|  | vts_config { name: "plain"} | 
|  | vts_config { name: "with_manifest", test_config: "manifest.xml" } | 
|  | `) | 
|  |  | 
|  | variants := ctx.ModuleVariantsForTests("plain") | 
|  | if len(variants) > 1 { | 
|  | t.Errorf("expected 1, got %d", len(variants)) | 
|  | } | 
|  | expectedOutputFilename := ctx.ModuleForTests( | 
|  | "plain", variants[0]).Module().(*VtsConfig).OutputFilePath.Base() | 
|  | if expectedOutputFilename != "plain" { | 
|  | t.Errorf("expected plain, got %q", expectedOutputFilename) | 
|  | } | 
|  | } |