ServerManagedPolicyTest.java

/*
 * Copyright (C) 2010 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.vending.licensing;


import com.example.android.market.licensing.MainActivity;
import com.google.android.vending.licensing.AESObfuscator;
import com.google.android.vending.licensing.Policy;
import com.google.android.vending.licensing.ResponseData;
import com.google.android.vending.licensing.ServerManagedPolicy;

import android.provider.Settings;
import android.test.ActivityInstrumentationTestCase2;

/**
 * Test suite for StrictPolicy.
 */
public class ServerManagedPolicyTest extends ActivityInstrumentationTestCase2<MainActivity> {

    ServerManagedPolicy p;

    public ServerManagedPolicyTest() {
        super("com.example.android.market.licensing", MainActivity.class);
    }

    public void setUp() {
        final byte[] SALT = new byte[] {
            104, -12, 112, 82, -85, -10, -11, 61, 15, 54, 44, -66, -117, -89, -64, 110, -53, 123, 33
        };

        String deviceId = Settings.Secure.getString(
                getActivity().getApplicationContext().getContentResolver(),
                Settings.Secure.ANDROID_ID);
        p = new ServerManagedPolicy(getActivity().getApplicationContext(),
                new AESObfuscator(SALT, getActivity().getPackageName(), deviceId));
    }

    /**
     * Verify that extra data is parsed correctly on a LICENSED resopnse..
     */
    public void testExtraDataParsed() {

        String sampleResponse = "0|1579380448|com.example.android.market.licensing|1|" +
                "ADf8I4ajjgc1P5ZI1S1DN/YIPIUNPECLrg==|1279578835423:VT=11>=22&GR=33";
        p.processServerResponse(Policy.LICENSED,
                ResponseData.parse(sampleResponse));
        assertEquals(11l, p.getValidityTimestamp());
        assertEquals(22l, p.getRetryUntil());
        assertEquals(33l, p.getMaxRetries());
    }

    /**
     * Verify that retry counts are cleared after getting a NOT_LICENSED response.
     */
    public void testRetryCountsCleared() {
        String sampleResponse = "0|1579380448|com.example.android.market.licensing|1|" +
                "ADf8I4ajjgc1P5ZI1S1DN/YIPIUNPECLrg==|1279578835423:VT=1>=2&GR=3";
        p.processServerResponse(Policy.LICENSED,
                ResponseData.parse(sampleResponse));
        // Sanity test
        assertTrue(0l != p.getValidityTimestamp());
        assertTrue(0l != p.getRetryUntil());
        assertTrue(0l != p.getMaxRetries());

        // Actual test
        p.processServerResponse(Policy.NOT_LICENSED, null);
        assertEquals(0l, p.getValidityTimestamp());
        assertEquals(0l, p.getRetryUntil());
        assertEquals(0l, p.getMaxRetries());
    }

    public void testNoFailureOnEncodedExtras() {
        String sampleResponse = "0|1579380448|com.example.android.market.licensing|1|" +
                "ADf8I4ajjgc1P5ZI1S1DN/YIPIUNPECLrg==|1279578835423:VT=1&test=hello%20world%20%26" +
                "%20friends>=2&GR=3";
        p.processServerResponse(Policy.LICENSED,
                ResponseData.parse(sampleResponse));
        assertEquals(1l, p.getValidityTimestamp());
        assertEquals(2l, p.getRetryUntil());
        assertEquals(3l, p.getMaxRetries());
    }

}