hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2019 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 17 | // Unit Test for MediaTranscodingService. |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 18 | |
| 19 | //#define LOG_NDEBUG 0 |
| 20 | #define LOG_TAG "MediaTranscodingServiceTest" |
| 21 | |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 22 | #include <aidl/android/media/BnTranscodingClientCallback.h> |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 23 | #include <aidl/android/media/IMediaTranscodingService.h> |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 24 | #include <aidl/android/media/ITranscodingClient.h> |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 25 | #include <aidl/android/media/ITranscodingClientCallback.h> |
| 26 | #include <aidl/android/media/TranscodingJobParcel.h> |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 27 | #include <aidl/android/media/TranscodingJobPriority.h> |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 28 | #include <aidl/android/media/TranscodingRequestParcel.h> |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 29 | #include <android-base/logging.h> |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 30 | #include <android/binder_manager.h> |
| 31 | #include <android/binder_process.h> |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 32 | #include <binder/PermissionController.h> |
| 33 | #include <cutils/multiuser.h> |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 34 | #include <gtest/gtest.h> |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 35 | #include <utils/Log.h> |
| 36 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 37 | #include <iostream> |
| 38 | #include <list> |
| 39 | |
| 40 | #include "SimulatedTranscoder.h" |
| 41 | |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 42 | namespace android { |
| 43 | |
| 44 | namespace media { |
| 45 | |
| 46 | using Status = ::ndk::ScopedAStatus; |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 47 | using aidl::android::media::BnTranscodingClientCallback; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 48 | using aidl::android::media::IMediaTranscodingService; |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 49 | using aidl::android::media::ITranscodingClient; |
| 50 | using aidl::android::media::ITranscodingClientCallback; |
| 51 | using aidl::android::media::TranscodingJobParcel; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 52 | using aidl::android::media::TranscodingJobPriority; |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 53 | using aidl::android::media::TranscodingRequestParcel; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 54 | |
| 55 | // Note that -1 is valid and means using calling pid/uid for the service. But only privilege caller could |
| 56 | // use them. This test is not a privilege caller. |
| 57 | constexpr int32_t kInvalidClientPid = -5; |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 58 | constexpr const char* kInvalidClientName = ""; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 59 | constexpr const char* kInvalidClientOpPackageName = ""; |
| 60 | |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 61 | constexpr int32_t kClientUseCallingPid = IMediaTranscodingService::USE_CALLING_PID; |
| 62 | constexpr int32_t kClientUseCallingUid = IMediaTranscodingService::USE_CALLING_UID; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 63 | |
| 64 | constexpr uid_t kClientUid = 5000; |
| 65 | #define UID(n) (kClientUid + (n)) |
| 66 | |
| 67 | constexpr int32_t kClientId = 0; |
| 68 | #define CLIENT(n) (kClientId + (n)) |
| 69 | |
Chong Zhang | dd7aeb4 | 2020-06-11 11:11:44 -0700 | [diff] [blame] | 70 | constexpr int64_t kPaddingUs = 1000000; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 71 | constexpr int64_t kJobWithPaddingUs = SimulatedTranscoder::kJobDurationUs + kPaddingUs; |
| 72 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 73 | constexpr const char* kClientName = "TestClient"; |
| 74 | constexpr const char* kClientOpPackageName = "TestClientPackage"; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 75 | constexpr const char* kClientPackageA = "com.android.tests.transcoding.testapp.A"; |
| 76 | constexpr const char* kClientPackageB = "com.android.tests.transcoding.testapp.B"; |
| 77 | constexpr const char* kClientPackageC = "com.android.tests.transcoding.testapp.C"; |
| 78 | constexpr const char* kTestActivityName = "/com.android.tests.transcoding.MainActivity"; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 79 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 80 | static status_t getUidForPackage(String16 packageName, userid_t userId, /*inout*/ uid_t& uid) { |
| 81 | PermissionController pc; |
| 82 | uid = pc.getPackageUid(packageName, 0); |
| 83 | if (uid <= 0) { |
| 84 | ALOGE("Unknown package: '%s'", String8(packageName).string()); |
| 85 | return BAD_VALUE; |
| 86 | } |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 87 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 88 | if (userId < 0) { |
| 89 | ALOGE("Invalid user: %d", userId); |
| 90 | return BAD_VALUE; |
| 91 | } |
| 92 | |
| 93 | uid = multiuser_get_uid(userId, uid); |
| 94 | return NO_ERROR; |
| 95 | } |
| 96 | |
| 97 | struct ShellHelper { |
| 98 | static bool RunCmd(const std::string& cmdStr) { |
| 99 | int ret = system(cmdStr.c_str()); |
| 100 | if (ret != 0) { |
| 101 | ALOGE("Failed to run cmd: %s, exitcode %d", cmdStr.c_str(), ret); |
| 102 | return false; |
| 103 | } |
| 104 | return true; |
| 105 | } |
| 106 | |
| 107 | static bool Start(const char* packageName, const char* activityName) { |
| 108 | return RunCmd("am start -W " + std::string(packageName) + std::string(activityName) + |
| 109 | " &> /dev/null"); |
| 110 | } |
| 111 | |
| 112 | static bool Stop(const char* packageName) { |
| 113 | return RunCmd("am force-stop " + std::string(packageName)); |
| 114 | } |
| 115 | }; |
| 116 | |
| 117 | struct EventTracker { |
| 118 | struct Event { |
| 119 | enum { NoEvent, Start, Pause, Resume, Finished, Failed } type; |
| 120 | int64_t clientId; |
| 121 | int32_t jobId; |
| 122 | }; |
| 123 | |
| 124 | #define DECLARE_EVENT(action) \ |
| 125 | static Event action(int32_t clientId, int32_t jobId) { \ |
| 126 | return {Event::action, clientId, jobId}; \ |
| 127 | } |
| 128 | |
| 129 | DECLARE_EVENT(Start); |
| 130 | DECLARE_EVENT(Pause); |
| 131 | DECLARE_EVENT(Resume); |
| 132 | DECLARE_EVENT(Finished); |
| 133 | DECLARE_EVENT(Failed); |
| 134 | |
| 135 | static constexpr Event NoEvent = {Event::NoEvent, 0, 0}; |
| 136 | |
| 137 | static std::string toString(const Event& event) { |
| 138 | std::string eventStr; |
| 139 | switch (event.type) { |
| 140 | case Event::Start: |
| 141 | eventStr = "Start"; |
| 142 | break; |
| 143 | case Event::Pause: |
| 144 | eventStr = "Pause"; |
| 145 | break; |
| 146 | case Event::Resume: |
| 147 | eventStr = "Resume"; |
| 148 | break; |
| 149 | case Event::Finished: |
| 150 | eventStr = "Finished"; |
| 151 | break; |
| 152 | case Event::Failed: |
| 153 | eventStr = "Failed"; |
| 154 | break; |
| 155 | default: |
| 156 | return "NoEvent"; |
| 157 | } |
| 158 | return "job {" + std::to_string(event.clientId) + ", " + std::to_string(event.jobId) + |
| 159 | "}: " + eventStr; |
| 160 | } |
| 161 | |
| 162 | // Pop 1 event from front, wait for up to timeoutUs if empty. |
| 163 | const Event& pop(int64_t timeoutUs = 0) { |
| 164 | std::unique_lock lock(mLock); |
| 165 | |
| 166 | if (mEventQueue.empty() && timeoutUs > 0) { |
| 167 | mCondition.wait_for(lock, std::chrono::microseconds(timeoutUs)); |
| 168 | } |
| 169 | |
| 170 | if (mEventQueue.empty()) { |
| 171 | mPoppedEvent = NoEvent; |
| 172 | } else { |
| 173 | mPoppedEvent = *mEventQueue.begin(); |
| 174 | mEventQueue.pop_front(); |
| 175 | } |
| 176 | |
| 177 | return mPoppedEvent; |
| 178 | } |
| 179 | |
| 180 | // Push 1 event to back. |
| 181 | void append(const Event& event) { |
| 182 | ALOGD("%s", toString(event).c_str()); |
| 183 | |
| 184 | std::unique_lock lock(mLock); |
| 185 | |
| 186 | mEventQueue.push_back(event); |
| 187 | mCondition.notify_one(); |
| 188 | } |
| 189 | |
| 190 | private: |
| 191 | std::mutex mLock; |
| 192 | std::condition_variable mCondition; |
| 193 | Event mPoppedEvent; |
| 194 | std::list<Event> mEventQueue; |
| 195 | }; |
| 196 | |
| 197 | // Operators for GTest macros. |
| 198 | bool operator==(const EventTracker::Event& lhs, const EventTracker::Event& rhs) { |
| 199 | return lhs.type == rhs.type && lhs.clientId == rhs.clientId && lhs.jobId == rhs.jobId; |
| 200 | } |
| 201 | |
| 202 | std::ostream& operator<<(std::ostream& str, const EventTracker::Event& v) { |
| 203 | str << EventTracker::toString(v); |
| 204 | return str; |
| 205 | } |
| 206 | |
| 207 | struct TestClientCallback : public BnTranscodingClientCallback, public EventTracker { |
| 208 | TestClientCallback(int32_t id) : mClientId(id) { |
| 209 | ALOGI("TestClientCallback %d Created", mClientId); |
| 210 | } |
| 211 | |
| 212 | virtual ~TestClientCallback() { ALOGI("TestClientCallback %d destroyed", mClientId); } |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 213 | |
hkuang | 1925309 | 2020-06-01 09:10:49 -0700 | [diff] [blame] | 214 | Status openFileDescriptor(const std::string& /*in_fileUri*/, const std::string& /*in_mode*/, |
| 215 | ::ndk::ScopedFileDescriptor* /*_aidl_return*/) override { |
| 216 | return Status::ok(); |
| 217 | } |
| 218 | |
Chong Zhang | de60f06 | 2020-06-11 17:05:10 -0700 | [diff] [blame] | 219 | Status onTranscodingStarted(int32_t in_jobId) override { |
| 220 | append(Start(mClientId, in_jobId)); |
| 221 | return Status::ok(); |
| 222 | } |
hkuang | 96471b8 | 2020-06-08 11:12:46 -0700 | [diff] [blame] | 223 | |
Chong Zhang | de60f06 | 2020-06-11 17:05:10 -0700 | [diff] [blame] | 224 | Status onTranscodingPaused(int32_t in_jobId) override { |
| 225 | append(Pause(mClientId, in_jobId)); |
| 226 | return Status::ok(); |
| 227 | } |
hkuang | 96471b8 | 2020-06-08 11:12:46 -0700 | [diff] [blame] | 228 | |
Chong Zhang | de60f06 | 2020-06-11 17:05:10 -0700 | [diff] [blame] | 229 | Status onTranscodingResumed(int32_t in_jobId) override { |
| 230 | append(Resume(mClientId, in_jobId)); |
| 231 | return Status::ok(); |
| 232 | } |
hkuang | 96471b8 | 2020-06-08 11:12:46 -0700 | [diff] [blame] | 233 | |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 234 | Status onTranscodingFinished( |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 235 | int32_t in_jobId, |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 236 | const ::aidl::android::media::TranscodingResultParcel& /* in_result */) override { |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 237 | append(Finished(mClientId, in_jobId)); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 238 | return Status::ok(); |
| 239 | } |
| 240 | |
| 241 | Status onTranscodingFailed( |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 242 | int32_t in_jobId, |
| 243 | ::aidl::android::media::TranscodingErrorCode /* in_errorCode */) override { |
| 244 | append(Failed(mClientId, in_jobId)); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 245 | return Status::ok(); |
| 246 | } |
| 247 | |
| 248 | Status onAwaitNumberOfJobsChanged(int32_t /* in_jobId */, int32_t /* in_oldAwaitNumber */, |
| 249 | int32_t /* in_newAwaitNumber */) override { |
| 250 | return Status::ok(); |
| 251 | } |
| 252 | |
Chong Zhang | de60f06 | 2020-06-11 17:05:10 -0700 | [diff] [blame] | 253 | Status onProgressUpdate(int32_t /*in_jobId*/, int32_t /*in_progress*/) override { |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 254 | return Status::ok(); |
| 255 | } |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 256 | |
| 257 | int32_t mClientId; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 258 | }; |
| 259 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 260 | class MediaTranscodingServiceTest : public ::testing::Test { |
| 261 | public: |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 262 | MediaTranscodingServiceTest() { ALOGI("MediaTranscodingServiceTest created"); } |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 263 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 264 | ~MediaTranscodingServiceTest() { ALOGI("MediaTranscodingingServiceTest destroyed"); } |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 265 | |
| 266 | void SetUp() override { |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 267 | // Need thread pool to receive callbacks, otherwise oneway callbacks are |
| 268 | // silently ignored. |
| 269 | ABinderProcess_startThreadPool(); |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 270 | ::ndk::SpAIBinder binder(AServiceManager_getService("media.transcoding")); |
| 271 | mService = IMediaTranscodingService::fromBinder(binder); |
| 272 | if (mService == nullptr) { |
| 273 | ALOGE("Failed to connect to the media.trascoding service."); |
| 274 | return; |
| 275 | } |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 276 | mClientCallback1 = ::ndk::SharedRefBase::make<TestClientCallback>(CLIENT(1)); |
| 277 | mClientCallback2 = ::ndk::SharedRefBase::make<TestClientCallback>(CLIENT(2)); |
| 278 | mClientCallback3 = ::ndk::SharedRefBase::make<TestClientCallback>(CLIENT(3)); |
| 279 | } |
| 280 | |
| 281 | std::shared_ptr<ITranscodingClient> registerOneClient( |
| 282 | const char* packageName, const std::shared_ptr<TestClientCallback>& callback, |
| 283 | uid_t defaultUid) { |
| 284 | uid_t uid; |
| 285 | if (getUidForPackage(String16(packageName), 0 /*userId*/, uid) != NO_ERROR) { |
| 286 | uid = defaultUid; |
| 287 | } |
| 288 | |
| 289 | ALOGD("registering %s with uid %d", packageName, uid); |
| 290 | |
| 291 | std::shared_ptr<ITranscodingClient> client; |
| 292 | Status status = mService->registerClient(callback, kClientName, packageName, uid, |
| 293 | kClientUseCallingPid, &client); |
| 294 | return status.isOk() ? client : nullptr; |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 295 | } |
| 296 | |
| 297 | void registerMultipleClients() { |
| 298 | // Register 3 clients. |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 299 | mClient1 = registerOneClient(kClientPackageA, mClientCallback1, UID(1)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 300 | EXPECT_TRUE(mClient1 != nullptr); |
| 301 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 302 | mClient2 = registerOneClient(kClientPackageB, mClientCallback2, UID(2)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 303 | EXPECT_TRUE(mClient2 != nullptr); |
| 304 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 305 | mClient3 = registerOneClient(kClientPackageC, mClientCallback3, UID(3)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 306 | EXPECT_TRUE(mClient3 != nullptr); |
| 307 | |
| 308 | // Check the number of clients. |
| 309 | int32_t numOfClients; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 310 | Status status = mService->getNumOfClients(&numOfClients); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 311 | EXPECT_TRUE(status.isOk()); |
| 312 | EXPECT_EQ(3, numOfClients); |
| 313 | } |
| 314 | |
| 315 | void unregisterMultipleClients() { |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 316 | Status status; |
| 317 | |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 318 | // Unregister the clients. |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 319 | status = mClient1->unregister(); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 320 | EXPECT_TRUE(status.isOk()); |
| 321 | |
| 322 | status = mClient2->unregister(); |
| 323 | EXPECT_TRUE(status.isOk()); |
| 324 | |
| 325 | status = mClient3->unregister(); |
| 326 | EXPECT_TRUE(status.isOk()); |
| 327 | |
| 328 | // Check the number of clients. |
| 329 | int32_t numOfClients; |
| 330 | status = mService->getNumOfClients(&numOfClients); |
| 331 | EXPECT_TRUE(status.isOk()); |
| 332 | EXPECT_EQ(0, numOfClients); |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 333 | } |
| 334 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 335 | static constexpr bool success = true; |
| 336 | static constexpr bool fail = false; |
| 337 | |
| 338 | template <bool expectation = success> |
| 339 | bool submit(const std::shared_ptr<ITranscodingClient>& client, int32_t jobId, |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 340 | const char* sourceFilePath, const char* destinationFilePath, |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 341 | TranscodingJobPriority priority = TranscodingJobPriority::kNormal) { |
| 342 | constexpr bool shouldSucceed = (expectation == success); |
| 343 | bool result; |
| 344 | TranscodingRequestParcel request; |
| 345 | TranscodingJobParcel job; |
| 346 | |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 347 | request.sourceFilePath = sourceFilePath; |
| 348 | request.destinationFilePath = destinationFilePath; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 349 | request.priority = priority; |
| 350 | Status status = client->submitRequest(request, &job, &result); |
| 351 | |
| 352 | EXPECT_TRUE(status.isOk()); |
| 353 | EXPECT_EQ(result, shouldSucceed); |
| 354 | if (shouldSucceed) { |
| 355 | EXPECT_EQ(job.jobId, jobId); |
| 356 | } |
| 357 | |
| 358 | return status.isOk() && (result == shouldSucceed) && (!shouldSucceed || job.jobId == jobId); |
| 359 | } |
| 360 | |
| 361 | template <bool expectation = success> |
| 362 | bool cancel(const std::shared_ptr<ITranscodingClient>& client, int32_t jobId) { |
| 363 | constexpr bool shouldSucceed = (expectation == success); |
| 364 | bool result; |
| 365 | Status status = client->cancelJob(jobId, &result); |
| 366 | |
| 367 | EXPECT_TRUE(status.isOk()); |
| 368 | EXPECT_EQ(result, shouldSucceed); |
| 369 | |
| 370 | return status.isOk() && (result == shouldSucceed); |
| 371 | } |
| 372 | |
| 373 | template <bool expectation = success> |
| 374 | bool getJob(const std::shared_ptr<ITranscodingClient>& client, int32_t jobId, |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 375 | const char* sourceFilePath, const char* destinationFilePath) { |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 376 | constexpr bool shouldSucceed = (expectation == success); |
| 377 | bool result; |
| 378 | TranscodingJobParcel job; |
| 379 | Status status = client->getJobWithId(jobId, &job, &result); |
| 380 | |
| 381 | EXPECT_TRUE(status.isOk()); |
| 382 | EXPECT_EQ(result, shouldSucceed); |
| 383 | if (shouldSucceed) { |
| 384 | EXPECT_EQ(job.jobId, jobId); |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 385 | EXPECT_EQ(job.request.sourceFilePath, sourceFilePath); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 386 | } |
| 387 | |
| 388 | return status.isOk() && (result == shouldSucceed) && |
Chong Zhang | de60f06 | 2020-06-11 17:05:10 -0700 | [diff] [blame] | 389 | (!shouldSucceed || |
| 390 | (job.jobId == jobId && job.request.sourceFilePath == sourceFilePath && |
| 391 | job.request.destinationFilePath == destinationFilePath)); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 392 | } |
| 393 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 394 | std::shared_ptr<IMediaTranscodingService> mService; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 395 | std::shared_ptr<TestClientCallback> mClientCallback1; |
| 396 | std::shared_ptr<TestClientCallback> mClientCallback2; |
| 397 | std::shared_ptr<TestClientCallback> mClientCallback3; |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 398 | std::shared_ptr<ITranscodingClient> mClient1; |
| 399 | std::shared_ptr<ITranscodingClient> mClient2; |
| 400 | std::shared_ptr<ITranscodingClient> mClient3; |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 401 | }; |
| 402 | |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 403 | TEST_F(MediaTranscodingServiceTest, TestRegisterNullClient) { |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 404 | std::shared_ptr<ITranscodingClient> client; |
| 405 | |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 406 | // Register the client with null callback. |
| 407 | Status status = mService->registerClient(nullptr, kClientName, kClientOpPackageName, |
| 408 | kClientUseCallingUid, kClientUseCallingPid, &client); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 409 | EXPECT_FALSE(status.isOk()); |
| 410 | } |
| 411 | |
| 412 | TEST_F(MediaTranscodingServiceTest, TestRegisterClientWithInvalidClientPid) { |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 413 | std::shared_ptr<ITranscodingClient> client; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 414 | |
| 415 | // Register the client with the service. |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 416 | Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName, |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 417 | kClientUseCallingUid, kInvalidClientPid, &client); |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 418 | EXPECT_FALSE(status.isOk()); |
| 419 | } |
| 420 | |
| 421 | TEST_F(MediaTranscodingServiceTest, TestRegisterClientWithInvalidClientName) { |
| 422 | std::shared_ptr<ITranscodingClient> client; |
| 423 | |
| 424 | // Register the client with the service. |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 425 | Status status = mService->registerClient(mClientCallback1, kInvalidClientName, |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 426 | kInvalidClientOpPackageName, kClientUseCallingUid, |
| 427 | kClientUseCallingPid, &client); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 428 | EXPECT_FALSE(status.isOk()); |
| 429 | } |
| 430 | |
| 431 | TEST_F(MediaTranscodingServiceTest, TestRegisterClientWithInvalidClientPackageName) { |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 432 | std::shared_ptr<ITranscodingClient> client; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 433 | |
| 434 | // Register the client with the service. |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 435 | Status status = |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 436 | mService->registerClient(mClientCallback1, kClientName, kInvalidClientOpPackageName, |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 437 | kClientUseCallingUid, kClientUseCallingPid, &client); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 438 | EXPECT_FALSE(status.isOk()); |
| 439 | } |
| 440 | |
| 441 | TEST_F(MediaTranscodingServiceTest, TestRegisterOneClient) { |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 442 | std::shared_ptr<ITranscodingClient> client; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 443 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 444 | Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName, |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 445 | kClientUseCallingUid, kClientUseCallingPid, &client); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 446 | EXPECT_TRUE(status.isOk()); |
| 447 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 448 | // Validate the client. |
| 449 | EXPECT_TRUE(client != nullptr); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 450 | |
| 451 | // Check the number of Clients. |
| 452 | int32_t numOfClients; |
| 453 | status = mService->getNumOfClients(&numOfClients); |
| 454 | EXPECT_TRUE(status.isOk()); |
| 455 | EXPECT_EQ(1, numOfClients); |
| 456 | |
| 457 | // Unregister the client. |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 458 | status = client->unregister(); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 459 | EXPECT_TRUE(status.isOk()); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 460 | |
| 461 | // Check the number of Clients. |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 462 | status = mService->getNumOfClients(&numOfClients); |
| 463 | EXPECT_TRUE(status.isOk()); |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 464 | EXPECT_EQ(0, numOfClients); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 465 | } |
| 466 | |
| 467 | TEST_F(MediaTranscodingServiceTest, TestRegisterClientTwice) { |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 468 | std::shared_ptr<ITranscodingClient> client; |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 469 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 470 | Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName, |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 471 | kClientUseCallingUid, kClientUseCallingPid, &client); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 472 | EXPECT_TRUE(status.isOk()); |
| 473 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 474 | // Validate the client. |
| 475 | EXPECT_TRUE(client != nullptr); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 476 | |
| 477 | // Register the client again and expects failure. |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 478 | std::shared_ptr<ITranscodingClient> client1; |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 479 | status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName, |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 480 | kClientUseCallingUid, kClientUseCallingPid, &client1); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 481 | EXPECT_FALSE(status.isOk()); |
| 482 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 483 | // Unregister the client. |
| 484 | status = client->unregister(); |
| 485 | EXPECT_TRUE(status.isOk()); |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 486 | } |
| 487 | |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 488 | TEST_F(MediaTranscodingServiceTest, TestRegisterMultipleClients) { |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 489 | registerMultipleClients(); |
| 490 | unregisterMultipleClients(); |
Chong Zhang | 8e06263 | 2020-03-31 10:56:37 -0700 | [diff] [blame] | 491 | } |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 492 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 493 | TEST_F(MediaTranscodingServiceTest, TestJobIdIndependence) { |
| 494 | registerMultipleClients(); |
| 495 | |
| 496 | // Submit 2 requests on client1 first. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 497 | EXPECT_TRUE(submit(mClient1, 0, "test_source_file", "test_destination_file")); |
| 498 | EXPECT_TRUE(submit(mClient1, 1, "test_source_file", "test_destination_file")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 499 | |
| 500 | // Submit 2 requests on client2, jobId should be independent for each client. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 501 | EXPECT_TRUE(submit(mClient2, 0, "test_source_file", "test_destination_file")); |
| 502 | EXPECT_TRUE(submit(mClient2, 1, "test_source_file", "test_destination_file")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 503 | |
| 504 | // Cancel all jobs. |
| 505 | EXPECT_TRUE(cancel(mClient1, 0)); |
| 506 | EXPECT_TRUE(cancel(mClient1, 1)); |
| 507 | EXPECT_TRUE(cancel(mClient2, 0)); |
| 508 | EXPECT_TRUE(cancel(mClient2, 1)); |
| 509 | |
| 510 | unregisterMultipleClients(); |
| 511 | } |
| 512 | |
| 513 | TEST_F(MediaTranscodingServiceTest, TestSubmitCancelJobs) { |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 514 | registerMultipleClients(); |
| 515 | |
| 516 | // Test jobId assignment. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 517 | EXPECT_TRUE(submit(mClient1, 0, "test_source_file_0", "test_destination_file")); |
| 518 | EXPECT_TRUE(submit(mClient1, 1, "test_source_file_1", "test_destination_file")); |
| 519 | EXPECT_TRUE(submit(mClient1, 2, "test_source_file_2", "test_destination_file")); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 520 | |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 521 | // Test submit bad request (no valid sourceFilePath) fails. |
| 522 | EXPECT_TRUE(submit<fail>(mClient1, 0, "", "")); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 523 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 524 | // Test cancel non-existent job fails. |
| 525 | EXPECT_TRUE(cancel<fail>(mClient1, 100)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 526 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 527 | // Job 0 should start immediately and finish in 2 seconds, followed by Job 1 start. |
| 528 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0)); |
| 529 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0)); |
| 530 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 531 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 532 | // Test cancel valid jobId in random order. |
| 533 | // Test cancel finished job fails. |
| 534 | EXPECT_TRUE(cancel(mClient1, 2)); |
| 535 | EXPECT_TRUE(cancel<fail>(mClient1, 0)); |
| 536 | EXPECT_TRUE(cancel(mClient1, 1)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 537 | |
| 538 | // Test cancel job again fails. |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 539 | EXPECT_TRUE(cancel<fail>(mClient1, 1)); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 540 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 541 | // Test no more events arriving after cancel. |
| 542 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::NoEvent); |
Chong Zhang | 182b06a | 2020-04-09 14:38:05 -0700 | [diff] [blame] | 543 | |
| 544 | unregisterMultipleClients(); |
| 545 | } |
| 546 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 547 | TEST_F(MediaTranscodingServiceTest, TestGetJobs) { |
| 548 | registerMultipleClients(); |
| 549 | |
| 550 | // Submit 3 requests. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 551 | EXPECT_TRUE(submit(mClient1, 0, "test_source_file_0", "test_destination_file_0")); |
| 552 | EXPECT_TRUE(submit(mClient1, 1, "test_source_file_1", "test_destination_file_1")); |
| 553 | EXPECT_TRUE(submit(mClient1, 2, "test_source_file_2", "test_destination_file_2")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 554 | |
| 555 | // Test get jobs by id. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 556 | EXPECT_TRUE(getJob(mClient1, 2, "test_source_file_2", "test_destination_file_2")); |
| 557 | EXPECT_TRUE(getJob(mClient1, 1, "test_source_file_1", "test_destination_file_1")); |
| 558 | EXPECT_TRUE(getJob(mClient1, 0, "test_source_file_0", "test_destination_file_0")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 559 | |
| 560 | // Test get job by invalid id fails. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 561 | EXPECT_TRUE(getJob<fail>(mClient1, 100, "", "")); |
| 562 | EXPECT_TRUE(getJob<fail>(mClient1, -1, "", "")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 563 | |
| 564 | // Test get job after cancel fails. |
| 565 | EXPECT_TRUE(cancel(mClient1, 2)); |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 566 | EXPECT_TRUE(getJob<fail>(mClient1, 2, "", "")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 567 | |
| 568 | // Job 0 should start immediately and finish in 2 seconds, followed by Job 1 start. |
| 569 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0)); |
| 570 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0)); |
| 571 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1)); |
| 572 | |
| 573 | // Test get job after finish fails. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 574 | EXPECT_TRUE(getJob<fail>(mClient1, 0, "", "")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 575 | |
| 576 | // Test get the remaining job 1. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 577 | EXPECT_TRUE(getJob(mClient1, 1, "test_source_file_1", "test_destination_file_1")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 578 | |
| 579 | // Cancel remaining job 1. |
| 580 | EXPECT_TRUE(cancel(mClient1, 1)); |
| 581 | |
| 582 | unregisterMultipleClients(); |
| 583 | } |
| 584 | |
| 585 | TEST_F(MediaTranscodingServiceTest, TestSubmitCancelWithOfflineJobs) { |
| 586 | registerMultipleClients(); |
| 587 | |
| 588 | // Submit some offline jobs first. |
Chong Zhang | de60f06 | 2020-06-11 17:05:10 -0700 | [diff] [blame] | 589 | EXPECT_TRUE(submit(mClient1, 0, "test_source_file_0", "test_destination_file_0", |
| 590 | TranscodingJobPriority::kUnspecified)); |
| 591 | EXPECT_TRUE(submit(mClient1, 1, "test_source_file_1", "test_destination_file_1", |
| 592 | TranscodingJobPriority::kUnspecified)); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 593 | |
| 594 | // Job 0 should start immediately. |
| 595 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0)); |
| 596 | |
| 597 | // Submit more real-time jobs. |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 598 | EXPECT_TRUE(submit(mClient1, 2, "test_source_file_2", "test_destination_file_2")); |
| 599 | EXPECT_TRUE(submit(mClient1, 3, "test_source_file_3", "test_destination_file_3")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 600 | |
| 601 | // Job 0 should pause immediately and job 2 should start. |
| 602 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Pause(CLIENT(1), 0)); |
| 603 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 2)); |
| 604 | |
| 605 | // Job 2 should finish in 2 seconds and job 3 should start. |
| 606 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 2)); |
| 607 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 3)); |
| 608 | |
| 609 | // Cancel job 3 now |
| 610 | EXPECT_TRUE(cancel(mClient1, 3)); |
| 611 | |
| 612 | // Job 0 should resume and finish in 2 seconds, followed by job 1 start. |
| 613 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Resume(CLIENT(1), 0)); |
| 614 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0)); |
| 615 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1)); |
| 616 | |
| 617 | // Cancel remaining job 1. |
| 618 | EXPECT_TRUE(cancel(mClient1, 1)); |
| 619 | |
| 620 | unregisterMultipleClients(); |
| 621 | } |
| 622 | |
Chong Zhang | 15c192a | 2020-05-05 16:24:00 -0700 | [diff] [blame] | 623 | TEST_F(MediaTranscodingServiceTest, TestClientUseAfterUnregister) { |
| 624 | std::shared_ptr<ITranscodingClient> client; |
| 625 | |
| 626 | // Register a client, then unregister. |
| 627 | Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName, |
| 628 | kClientUseCallingUid, kClientUseCallingPid, &client); |
| 629 | EXPECT_TRUE(status.isOk()); |
| 630 | |
| 631 | status = client->unregister(); |
| 632 | EXPECT_TRUE(status.isOk()); |
| 633 | |
| 634 | // Test various operations on the client, should fail with ERROR_DISCONNECTED. |
| 635 | TranscodingJobParcel job; |
| 636 | bool result; |
| 637 | status = client->getJobWithId(0, &job, &result); |
| 638 | EXPECT_EQ(status.getServiceSpecificError(), IMediaTranscodingService::ERROR_DISCONNECTED); |
| 639 | |
| 640 | status = client->cancelJob(0, &result); |
| 641 | EXPECT_EQ(status.getServiceSpecificError(), IMediaTranscodingService::ERROR_DISCONNECTED); |
| 642 | |
| 643 | TranscodingRequestParcel request; |
| 644 | status = client->submitRequest(request, &job, &result); |
| 645 | EXPECT_EQ(status.getServiceSpecificError(), IMediaTranscodingService::ERROR_DISCONNECTED); |
| 646 | } |
| 647 | |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 648 | TEST_F(MediaTranscodingServiceTest, TestTranscodingUidPolicy) { |
| 649 | ALOGD("TestTranscodingUidPolicy starting..."); |
| 650 | |
| 651 | EXPECT_TRUE(ShellHelper::RunCmd("input keyevent KEYCODE_WAKEUP")); |
| 652 | EXPECT_TRUE(ShellHelper::RunCmd("wm dismiss-keyguard")); |
| 653 | EXPECT_TRUE(ShellHelper::Stop(kClientPackageA)); |
| 654 | EXPECT_TRUE(ShellHelper::Stop(kClientPackageB)); |
| 655 | EXPECT_TRUE(ShellHelper::Stop(kClientPackageC)); |
| 656 | |
| 657 | registerMultipleClients(); |
| 658 | |
| 659 | ALOGD("Moving app A to top..."); |
| 660 | EXPECT_TRUE(ShellHelper::Start(kClientPackageA, kTestActivityName)); |
| 661 | |
| 662 | // Submit 3 requests. |
| 663 | ALOGD("Submitting job to client1 (app A) ..."); |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 664 | EXPECT_TRUE(submit(mClient1, 0, "test_source_file_0", "test_destination_file_0")); |
| 665 | EXPECT_TRUE(submit(mClient1, 1, "test_source_file_1", "test_destination_file_1")); |
| 666 | EXPECT_TRUE(submit(mClient1, 2, "test_source_file_2", "test_destination_file_2")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 667 | |
| 668 | // Job 0 should start immediately. |
| 669 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0)); |
| 670 | |
| 671 | ALOGD("Moving app B to top..."); |
| 672 | EXPECT_TRUE(ShellHelper::Start(kClientPackageB, kTestActivityName)); |
| 673 | |
| 674 | // Job 0 should continue and finish in 2 seconds, then job 1 should start. |
| 675 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0)); |
| 676 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1)); |
| 677 | |
| 678 | ALOGD("Submitting job to client2 (app B) ..."); |
hkuang | 72d105f | 2020-05-21 10:48:55 -0700 | [diff] [blame] | 679 | EXPECT_TRUE(submit(mClient2, 0, "test_source_file_0", "test_destination_file_0")); |
Chong Zhang | 7522218 | 2020-04-29 14:43:42 -0700 | [diff] [blame] | 680 | |
| 681 | // Client1's job should pause, client2's job should start. |
| 682 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Pause(CLIENT(1), 1)); |
| 683 | EXPECT_EQ(mClientCallback2->pop(kPaddingUs), EventTracker::Start(CLIENT(2), 0)); |
| 684 | |
| 685 | ALOGD("Moving app A back to top..."); |
| 686 | EXPECT_TRUE(ShellHelper::Start(kClientPackageA, kTestActivityName)); |
| 687 | |
| 688 | // Client2's job should pause, client1's job 1 should resume. |
| 689 | EXPECT_EQ(mClientCallback2->pop(kPaddingUs), EventTracker::Pause(CLIENT(2), 0)); |
| 690 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Resume(CLIENT(1), 1)); |
| 691 | |
| 692 | // Client2's job 1 should finish in 2 seconds, then its job 2 should start. |
| 693 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 1)); |
| 694 | EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 2)); |
| 695 | |
| 696 | // After client2's jobs finish, client1's job should resume. |
| 697 | EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 2)); |
| 698 | EXPECT_EQ(mClientCallback2->pop(kPaddingUs), EventTracker::Resume(CLIENT(2), 0)); |
| 699 | |
| 700 | unregisterMultipleClients(); |
| 701 | |
| 702 | EXPECT_TRUE(ShellHelper::Stop(kClientPackageA)); |
| 703 | EXPECT_TRUE(ShellHelper::Stop(kClientPackageB)); |
| 704 | EXPECT_TRUE(ShellHelper::Stop(kClientPackageC)); |
| 705 | |
| 706 | ALOGD("TestTranscodingUidPolicy finished."); |
| 707 | } |
| 708 | |
hkuang | 9c04b8d | 2020-01-22 10:03:21 -0800 | [diff] [blame] | 709 | } // namespace media |
| 710 | } // namespace android |