1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/android/jni_array.h" 6 7 #include "base/android/jni_android.h" 8 #include "base/android/scoped_java_ref.h" 9 #include "testing/gtest/include/gtest/gtest.h" 10 11 namespace base { 12 namespace android { 13 14 TEST(JniArray, BasicConversions) { 15 const uint8 kBytes[] = { 0, 1, 2, 3 }; 16 const size_t kLen = arraysize(kBytes); 17 JNIEnv* env = AttachCurrentThread(); 18 ScopedJavaLocalRef<jbyteArray> bytes = ToJavaByteArray(env, kBytes, kLen); 19 ASSERT_TRUE(bytes.obj()); 20 21 std::vector<uint8> vec(5); 22 JavaByteArrayToByteVector(env, bytes.obj(), &vec); 23 EXPECT_EQ(4U, vec.size()); 24 EXPECT_EQ(std::vector<uint8>(kBytes, kBytes + kLen), vec); 25 26 AppendJavaByteArrayToByteVector(env, bytes.obj(), &vec); 27 EXPECT_EQ(8U, vec.size()); 28 } 29 30 void CheckLongConversion( 31 JNIEnv* env, 32 const int64* long_array, 33 const size_t len, 34 const ScopedJavaLocalRef<jlongArray>& longs) { 35 ASSERT_TRUE(longs.obj()); 36 37 jsize java_array_len = env->GetArrayLength(longs.obj()); 38 ASSERT_EQ(static_cast<jsize>(len), java_array_len); 39 40 jlong value; 41 for (size_t i = 0; i < len; ++i) { 42 env->GetLongArrayRegion(longs.obj(), i, 1, &value); 43 ASSERT_EQ(long_array[i], value); 44 } 45 } 46 47 TEST(JniArray, LongConversions) { 48 const int64 kLongs[] = { 0, 1, -1, kint64min, kint64max}; 49 const size_t kLen = arraysize(kLongs); 50 51 JNIEnv* env = AttachCurrentThread(); 52 CheckLongConversion(env, kLongs, kLen, ToJavaLongArray(env, kLongs, kLen)); 53 54 const std::vector<int64> vec(kLongs, kLongs + kLen); 55 CheckLongConversion(env, kLongs, kLen, ToJavaLongArray(env, vec)); 56 } 57 58 TEST(JniArray, JavaIntArrayToIntVector) { 59 const int kInts[] = {0, 1, -1}; 60 const size_t kLen = arraysize(kInts); 61 62 JNIEnv* env = AttachCurrentThread(); 63 ScopedJavaLocalRef<jintArray> jints(env, env->NewIntArray(kLen)); 64 ASSERT_TRUE(jints.obj()); 65 66 for (size_t i = 0; i < kLen; ++i) { 67 jint j = static_cast<jint>(kInts[i]); 68 env->SetIntArrayRegion(jints.obj(), i, 1, &j); 69 ASSERT_FALSE(HasException(env)); 70 } 71 72 std::vector<int> ints; 73 JavaIntArrayToIntVector(env, jints.obj(), &ints); 74 75 ASSERT_EQ(static_cast<jsize>(ints.size()), env->GetArrayLength(jints.obj())); 76 77 jint value; 78 for (size_t i = 0; i < kLen; ++i) { 79 env->GetIntArrayRegion(jints.obj(), i, 1, &value); 80 ASSERT_EQ(ints[i], value); 81 } 82 } 83 84 TEST(JniArray, JavaFloatArrayToFloatVector) { 85 const float kFloats[] = {0.0, 0.5, -0.5}; 86 const size_t kLen = arraysize(kFloats); 87 88 JNIEnv* env = AttachCurrentThread(); 89 ScopedJavaLocalRef<jfloatArray> jfloats(env, env->NewFloatArray(kLen)); 90 ASSERT_TRUE(jfloats.obj()); 91 92 for (size_t i = 0; i < kLen; ++i) { 93 jfloat j = static_cast<jfloat>(kFloats[i]); 94 env->SetFloatArrayRegion(jfloats.obj(), i, 1, &j); 95 ASSERT_FALSE(HasException(env)); 96 } 97 98 std::vector<float> floats; 99 JavaFloatArrayToFloatVector(env, jfloats.obj(), &floats); 100 101 ASSERT_EQ(static_cast<jsize>(floats.size()), 102 env->GetArrayLength(jfloats.obj())); 103 104 jfloat value; 105 for (size_t i = 0; i < kLen; ++i) { 106 env->GetFloatArrayRegion(jfloats.obj(), i, 1, &value); 107 ASSERT_EQ(floats[i], value); 108 } 109 } 110 111 TEST(JniArray, JavaArrayOfByteArrayToStringVector) { 112 const int kMaxItems = 50; 113 JNIEnv* env = AttachCurrentThread(); 114 115 // Create a byte[][] object. 116 ScopedJavaLocalRef<jclass> byte_array_clazz(env, env->FindClass("[B")); 117 ASSERT_TRUE(byte_array_clazz.obj()); 118 119 ScopedJavaLocalRef<jobjectArray> array( 120 env, env->NewObjectArray(kMaxItems, byte_array_clazz.obj(), NULL)); 121 ASSERT_TRUE(array.obj()); 122 123 // Create kMaxItems byte buffers. 124 char text[16]; 125 for (int i = 0; i < kMaxItems; ++i) { 126 snprintf(text, sizeof text, "%d", i); 127 ScopedJavaLocalRef<jbyteArray> byte_array = ToJavaByteArray( 128 env, reinterpret_cast<uint8*>(text), 129 static_cast<size_t>(strlen(text))); 130 ASSERT_TRUE(byte_array.obj()); 131 132 env->SetObjectArrayElement(array.obj(), i, byte_array.obj()); 133 ASSERT_FALSE(HasException(env)); 134 } 135 136 // Convert to std::vector<std::string>, check the content. 137 std::vector<std::string> vec; 138 JavaArrayOfByteArrayToStringVector(env, array.obj(), &vec); 139 140 EXPECT_EQ(static_cast<size_t>(kMaxItems), vec.size()); 141 for (int i = 0; i < kMaxItems; ++i) { 142 snprintf(text, sizeof text, "%d", i); 143 EXPECT_STREQ(text, vec[i].c_str()); 144 } 145 } 146 147 } // namespace android 148 } // namespace base 149