123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885 |
- // 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 main
- import (
- "bytes"
- "fmt"
- "reflect"
- "testing"
- "google.golang.org/protobuf/encoding/prototext"
- bp "android/soong/cmd/extract_apks/bundle_proto"
- "android/soong/third_party/zip"
- )
- type testConfigDesc struct {
- name string
- targetConfig TargetConfig
- expected SelectionResult
- }
- type testDesc struct {
- protoText string
- configs []testConfigDesc
- }
- func TestSelectApks_ApkSet(t *testing.T) {
- testCases := []testDesc{
- {
- protoText: `
- variant {
- targeting {
- sdk_version_targeting {
- value { min { value: 29 } } } }
- apk_set {
- module_metadata {
- name: "base" targeting {} delivery_type: INSTALL_TIME }
- apk_description {
- targeting {
- screen_density_targeting {
- value { density_alias: LDPI } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-ldpi.apk"
- split_apk_metadata { split_id: "config.ldpi" } }
- apk_description {
- targeting {
- screen_density_targeting {
- value { density_alias: MDPI } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-mdpi.apk"
- split_apk_metadata { split_id: "config.mdpi" } }
- apk_description {
- targeting {
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-master.apk"
- split_apk_metadata { is_master_split: true } }
- apk_description {
- targeting {
- abi_targeting {
- value { alias: ARMEABI_V7A }
- alternatives { alias: ARM64_V8A }
- alternatives { alias: X86 }
- alternatives { alias: X86_64 } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-armeabi_v7a.apk"
- split_apk_metadata { split_id: "config.armeabi_v7a" } }
- apk_description {
- targeting {
- abi_targeting {
- value { alias: ARM64_V8A }
- alternatives { alias: ARMEABI_V7A }
- alternatives { alias: X86 }
- alternatives { alias: X86_64 } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-arm64_v8a.apk"
- split_apk_metadata { split_id: "config.arm64_v8a" } }
- apk_description {
- targeting {
- abi_targeting {
- value { alias: X86 }
- alternatives { alias: ARMEABI_V7A }
- alternatives { alias: ARM64_V8A }
- alternatives { alias: X86_64 } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-x86.apk"
- split_apk_metadata { split_id: "config.x86" } }
- apk_description {
- targeting {
- abi_targeting {
- value { alias: X86_64 }
- alternatives { alias: ARMEABI_V7A }
- alternatives { alias: ARM64_V8A }
- alternatives { alias: X86 } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-x86_64.apk"
- split_apk_metadata { split_id: "config.x86_64" } } }
- }
- bundletool {
- version: "0.10.3" }
- `,
- configs: []testConfigDesc{
- {
- name: "one",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARMEABI_V7A: 0,
- bp.Abi_ARM64_V8A: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "splits/base-ldpi.apk",
- "splits/base-mdpi.apk",
- "splits/base-master.apk",
- "splits/base-armeabi_v7a.apk",
- },
- },
- },
- {
- name: "two",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_LDPI: true,
- },
- abis: map[bp.Abi_AbiAlias]int{},
- },
- expected: SelectionResult{
- "base",
- []string{
- "splits/base-ldpi.apk",
- "splits/base-master.apk",
- },
- },
- },
- {
- name: "three",
- targetConfig: TargetConfig{
- sdkVersion: 20,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_LDPI: true,
- },
- abis: map[bp.Abi_AbiAlias]int{},
- },
- expected: SelectionResult{
- "",
- nil,
- },
- },
- {
- name: "four",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_MDPI: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARM64_V8A: 0,
- bp.Abi_ARMEABI_V7A: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "splits/base-mdpi.apk",
- "splits/base-master.apk",
- "splits/base-arm64_v8a.apk",
- },
- },
- },
- },
- },
- {
- protoText: `
- variant {
- targeting {
- sdk_version_targeting {
- value { min { value: 10000 } } } }
- apk_set {
- module_metadata {
- name: "base" targeting {} delivery_type: INSTALL_TIME }
- apk_description {
- targeting {
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "splits/base-master.apk"
- split_apk_metadata { is_master_split: true } } } }`,
- configs: []testConfigDesc{
- {
- name: "Prerelease",
- targetConfig: TargetConfig{
- sdkVersion: 30,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{},
- abis: map[bp.Abi_AbiAlias]int{},
- allowPrereleased: true,
- },
- expected: SelectionResult{
- "base",
- []string{"splits/base-master.apk"},
- },
- },
- },
- },
- {
- protoText: `
- variant {
- targeting {
- sdk_version_targeting {
- value { min { value: 29 } } } }
- apk_set {
- module_metadata {
- name: "base" targeting {} delivery_type: INSTALL_TIME }
- apk_description {
- targeting {}
- path: "universal.apk"
- standalone_apk_metadata { fused_module_name: "base" } } } }`,
- configs: []testConfigDesc{
- {
- name: "Universal",
- targetConfig: TargetConfig{sdkVersion: 30},
- expected: SelectionResult{
- "base",
- []string{"universal.apk"},
- },
- },
- },
- },
- }
- for _, testCase := range testCases {
- var toc bp.BuildApksResult
- if err := prototext.Unmarshal([]byte(testCase.protoText), &toc); err != nil {
- t.Fatal(err)
- }
- for _, config := range testCase.configs {
- actual := selectApks(&toc, config.targetConfig)
- if !reflect.DeepEqual(config.expected, actual) {
- t.Errorf("%s: expected %v, got %v", config.name, config.expected, actual)
- }
- }
- }
- }
- func TestSelectApks_ApexSet(t *testing.T) {
- testCases := []testDesc{
- {
- protoText: `
- variant {
- targeting {
- sdk_version_targeting {
- value { min { value: 29 } } } }
- apk_set {
- module_metadata {
- name: "base" targeting {} delivery_type: INSTALL_TIME }
- apk_description {
- targeting {
- multi_abi_targeting {
- value { abi { alias: ARMEABI_V7A } }
- alternatives { abi { alias: ARM64_V8A } }
- alternatives { abi { alias: X86 } }
- alternatives { abi { alias: X86_64 } } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "standalones/standalone-armeabi_v7a.apex"
- apex_apk_metadata { } }
- apk_description {
- targeting {
- multi_abi_targeting {
- value { abi { alias: ARM64_V8A } }
- alternatives { abi { alias: ARMEABI_V7A } }
- alternatives { abi { alias: X86 } }
- alternatives { abi { alias: X86_64 } } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "standalones/standalone-arm64_v8a.apex"
- apex_apk_metadata { } }
- apk_description {
- targeting {
- multi_abi_targeting {
- value { abi { alias: X86 } }
- alternatives { abi { alias: ARMEABI_V7A } }
- alternatives { abi { alias: ARM64_V8A } }
- alternatives { abi { alias: X86_64 } } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "standalones/standalone-x86.apex"
- apex_apk_metadata { } }
- apk_description {
- targeting {
- multi_abi_targeting {
- value { abi { alias: X86_64 } }
- alternatives { abi { alias: ARMEABI_V7A } }
- alternatives { abi { alias: ARM64_V8A } }
- alternatives { abi { alias: X86 } } }
- sdk_version_targeting {
- value { min { value: 21 } } } }
- path: "standalones/standalone-x86_64.apex"
- apex_apk_metadata { } } }
- }
- bundletool {
- version: "0.10.3" }
- `,
- configs: []testConfigDesc{
- {
- name: "order matches priorities",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARM64_V8A: 0,
- bp.Abi_ARMEABI_V7A: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-arm64_v8a.apex",
- },
- },
- },
- {
- name: "order doesn't match priorities",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARMEABI_V7A: 0,
- bp.Abi_ARM64_V8A: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-arm64_v8a.apex",
- },
- },
- },
- {
- name: "single choice",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARMEABI_V7A: 0,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-armeabi_v7a.apex",
- },
- },
- },
- {
- name: "cross platform",
- targetConfig: TargetConfig{
- sdkVersion: 29,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARM64_V8A: 0,
- bp.Abi_MIPS64: 1,
- bp.Abi_X86: 2,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-x86.apex",
- },
- },
- },
- },
- },
- }
- for _, testCase := range testCases {
- var toc bp.BuildApksResult
- if err := prototext.Unmarshal([]byte(testCase.protoText), &toc); err != nil {
- t.Fatal(err)
- }
- for _, config := range testCase.configs {
- actual := selectApks(&toc, config.targetConfig)
- if !reflect.DeepEqual(config.expected, actual) {
- t.Errorf("%s: expected %v, got %v", config.name, config.expected, actual)
- }
- }
- }
- }
- func TestSelectApks_ApexSet_Variants(t *testing.T) {
- testCases := []testDesc{
- {
- protoText: `
- variant {
- targeting {
- sdk_version_targeting {value {min {value: 29}}}
- multi_abi_targeting {
- value {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: X86}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- apk_set {
- module_metadata {
- name: "base"
- delivery_type: INSTALL_TIME
- }
- apk_description {
- targeting {
- multi_abi_targeting {
- value {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: X86}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- path: "standalones/standalone-armeabi_v7a.apex"
- }
- }
- variant_number: 0
- }
- variant {
- targeting {
- sdk_version_targeting {value {min {value: 29}}}
- multi_abi_targeting {
- value {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: X86}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- apk_set {
- module_metadata {
- name: "base"
- delivery_type: INSTALL_TIME
- }
- apk_description {
- targeting {
- multi_abi_targeting {
- value {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: X86}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- path: "standalones/standalone-arm64_v8a.apex"
- }
- }
- variant_number: 1
- }
- variant {
- targeting {
- sdk_version_targeting {value {min {value: 29}}}
- multi_abi_targeting {
- value {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: X86}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- apk_set {
- module_metadata {
- name: "base"
- delivery_type: INSTALL_TIME
- }
- apk_description {
- targeting {
- multi_abi_targeting {
- value {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: X86}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- path: "standalones/standalone-armeabi_v7a.arm64_v8a.apex"
- }
- }
- variant_number: 2
- }
- variant {
- targeting {
- sdk_version_targeting {value {min {value: 29}}}
- multi_abi_targeting {
- value {abi {alias: X86}}
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- apk_set {
- module_metadata {
- name: "base"
- delivery_type: INSTALL_TIME
- }
- apk_description {
- targeting {
- multi_abi_targeting {
- value {abi {alias: X86}}
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- }
- }
- path: "standalones/standalone-x86.apex"
- }
- }
- variant_number: 3
- }
- variant {
- targeting {
- sdk_version_targeting {value {min {value: 29}}}
- multi_abi_targeting {
- value {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: X86}}
- }
- }
- apk_set {
- module_metadata {
- name: "base"
- delivery_type: INSTALL_TIME
- }
- apk_description {
- targeting {
- multi_abi_targeting {
- value {
- abi {alias: X86}
- abi {alias: X86_64}
- }
- alternatives {abi {alias: ARMEABI_V7A}}
- alternatives {
- abi {alias: ARMEABI_V7A}
- abi {alias: ARM64_V8A}
- }
- alternatives {abi {alias: ARM64_V8A}}
- alternatives {abi {alias: X86}}
- }
- }
- path: "standalones/standalone-x86.x86_64.apex"
- }
- }
- variant_number: 4
- }
- `,
- configs: []testConfigDesc{
- {
- name: "multi-variant multi-target ARM",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARM64_V8A: 0,
- bp.Abi_ARMEABI_V7A: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-armeabi_v7a.arm64_v8a.apex",
- },
- },
- },
- {
- name: "multi-variant single-target arm",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARMEABI_V7A: 0,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-armeabi_v7a.apex",
- },
- },
- },
- {
- name: "multi-variant single-target arm64",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARM64_V8A: 0,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-arm64_v8a.apex",
- },
- },
- },
- {
- name: "multi-variant multi-target x86",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_X86: 0,
- bp.Abi_X86_64: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-x86.x86_64.apex",
- },
- },
- },
- {
- name: "multi-variant single-target x86",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_X86: 0,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-x86.apex",
- },
- },
- },
- {
- name: "multi-variant single-target x86_64",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_X86_64: 0,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-x86.x86_64.apex",
- }},
- },
- {
- name: "multi-variant multi-target cross-target",
- targetConfig: TargetConfig{
- sdkVersion: 33,
- screenDpi: map[bp.ScreenDensity_DensityAlias]bool{
- bp.ScreenDensity_DENSITY_UNSPECIFIED: true,
- },
- abis: map[bp.Abi_AbiAlias]int{
- bp.Abi_ARM64_V8A: 0,
- bp.Abi_X86_64: 1,
- },
- },
- expected: SelectionResult{
- "base",
- []string{
- "standalones/standalone-arm64_v8a.apex",
- },
- },
- },
- },
- },
- }
- for _, testCase := range testCases {
- var toc bp.BuildApksResult
- if err := prototext.Unmarshal([]byte(testCase.protoText), &toc); err != nil {
- t.Fatal(err)
- }
- for _, config := range testCase.configs {
- t.Run(config.name, func(t *testing.T) {
- actual := selectApks(&toc, config.targetConfig)
- if !reflect.DeepEqual(config.expected, actual) {
- t.Errorf("expected %v, got %v", config.expected, actual)
- }
- })
- }
- }
- }
- type testZip2ZipWriter struct {
- entries map[string]string
- }
- func (w testZip2ZipWriter) CopyFrom(file *zip.File, out string) error {
- if x, ok := w.entries[out]; ok {
- return fmt.Errorf("%s and %s both write to %s", x, file.Name, out)
- }
- w.entries[out] = file.Name
- return nil
- }
- type testCaseWriteApks struct {
- name string
- moduleName string
- stem string
- partition string
- // what we write from what
- zipEntries map[string]string
- expectedApkcerts []string
- }
- func TestWriteApks(t *testing.T) {
- testCases := []testCaseWriteApks{
- {
- name: "splits",
- moduleName: "mybase",
- stem: "Foo",
- partition: "system",
- zipEntries: map[string]string{
- "Foo.apk": "splits/mybase-master.apk",
- "Foo-xhdpi.apk": "splits/mybase-xhdpi.apk",
- },
- expectedApkcerts: []string{
- `name="Foo-xhdpi.apk" certificate="PRESIGNED" private_key="" partition="system"`,
- `name="Foo.apk" certificate="PRESIGNED" private_key="" partition="system"`,
- },
- },
- {
- name: "universal",
- moduleName: "base",
- stem: "Bar",
- partition: "product",
- zipEntries: map[string]string{
- "Bar.apk": "universal.apk",
- },
- expectedApkcerts: []string{
- `name="Bar.apk" certificate="PRESIGNED" private_key="" partition="product"`,
- },
- },
- }
- for _, testCase := range testCases {
- t.Run(testCase.name, func(t *testing.T) {
- testZipBuf := &bytes.Buffer{}
- testZip := zip.NewWriter(testZipBuf)
- for _, in := range testCase.zipEntries {
- f, _ := testZip.Create(in)
- f.Write([]byte(in))
- }
- testZip.Close()
- zipReader, _ := zip.NewReader(bytes.NewReader(testZipBuf.Bytes()), int64(testZipBuf.Len()))
- apkSet := ApkSet{entries: make(map[string]*zip.File)}
- sel := SelectionResult{moduleName: testCase.moduleName}
- for _, f := range zipReader.File {
- apkSet.entries[f.Name] = f
- sel.entries = append(sel.entries, f.Name)
- }
- zipWriter := testZip2ZipWriter{make(map[string]string)}
- outWriter := &bytes.Buffer{}
- config := TargetConfig{stem: testCase.stem}
- apkcerts, err := apkSet.writeApks(sel, config, outWriter, zipWriter, testCase.partition)
- if err != nil {
- t.Error(err)
- }
- expectedZipEntries := make(map[string]string)
- for k, v := range testCase.zipEntries {
- if k != testCase.stem+".apk" {
- expectedZipEntries[k] = v
- }
- }
- if !reflect.DeepEqual(expectedZipEntries, zipWriter.entries) {
- t.Errorf("expected zip entries %v, got %v", testCase.zipEntries, zipWriter.entries)
- }
- if !reflect.DeepEqual(testCase.expectedApkcerts, apkcerts) {
- t.Errorf("expected apkcerts %v, got %v", testCase.expectedApkcerts, apkcerts)
- }
- if g, w := outWriter.String(), testCase.zipEntries[testCase.stem+".apk"]; !reflect.DeepEqual(g, w) {
- t.Errorf("expected output file contents %q, got %q", testCase.stem+".apk", outWriter.String())
- }
- })
- }
- }
|