tests/src/com/android/exchange/eas/EasProvisionTests.java

/*
 * 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));
    }

}