| /* |
| * Copyright (C) 2019 The Android Open Source Project |
| * |
| * 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. |
| */ |
| |
| @file:JvmName("ParcelUtils") |
| |
| package com.android.testutils |
| |
| import android.os.Parcel |
| import android.os.Parcelable |
| import kotlin.test.assertTrue |
| import kotlin.test.fail |
| |
| /** |
| * Return a new instance of `T` after being parceled then unparceled. |
| */ |
| fun <T : Parcelable> parcelingRoundTrip(source: T): T { |
| val creator: Parcelable.Creator<T> |
| try { |
| creator = source.javaClass.getField("CREATOR").get(null) as Parcelable.Creator<T> |
| } catch (e: IllegalAccessException) { |
| fail("Missing CREATOR field: " + e.message) |
| } catch (e: NoSuchFieldException) { |
| fail("Missing CREATOR field: " + e.message) |
| } |
| |
| var p = Parcel.obtain() |
| source.writeToParcel(p, /* flags */ 0) |
| p.setDataPosition(0) |
| val marshalled = p.marshall() |
| p = Parcel.obtain() |
| p.unmarshall(marshalled, 0, marshalled.size) |
| p.setDataPosition(0) |
| return creator.createFromParcel(p) |
| } |
| |
| /** |
| * Assert that after being parceled then unparceled, `source` is equal to the original |
| * object. If a customized equals function is provided, uses the provided one. |
| */ |
| @JvmOverloads |
| fun <T : Parcelable> assertParcelingIsLossless( |
| source: T, |
| equals: (T, T) -> Boolean = { a, b -> a == b } |
| ) { |
| val actual = parcelingRoundTrip(source) |
| assertTrue(equals(source, actual), "Expected $source, but was $actual") |
| } |
| |
| @JvmOverloads |
| fun <T : Parcelable> assertParcelSane( |
| obj: T, |
| fieldCount: Int, |
| equals: (T, T) -> Boolean = { a, b -> a == b } |
| ) { |
| assertFieldCountEquals(fieldCount, obj::class.java) |
| assertParcelingIsLossless(obj, equals) |
| } |