Initial test cases for DrmInfoRequest
Change-Id: Iebe616b637514fdc97bac3963cce82d4beda6ee0
diff --git a/tests/tests/drm/src/android/drm/cts/DrmInfoRequestTest.java b/tests/tests/drm/src/android/drm/cts/DrmInfoRequestTest.java
new file mode 100644
index 0000000..0ec8d49
--- /dev/null
+++ b/tests/tests/drm/src/android/drm/cts/DrmInfoRequestTest.java
@@ -0,0 +1,123 @@
+/*
+ * Copyright (C) 2012 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 android.drm.cts;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import android.util.Log;
+import android.test.AndroidTestCase;
+import android.drm.DrmInfoRequest;
+
+public class DrmInfoRequestTest extends AndroidTestCase {
+ private static final String TAG = "CtsDrmInfoRequestTest";
+ private static final String DEFAULT_MIME = "video/";
+ private static final int DEFAULT_TYPE =
+ DrmInfoRequest.TYPE_REGISTRATION_INFO;
+
+ public static void testInvalidInfoTypes() throws Exception {
+ checkInvalidInfoType(DrmInfoRequest.TYPE_REGISTRATION_INFO - 1);
+ checkInvalidInfoType(
+ DrmInfoRequest.TYPE_RIGHTS_ACQUISITION_PROGRESS_INFO + 1);
+ }
+
+ public static void testValidInfoTypes() throws Exception {
+ checkValidInfoType(DrmInfoRequest.TYPE_REGISTRATION_INFO);
+ checkValidInfoType(DrmInfoRequest.TYPE_UNREGISTRATION_INFO);
+ checkValidInfoType(DrmInfoRequest.TYPE_RIGHTS_ACQUISITION_INFO);
+ checkValidInfoType(
+ DrmInfoRequest.TYPE_RIGHTS_ACQUISITION_PROGRESS_INFO);
+ }
+
+ public static void testGetInfoType() throws Exception {
+ checkGetInfoType(DrmInfoRequest.TYPE_REGISTRATION_INFO);
+ checkGetInfoType(DrmInfoRequest.TYPE_UNREGISTRATION_INFO);
+ checkGetInfoType(DrmInfoRequest.TYPE_RIGHTS_ACQUISITION_INFO);
+ checkGetInfoType(
+ DrmInfoRequest.TYPE_RIGHTS_ACQUISITION_PROGRESS_INFO);
+ }
+
+ public static void testInvalidMimeTypes() throws Exception {
+ checkInvalidMimeType("");
+ checkInvalidMimeType(null);
+ }
+
+ public static void testGetMimeType() throws Exception {
+ checkGetMimeType("Hello");
+ checkGetMimeType("World");
+ checkGetMimeType("Hello World");
+ }
+
+ public static void testPutAndGetKeys() throws Exception {
+ HashMap<String, Object> attributes = new HashMap<String, Object>(3);
+ attributes.put("Hello", "");
+ attributes.put("World", null);
+ attributes.put("Hello World", "Hello World");
+
+ // Store all the attributes in DrmInfoRequest object request.
+ DrmInfoRequest request = new DrmInfoRequest(DEFAULT_TYPE, DEFAULT_MIME);
+ Iterator<String> keys = attributes.keySet().iterator();
+ while (keys.hasNext()) {
+ final String key = (String) keys.next();
+ request.put(key, attributes.get(key));
+ }
+
+ // Request object must have all the keys that attributes does.
+ Iterator<String> infoKeys = request.keyIterator();
+ while (keys.hasNext()) {
+ final String key = (String) keys.next();
+ assertEquals(request.get(key), attributes.get(key));
+ }
+
+ // Attributes object must have all the keys that request does.
+ while (infoKeys.hasNext()) {
+ final String key = (String) infoKeys.next();
+ assertEquals(request.get(key), attributes.get(key));
+ }
+ }
+
+ private static void checkInvalidMimeType(String mimeType) throws Exception {
+ try {
+ DrmInfoRequest request = new DrmInfoRequest(DEFAULT_TYPE, mimeType);
+ fail("Mime type " + mimeType + " was accepted for DrmInfoRequest");
+ } catch(IllegalArgumentException e) {
+ // Expected and thus intentionally ignored.
+ }
+ }
+
+ private static void checkGetMimeType(String mimeType) throws Exception {
+ DrmInfoRequest request = new DrmInfoRequest(DEFAULT_TYPE, mimeType);
+ assertEquals(request.getMimeType(), mimeType);
+ }
+
+ private static void checkGetInfoType(int type) throws Exception {
+ DrmInfoRequest request = new DrmInfoRequest(type, DEFAULT_MIME);
+ assertEquals(request.getInfoType(), type);
+ }
+
+ private static void checkInvalidInfoType(int type) throws Exception {
+ try {
+ DrmInfoRequest request = new DrmInfoRequest(type, DEFAULT_MIME);
+ fail("Info type " + type + " was accepted for DrmInfoRequest");
+ } catch(IllegalArgumentException e) {
+ // Expected, and thus intentionally ignored.
+ }
+ }
+
+ private static void checkValidInfoType(int type) throws Exception {
+ DrmInfoRequest request = new DrmInfoRequest(type, DEFAULT_MIME);
+ }
+}
