Do not send extraneous tags when confirming a remote wipe. b/12921830

This is an integration of an Exchange patch from Motorola.
getRequestEntity() was modified to not add tags that were
not part of the protocol spec for remote wipe ack.
Another big change was to unwrap this function so it could
be unit tested w/o the need for bigger, bulkier classes.
This lead to the creation of populateRequestEntitySerializer()
and changes to other functions to make then reentrant and
potentially static if possible. Oh...fixed a typo too.

Change-Id: I0c9167c9609b3d6a6f77f3f3118d9ae790077b2d
(cherry picked from commit 60b69ecd47d872c845b6bca46e4937d505a78101)

1 files changed, 161 insertions, 0 deletions

diff --git a/tests/src/com/android/exchange/eas/EasProvisionTests.java b/tests/src/com/android/exchange/eas/EasProvisionTests.java
new file mode 100644
index 00000000..7d042339
--- /dev/null
+++ b/tests/src/com/android/exchange/eas/EasProvisionTests.java
@@ -0,0 +1,161 @@
+/*
+ * Copyright (C) 2014 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.
+ */
+
+package com.android.exchange.eas;
+
+import android.content.Context;
+import android.os.Bundle;
+import android.telephony.TelephonyManager;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import com.android.emailcommon.provider.EmailContent;
+import com.android.exchange.adapter.Serializer;
+import com.android.exchange.Eas;
+import com.android.exchange.adapter.Tags;
+import com.android.exchange.utility.ExchangeTestCase;
+
+import java.io.IOException;
+import java.util.Arrays;
+
+/**
+ * You can run this entire test case with:
+ *   runtest -c com.android.exchange.eas.EasProvisionTests exchange
+ */
+@SmallTest
+public class EasProvisionTests extends ExchangeTestCase {
+
+    /**
+     * This test case will test PHASE_INITIAL along with a protocol version of Ex2007.
+     */
+    public void testPopulateRequestEntitySerializerPhaseInitialEx2007() throws IOException {
+        // Set up some parameters for the test case
+        final String policyType = "Test_Policy";
+        final String userAgent = "User_Agent";
+        final String status = "Test_Status";
+        final String policyKey = "Test_Policy_Key";
+        final int phase = EasProvision.PHASE_INITIAL;
+        final double protocolVersion = Eas.SUPPORTED_PROTOCOL_EX2007_DOUBLE;
+
+        // Build the result that we are expecting
+        final Serializer expectedResult = new Serializer();
+        expectedResult.start(Tags.PROVISION_PROVISION);
+        expectedResult.start(Tags.PROVISION_POLICIES);
+        expectedResult.start(Tags.PROVISION_POLICY);
+        expectedResult.data(Tags.PROVISION_POLICY_TYPE, policyType);
+        // PROVISION_POLICY, PROVISION_POLICIES, PROVISION_PROVISION
+        expectedResult.end().end().end().done();
+        final byte[] expectedBytes = expectedResult.toByteArray();
+
+        // Now run it through the code that we are testing
+        final Serializer generatedResult = EasProvision.generateRequestEntitySerializer(
+                mContext, userAgent, policyKey, policyType, status, phase, protocolVersion);
+
+        // Now let's analyze the results
+        assertTrue(Arrays.equals(generatedResult.toByteArray(), expectedBytes));
+    }
+
+    /**
+     * This test case will test PHASE_INITIAL along with a protocol version of Ex2010.
+     */
+    public void testPopulateRequestEntitySerializerPhaseInitialEx2010() throws IOException {
+        // Set up some parameters for the test case
+        final String policyType = "Test_Policy";
+        final String userAgent = "User_Agent";
+        final String status = "Test_Status";
+        final String policyKey = "Test_Policy_Key";
+        final int phase = EasProvision.PHASE_INITIAL;
+        final double protocolVersion = Eas.SUPPORTED_PROTOCOL_EX2010_SP1_DOUBLE;
+
+        // Build the result that we are expecting
+        final Serializer expectedResult = new Serializer();
+        expectedResult.start(Tags.PROVISION_PROVISION);
+        EasProvision.expandedAddDeviceInformationToSerializer(expectedResult, mContext, userAgent);
+        expectedResult.start(Tags.PROVISION_POLICIES);
+        expectedResult.start(Tags.PROVISION_POLICY);
+        expectedResult.data(Tags.PROVISION_POLICY_TYPE, policyType);
+        // PROVISION_POLICY, PROVISION_POLICIES, PROVISION_PROVISION
+        expectedResult.end().end().end().done();
+        final byte[] expectedBytes = expectedResult.toByteArray();
+
+        // Now run it through the code that we are testing
+        final Serializer generatedResult = EasProvision.generateRequestEntitySerializer(
+                mContext, userAgent, policyKey, policyType, status, phase, protocolVersion);
+
+        // Now let's analyze the results
+        assertTrue(Arrays.equals(generatedResult.toByteArray(), expectedBytes));
+    }
+
+    /**
+     * This test case will test PHASE_WIPE.
+     */
+    public void testPopulateRequestEntitySerializerPhaseWipe() throws IOException {
+        // Set up some parameters for the test case
+        final String policyType = "Test_Policy";
+        final String userAgent = "User_Agent";
+        final String status = "Test_Status";
+        final String policyKey = "Test_Policy_Key";
+        final int phase = EasProvision.PHASE_WIPE;
+        final double protocolVersion = Eas.SUPPORTED_PROTOCOL_EX2007_DOUBLE;
+
+        // Build the result that we are expecting
+        final Serializer expectedResult = new Serializer();
+        expectedResult.start(Tags.PROVISION_PROVISION);
+        expectedResult.start(Tags.PROVISION_REMOTE_WIPE);
+        expectedResult.data(Tags.PROVISION_STATUS, EasProvision.PROVISION_STATUS_OK);
+        expectedResult.end().end().done(); // PROVISION_REMOTE_WIPE, PROVISION_PROVISION
+        final byte[] expectedBytes = expectedResult.toByteArray();
+
+        // Now run it through the code that we are testing
+        final Serializer generatedResult = EasProvision.generateRequestEntitySerializer(
+                mContext, userAgent, policyKey, policyType, status, phase, protocolVersion);
+
+        // Now let's analyze the results
+        assertTrue(Arrays.equals(generatedResult.toByteArray(), expectedBytes));
+    }
+
+    /**
+     * This test case will test PHASE_ACKNOWLEDGE.
+     */
+    public void testPopulateRequestEntitySerializerPhaseAcknowledge() throws IOException {
+        // Set up some parameters for the test case
+        final String policyType = "Test_Policy";
+        final String userAgent = "User_Agent";
+        final String status = "Test_Status";
+        final String policyKey = "Test_Policy_Key";
+        final int phase = EasProvision.PHASE_ACKNOWLEDGE;
+        final double protocolVersion = Eas.SUPPORTED_PROTOCOL_EX2007_DOUBLE;
+
+        // Build the result that we are expecting
+        final Serializer expectedResult = new Serializer();
+        expectedResult.start(Tags.PROVISION_PROVISION);
+        expectedResult.start(Tags.PROVISION_POLICIES);
+        expectedResult.start(Tags.PROVISION_POLICY);
+        expectedResult.data(Tags.PROVISION_POLICY_TYPE, policyType);
+        expectedResult.data(Tags.PROVISION_POLICY_KEY, policyKey);
+        expectedResult.data(Tags.PROVISION_STATUS, status);
+        // PROVISION_POLICY, PROVISION_POLICIES, PROVISION_PROVISION
+        expectedResult.end().end().end().done();
+        final byte[] expectedBytes = expectedResult.toByteArray();
+
+        // Now run it through the code that we are testing
+        final Serializer generatedResult = EasProvision.generateRequestEntitySerializer(
+                mContext, userAgent, policyKey, policyType, status, phase, protocolVersion);
+
+        // Now let's analyze the results
+        assertTrue(Arrays.equals(generatedResult.toByteArray(), expectedBytes));
+    }
+
+}