From 0f705373480777635c331ed5ce3c04c2381704f3 Mon Sep 17 00:00:00 2001 From: paulhu Date: Fri, 13 May 2022 07:17:45 +0000 Subject: [PATCH] Add setter/getter methods test for DscpPolicy.Builder Bug: 220029526 Test: atest FrameworksNetTests CtsNetTestCases Change-Id: Iba8b1c178f04f3044a47a30085ead0ad36c8c7c8 (cherry picked from commit e849ea4dbea9267977d4a8ac53d367ea14e9c9c4) Merged-In: Iba8b1c178f04f3044a47a30085ead0ad36c8c7c8 --- .../net/src/android/net/cts/DscpPolicyTest.kt | 20 +++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/tests/cts/net/src/android/net/cts/DscpPolicyTest.kt b/tests/cts/net/src/android/net/cts/DscpPolicyTest.kt index 1e42fe6983..bbac09bb44 100644 --- a/tests/cts/net/src/android/net/cts/DscpPolicyTest.kt +++ b/tests/cts/net/src/android/net/cts/DscpPolicyTest.kt @@ -626,15 +626,31 @@ class DscpPolicyTest { @Test fun testParcelingDscpPolicyIsLossless(): Unit = createConnectedNetworkAgent().let { (agent, callback) -> + val policyId = 1 + val dscpValue = 1 + val range = Range(4444, 4444) + val srcPort = 555 + // Check that policy with partial parameters is lossless. - val policy = DscpPolicy.Builder(1, 1).setDestinationPortRange(Range(4444, 4444)).build() + val policy = DscpPolicy.Builder(policyId, dscpValue).setDestinationPortRange(range).build() + assertEquals(policyId, policy.policyId) + assertEquals(dscpValue, policy.dscpValue) + assertEquals(range, policy.destinationPortRange) assertParcelingIsLossless(policy) // Check that policy with all parameters is lossless. - val policy2 = DscpPolicy.Builder(1, 1).setDestinationPortRange(Range(4444, 4444)) + val policy2 = DscpPolicy.Builder(policyId, dscpValue).setDestinationPortRange(range) .setSourceAddress(LOCAL_IPV4_ADDRESS) .setDestinationAddress(TEST_TARGET_IPV4_ADDR) + .setSourcePort(srcPort) .setProtocol(IPPROTO_UDP).build() + assertEquals(policyId, policy2.policyId) + assertEquals(dscpValue, policy2.dscpValue) + assertEquals(range, policy2.destinationPortRange) + assertEquals(TEST_TARGET_IPV4_ADDR, policy2.destinationAddress) + assertEquals(LOCAL_IPV4_ADDRESS, policy2.sourceAddress) + assertEquals(srcPort, policy2.sourcePort) + assertEquals(IPPROTO_UDP, policy2.protocol) assertParcelingIsLossless(policy2) } }