blob: a8d42410e4d682f615e6a6e7560ac61a02055750 [file] [log] [blame]
hkuang9c04b8d2020-01-22 10:03:21 -08001/*
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 Zhang182b06a2020-04-09 14:38:05 -070017// Unit Test for MediaTranscodingService.
hkuang9c04b8d2020-01-22 10:03:21 -080018
19//#define LOG_NDEBUG 0
20#define LOG_TAG "MediaTranscodingServiceTest"
21
Chong Zhang182b06a2020-04-09 14:38:05 -070022#include <aidl/android/media/BnTranscodingClientCallback.h>
hkuang9c04b8d2020-01-22 10:03:21 -080023#include <aidl/android/media/IMediaTranscodingService.h>
Chong Zhang8e062632020-03-31 10:56:37 -070024#include <aidl/android/media/ITranscodingClient.h>
Chong Zhang182b06a2020-04-09 14:38:05 -070025#include <aidl/android/media/ITranscodingClientCallback.h>
26#include <aidl/android/media/TranscodingJobParcel.h>
Chong Zhang75222182020-04-29 14:43:42 -070027#include <aidl/android/media/TranscodingJobPriority.h>
Chong Zhang182b06a2020-04-09 14:38:05 -070028#include <aidl/android/media/TranscodingRequestParcel.h>
hkuang9c04b8d2020-01-22 10:03:21 -080029#include <android-base/logging.h>
hkuang9c04b8d2020-01-22 10:03:21 -080030#include <android/binder_manager.h>
31#include <android/binder_process.h>
Chong Zhang75222182020-04-29 14:43:42 -070032#include <binder/PermissionController.h>
33#include <cutils/multiuser.h>
hkuang9c04b8d2020-01-22 10:03:21 -080034#include <gtest/gtest.h>
hkuang9c04b8d2020-01-22 10:03:21 -080035#include <utils/Log.h>
36
Chong Zhang75222182020-04-29 14:43:42 -070037#include <iostream>
38#include <list>
39
40#include "SimulatedTranscoder.h"
41
hkuang9c04b8d2020-01-22 10:03:21 -080042namespace android {
43
44namespace media {
45
46using Status = ::ndk::ScopedAStatus;
Chong Zhang182b06a2020-04-09 14:38:05 -070047using aidl::android::media::BnTranscodingClientCallback;
hkuang9c04b8d2020-01-22 10:03:21 -080048using aidl::android::media::IMediaTranscodingService;
Chong Zhang182b06a2020-04-09 14:38:05 -070049using aidl::android::media::ITranscodingClient;
50using aidl::android::media::ITranscodingClientCallback;
51using aidl::android::media::TranscodingJobParcel;
Chong Zhang75222182020-04-29 14:43:42 -070052using aidl::android::media::TranscodingJobPriority;
Chong Zhang182b06a2020-04-09 14:38:05 -070053using aidl::android::media::TranscodingRequestParcel;
hkuang9c04b8d2020-01-22 10:03:21 -080054
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.
57constexpr int32_t kInvalidClientPid = -5;
Chong Zhang8e062632020-03-31 10:56:37 -070058constexpr const char* kInvalidClientName = "";
hkuang9c04b8d2020-01-22 10:03:21 -080059constexpr const char* kInvalidClientOpPackageName = "";
60
Chong Zhang182b06a2020-04-09 14:38:05 -070061constexpr int32_t kClientUseCallingPid = IMediaTranscodingService::USE_CALLING_PID;
62constexpr int32_t kClientUseCallingUid = IMediaTranscodingService::USE_CALLING_UID;
Chong Zhang75222182020-04-29 14:43:42 -070063
64constexpr uid_t kClientUid = 5000;
65#define UID(n) (kClientUid + (n))
66
67constexpr int32_t kClientId = 0;
68#define CLIENT(n) (kClientId + (n))
69
70constexpr int64_t kPaddingUs = 200000;
71constexpr int64_t kJobWithPaddingUs = SimulatedTranscoder::kJobDurationUs + kPaddingUs;
72
Chong Zhang8e062632020-03-31 10:56:37 -070073constexpr const char* kClientName = "TestClient";
74constexpr const char* kClientOpPackageName = "TestClientPackage";
Chong Zhang75222182020-04-29 14:43:42 -070075constexpr const char* kClientPackageA = "com.android.tests.transcoding.testapp.A";
76constexpr const char* kClientPackageB = "com.android.tests.transcoding.testapp.B";
77constexpr const char* kClientPackageC = "com.android.tests.transcoding.testapp.C";
78constexpr const char* kTestActivityName = "/com.android.tests.transcoding.MainActivity";
hkuang9c04b8d2020-01-22 10:03:21 -080079
Chong Zhang75222182020-04-29 14:43:42 -070080static 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 }
hkuang9c04b8d2020-01-22 10:03:21 -080087
Chong Zhang75222182020-04-29 14:43:42 -070088 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
97struct 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
117struct 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
190private:
191 std::mutex mLock;
192 std::condition_variable mCondition;
193 Event mPoppedEvent;
194 std::list<Event> mEventQueue;
195};
196
197// Operators for GTest macros.
198bool 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
202std::ostream& operator<<(std::ostream& str, const EventTracker::Event& v) {
203 str << EventTracker::toString(v);
204 return str;
205}
206
207struct 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); }
hkuang9c04b8d2020-01-22 10:03:21 -0800213
214 Status onTranscodingFinished(
Chong Zhang75222182020-04-29 14:43:42 -0700215 int32_t in_jobId,
hkuang9c04b8d2020-01-22 10:03:21 -0800216 const ::aidl::android::media::TranscodingResultParcel& /* in_result */) override {
Chong Zhang75222182020-04-29 14:43:42 -0700217 append(Finished(mClientId, in_jobId));
hkuang9c04b8d2020-01-22 10:03:21 -0800218 return Status::ok();
219 }
220
221 Status onTranscodingFailed(
Chong Zhang75222182020-04-29 14:43:42 -0700222 int32_t in_jobId,
223 ::aidl::android::media::TranscodingErrorCode /* in_errorCode */) override {
224 append(Failed(mClientId, in_jobId));
hkuang9c04b8d2020-01-22 10:03:21 -0800225 return Status::ok();
226 }
227
228 Status onAwaitNumberOfJobsChanged(int32_t /* in_jobId */, int32_t /* in_oldAwaitNumber */,
229 int32_t /* in_newAwaitNumber */) override {
230 return Status::ok();
231 }
232
Chong Zhang75222182020-04-29 14:43:42 -0700233 Status onProgressUpdate(int32_t in_jobId, int32_t in_progress) override {
234 // The progress numbers from the SimulatedTranscoder represents the
235 // event's type in the transcoder.
236 switch (in_progress) {
237 case SimulatedTranscoder::Event::Start:
238 append(EventTracker::Start(mClientId, in_jobId));
239 break;
240 case SimulatedTranscoder::Event::Pause:
241 append(EventTracker::Pause(mClientId, in_jobId));
242 break;
243 case SimulatedTranscoder::Event::Resume:
244 append(EventTracker::Resume(mClientId, in_jobId));
245 break;
246 default:
247 ALOGE("unrecognized progress number %d, ignored by test", in_progress);
248 break;
249 }
hkuang9c04b8d2020-01-22 10:03:21 -0800250 return Status::ok();
251 }
Chong Zhang75222182020-04-29 14:43:42 -0700252
253 int32_t mClientId;
hkuang9c04b8d2020-01-22 10:03:21 -0800254};
255
Chong Zhang8e062632020-03-31 10:56:37 -0700256class MediaTranscodingServiceTest : public ::testing::Test {
257public:
Chong Zhang75222182020-04-29 14:43:42 -0700258 MediaTranscodingServiceTest() { ALOGI("MediaTranscodingServiceTest created"); }
Chong Zhang8e062632020-03-31 10:56:37 -0700259
Chong Zhang75222182020-04-29 14:43:42 -0700260 ~MediaTranscodingServiceTest() { ALOGI("MediaTranscodingingServiceTest destroyed"); }
Chong Zhang8e062632020-03-31 10:56:37 -0700261
262 void SetUp() override {
Chong Zhang75222182020-04-29 14:43:42 -0700263 // Need thread pool to receive callbacks, otherwise oneway callbacks are
264 // silently ignored.
265 ABinderProcess_startThreadPool();
Chong Zhang8e062632020-03-31 10:56:37 -0700266 ::ndk::SpAIBinder binder(AServiceManager_getService("media.transcoding"));
267 mService = IMediaTranscodingService::fromBinder(binder);
268 if (mService == nullptr) {
269 ALOGE("Failed to connect to the media.trascoding service.");
270 return;
271 }
Chong Zhang75222182020-04-29 14:43:42 -0700272 mClientCallback1 = ::ndk::SharedRefBase::make<TestClientCallback>(CLIENT(1));
273 mClientCallback2 = ::ndk::SharedRefBase::make<TestClientCallback>(CLIENT(2));
274 mClientCallback3 = ::ndk::SharedRefBase::make<TestClientCallback>(CLIENT(3));
275 }
276
277 std::shared_ptr<ITranscodingClient> registerOneClient(
278 const char* packageName, const std::shared_ptr<TestClientCallback>& callback,
279 uid_t defaultUid) {
280 uid_t uid;
281 if (getUidForPackage(String16(packageName), 0 /*userId*/, uid) != NO_ERROR) {
282 uid = defaultUid;
283 }
284
285 ALOGD("registering %s with uid %d", packageName, uid);
286
287 std::shared_ptr<ITranscodingClient> client;
288 Status status = mService->registerClient(callback, kClientName, packageName, uid,
289 kClientUseCallingPid, &client);
290 return status.isOk() ? client : nullptr;
Chong Zhang182b06a2020-04-09 14:38:05 -0700291 }
292
293 void registerMultipleClients() {
294 // Register 3 clients.
Chong Zhang75222182020-04-29 14:43:42 -0700295 mClient1 = registerOneClient(kClientPackageA, mClientCallback1, UID(1));
Chong Zhang182b06a2020-04-09 14:38:05 -0700296 EXPECT_TRUE(mClient1 != nullptr);
297
Chong Zhang75222182020-04-29 14:43:42 -0700298 mClient2 = registerOneClient(kClientPackageB, mClientCallback2, UID(2));
Chong Zhang182b06a2020-04-09 14:38:05 -0700299 EXPECT_TRUE(mClient2 != nullptr);
300
Chong Zhang75222182020-04-29 14:43:42 -0700301 mClient3 = registerOneClient(kClientPackageC, mClientCallback3, UID(3));
Chong Zhang182b06a2020-04-09 14:38:05 -0700302 EXPECT_TRUE(mClient3 != nullptr);
303
304 // Check the number of clients.
305 int32_t numOfClients;
Chong Zhang75222182020-04-29 14:43:42 -0700306 Status status = mService->getNumOfClients(&numOfClients);
Chong Zhang182b06a2020-04-09 14:38:05 -0700307 EXPECT_TRUE(status.isOk());
308 EXPECT_EQ(3, numOfClients);
309 }
310
311 void unregisterMultipleClients() {
Chong Zhang75222182020-04-29 14:43:42 -0700312 Status status;
313
Chong Zhang182b06a2020-04-09 14:38:05 -0700314 // Unregister the clients.
Chong Zhang75222182020-04-29 14:43:42 -0700315 status = mClient1->unregister();
Chong Zhang182b06a2020-04-09 14:38:05 -0700316 EXPECT_TRUE(status.isOk());
317
318 status = mClient2->unregister();
319 EXPECT_TRUE(status.isOk());
320
321 status = mClient3->unregister();
322 EXPECT_TRUE(status.isOk());
323
324 // Check the number of clients.
325 int32_t numOfClients;
326 status = mService->getNumOfClients(&numOfClients);
327 EXPECT_TRUE(status.isOk());
328 EXPECT_EQ(0, numOfClients);
Chong Zhang8e062632020-03-31 10:56:37 -0700329 }
330
Chong Zhang75222182020-04-29 14:43:42 -0700331 static constexpr bool success = true;
332 static constexpr bool fail = false;
333
334 template <bool expectation = success>
335 bool submit(const std::shared_ptr<ITranscodingClient>& client, int32_t jobId,
336 const char* filename,
337 TranscodingJobPriority priority = TranscodingJobPriority::kNormal) {
338 constexpr bool shouldSucceed = (expectation == success);
339 bool result;
340 TranscodingRequestParcel request;
341 TranscodingJobParcel job;
342
343 request.fileName = filename;
344 request.priority = priority;
345 Status status = client->submitRequest(request, &job, &result);
346
347 EXPECT_TRUE(status.isOk());
348 EXPECT_EQ(result, shouldSucceed);
349 if (shouldSucceed) {
350 EXPECT_EQ(job.jobId, jobId);
351 }
352
353 return status.isOk() && (result == shouldSucceed) && (!shouldSucceed || job.jobId == jobId);
354 }
355
356 template <bool expectation = success>
357 bool cancel(const std::shared_ptr<ITranscodingClient>& client, int32_t jobId) {
358 constexpr bool shouldSucceed = (expectation == success);
359 bool result;
360 Status status = client->cancelJob(jobId, &result);
361
362 EXPECT_TRUE(status.isOk());
363 EXPECT_EQ(result, shouldSucceed);
364
365 return status.isOk() && (result == shouldSucceed);
366 }
367
368 template <bool expectation = success>
369 bool getJob(const std::shared_ptr<ITranscodingClient>& client, int32_t jobId,
370 const char* filename) {
371 constexpr bool shouldSucceed = (expectation == success);
372 bool result;
373 TranscodingJobParcel job;
374 Status status = client->getJobWithId(jobId, &job, &result);
375
376 EXPECT_TRUE(status.isOk());
377 EXPECT_EQ(result, shouldSucceed);
378 if (shouldSucceed) {
379 EXPECT_EQ(job.jobId, jobId);
380 EXPECT_EQ(job.request.fileName, filename);
381 }
382
383 return status.isOk() && (result == shouldSucceed) &&
384 (!shouldSucceed || (job.jobId == jobId && job.request.fileName == filename));
385 }
386
Chong Zhang8e062632020-03-31 10:56:37 -0700387 std::shared_ptr<IMediaTranscodingService> mService;
Chong Zhang75222182020-04-29 14:43:42 -0700388 std::shared_ptr<TestClientCallback> mClientCallback1;
389 std::shared_ptr<TestClientCallback> mClientCallback2;
390 std::shared_ptr<TestClientCallback> mClientCallback3;
Chong Zhang182b06a2020-04-09 14:38:05 -0700391 std::shared_ptr<ITranscodingClient> mClient1;
392 std::shared_ptr<ITranscodingClient> mClient2;
393 std::shared_ptr<ITranscodingClient> mClient3;
Chong Zhang8e062632020-03-31 10:56:37 -0700394};
395
hkuang9c04b8d2020-01-22 10:03:21 -0800396TEST_F(MediaTranscodingServiceTest, TestRegisterNullClient) {
Chong Zhang8e062632020-03-31 10:56:37 -0700397 std::shared_ptr<ITranscodingClient> client;
398
Chong Zhang182b06a2020-04-09 14:38:05 -0700399 // Register the client with null callback.
400 Status status = mService->registerClient(nullptr, kClientName, kClientOpPackageName,
401 kClientUseCallingUid, kClientUseCallingPid, &client);
hkuang9c04b8d2020-01-22 10:03:21 -0800402 EXPECT_FALSE(status.isOk());
403}
404
405TEST_F(MediaTranscodingServiceTest, TestRegisterClientWithInvalidClientPid) {
Chong Zhang8e062632020-03-31 10:56:37 -0700406 std::shared_ptr<ITranscodingClient> client;
hkuang9c04b8d2020-01-22 10:03:21 -0800407
408 // Register the client with the service.
Chong Zhang75222182020-04-29 14:43:42 -0700409 Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName,
Chong Zhang182b06a2020-04-09 14:38:05 -0700410 kClientUseCallingUid, kInvalidClientPid, &client);
Chong Zhang8e062632020-03-31 10:56:37 -0700411 EXPECT_FALSE(status.isOk());
412}
413
414TEST_F(MediaTranscodingServiceTest, TestRegisterClientWithInvalidClientName) {
415 std::shared_ptr<ITranscodingClient> client;
416
417 // Register the client with the service.
Chong Zhang75222182020-04-29 14:43:42 -0700418 Status status = mService->registerClient(mClientCallback1, kInvalidClientName,
Chong Zhang182b06a2020-04-09 14:38:05 -0700419 kInvalidClientOpPackageName, kClientUseCallingUid,
420 kClientUseCallingPid, &client);
hkuang9c04b8d2020-01-22 10:03:21 -0800421 EXPECT_FALSE(status.isOk());
422}
423
424TEST_F(MediaTranscodingServiceTest, TestRegisterClientWithInvalidClientPackageName) {
Chong Zhang8e062632020-03-31 10:56:37 -0700425 std::shared_ptr<ITranscodingClient> client;
hkuang9c04b8d2020-01-22 10:03:21 -0800426
427 // Register the client with the service.
Chong Zhang182b06a2020-04-09 14:38:05 -0700428 Status status =
Chong Zhang75222182020-04-29 14:43:42 -0700429 mService->registerClient(mClientCallback1, kClientName, kInvalidClientOpPackageName,
Chong Zhang182b06a2020-04-09 14:38:05 -0700430 kClientUseCallingUid, kClientUseCallingPid, &client);
hkuang9c04b8d2020-01-22 10:03:21 -0800431 EXPECT_FALSE(status.isOk());
432}
433
434TEST_F(MediaTranscodingServiceTest, TestRegisterOneClient) {
Chong Zhang8e062632020-03-31 10:56:37 -0700435 std::shared_ptr<ITranscodingClient> client;
hkuang9c04b8d2020-01-22 10:03:21 -0800436
Chong Zhang75222182020-04-29 14:43:42 -0700437 Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName,
Chong Zhang182b06a2020-04-09 14:38:05 -0700438 kClientUseCallingUid, kClientUseCallingPid, &client);
hkuang9c04b8d2020-01-22 10:03:21 -0800439 EXPECT_TRUE(status.isOk());
440
Chong Zhang8e062632020-03-31 10:56:37 -0700441 // Validate the client.
442 EXPECT_TRUE(client != nullptr);
hkuang9c04b8d2020-01-22 10:03:21 -0800443
444 // Check the number of Clients.
445 int32_t numOfClients;
446 status = mService->getNumOfClients(&numOfClients);
447 EXPECT_TRUE(status.isOk());
448 EXPECT_EQ(1, numOfClients);
449
450 // Unregister the client.
Chong Zhang8e062632020-03-31 10:56:37 -0700451 status = client->unregister();
hkuang9c04b8d2020-01-22 10:03:21 -0800452 EXPECT_TRUE(status.isOk());
hkuang9c04b8d2020-01-22 10:03:21 -0800453
454 // Check the number of Clients.
hkuang9c04b8d2020-01-22 10:03:21 -0800455 status = mService->getNumOfClients(&numOfClients);
456 EXPECT_TRUE(status.isOk());
Chong Zhang8e062632020-03-31 10:56:37 -0700457 EXPECT_EQ(0, numOfClients);
hkuang9c04b8d2020-01-22 10:03:21 -0800458}
459
460TEST_F(MediaTranscodingServiceTest, TestRegisterClientTwice) {
Chong Zhang8e062632020-03-31 10:56:37 -0700461 std::shared_ptr<ITranscodingClient> client;
hkuang9c04b8d2020-01-22 10:03:21 -0800462
Chong Zhang75222182020-04-29 14:43:42 -0700463 Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName,
Chong Zhang182b06a2020-04-09 14:38:05 -0700464 kClientUseCallingUid, kClientUseCallingPid, &client);
hkuang9c04b8d2020-01-22 10:03:21 -0800465 EXPECT_TRUE(status.isOk());
466
Chong Zhang8e062632020-03-31 10:56:37 -0700467 // Validate the client.
468 EXPECT_TRUE(client != nullptr);
hkuang9c04b8d2020-01-22 10:03:21 -0800469
470 // Register the client again and expects failure.
Chong Zhang8e062632020-03-31 10:56:37 -0700471 std::shared_ptr<ITranscodingClient> client1;
Chong Zhang75222182020-04-29 14:43:42 -0700472 status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName,
Chong Zhang182b06a2020-04-09 14:38:05 -0700473 kClientUseCallingUid, kClientUseCallingPid, &client1);
hkuang9c04b8d2020-01-22 10:03:21 -0800474 EXPECT_FALSE(status.isOk());
475
Chong Zhang8e062632020-03-31 10:56:37 -0700476 // Unregister the client.
477 status = client->unregister();
478 EXPECT_TRUE(status.isOk());
hkuang9c04b8d2020-01-22 10:03:21 -0800479}
480
Chong Zhang8e062632020-03-31 10:56:37 -0700481TEST_F(MediaTranscodingServiceTest, TestRegisterMultipleClients) {
Chong Zhang182b06a2020-04-09 14:38:05 -0700482 registerMultipleClients();
483 unregisterMultipleClients();
Chong Zhang8e062632020-03-31 10:56:37 -0700484}
Chong Zhang182b06a2020-04-09 14:38:05 -0700485
Chong Zhang75222182020-04-29 14:43:42 -0700486TEST_F(MediaTranscodingServiceTest, TestJobIdIndependence) {
487 registerMultipleClients();
488
489 // Submit 2 requests on client1 first.
490 EXPECT_TRUE(submit(mClient1, 0, "test_file"));
491 EXPECT_TRUE(submit(mClient1, 1, "test_file"));
492
493 // Submit 2 requests on client2, jobId should be independent for each client.
494 EXPECT_TRUE(submit(mClient2, 0, "test_file"));
495 EXPECT_TRUE(submit(mClient2, 1, "test_file"));
496
497 // Cancel all jobs.
498 EXPECT_TRUE(cancel(mClient1, 0));
499 EXPECT_TRUE(cancel(mClient1, 1));
500 EXPECT_TRUE(cancel(mClient2, 0));
501 EXPECT_TRUE(cancel(mClient2, 1));
502
503 unregisterMultipleClients();
504}
505
506TEST_F(MediaTranscodingServiceTest, TestSubmitCancelJobs) {
Chong Zhang182b06a2020-04-09 14:38:05 -0700507 registerMultipleClients();
508
509 // Test jobId assignment.
Chong Zhang75222182020-04-29 14:43:42 -0700510 EXPECT_TRUE(submit(mClient1, 0, "test_file_0"));
511 EXPECT_TRUE(submit(mClient1, 1, "test_file_1"));
512 EXPECT_TRUE(submit(mClient1, 2, "test_file_2"));
Chong Zhang182b06a2020-04-09 14:38:05 -0700513
514 // Test submit bad request (no valid fileName) fails.
Chong Zhang75222182020-04-29 14:43:42 -0700515 EXPECT_TRUE(submit<fail>(mClient1, 0, ""));
Chong Zhang182b06a2020-04-09 14:38:05 -0700516
Chong Zhang75222182020-04-29 14:43:42 -0700517 // Test cancel non-existent job fails.
518 EXPECT_TRUE(cancel<fail>(mClient1, 100));
Chong Zhang182b06a2020-04-09 14:38:05 -0700519
Chong Zhang75222182020-04-29 14:43:42 -0700520 // Job 0 should start immediately and finish in 2 seconds, followed by Job 1 start.
521 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0));
522 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0));
523 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1));
Chong Zhang182b06a2020-04-09 14:38:05 -0700524
Chong Zhang75222182020-04-29 14:43:42 -0700525 // Test cancel valid jobId in random order.
526 // Test cancel finished job fails.
527 EXPECT_TRUE(cancel(mClient1, 2));
528 EXPECT_TRUE(cancel<fail>(mClient1, 0));
529 EXPECT_TRUE(cancel(mClient1, 1));
Chong Zhang182b06a2020-04-09 14:38:05 -0700530
531 // Test cancel job again fails.
Chong Zhang75222182020-04-29 14:43:42 -0700532 EXPECT_TRUE(cancel<fail>(mClient1, 1));
Chong Zhang182b06a2020-04-09 14:38:05 -0700533
Chong Zhang75222182020-04-29 14:43:42 -0700534 // Test no more events arriving after cancel.
535 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::NoEvent);
Chong Zhang182b06a2020-04-09 14:38:05 -0700536
537 unregisterMultipleClients();
538}
539
Chong Zhang75222182020-04-29 14:43:42 -0700540TEST_F(MediaTranscodingServiceTest, TestGetJobs) {
541 registerMultipleClients();
542
543 // Submit 3 requests.
544 EXPECT_TRUE(submit(mClient1, 0, "test_file_0"));
545 EXPECT_TRUE(submit(mClient1, 1, "test_file_1"));
546 EXPECT_TRUE(submit(mClient1, 2, "test_file_2"));
547
548 // Test get jobs by id.
549 EXPECT_TRUE(getJob(mClient1, 2, "test_file_2"));
550 EXPECT_TRUE(getJob(mClient1, 1, "test_file_1"));
551 EXPECT_TRUE(getJob(mClient1, 0, "test_file_0"));
552
553 // Test get job by invalid id fails.
554 EXPECT_TRUE(getJob<fail>(mClient1, 100, ""));
555 EXPECT_TRUE(getJob<fail>(mClient1, -1, ""));
556
557 // Test get job after cancel fails.
558 EXPECT_TRUE(cancel(mClient1, 2));
559 EXPECT_TRUE(getJob<fail>(mClient1, 2, ""));
560
561 // Job 0 should start immediately and finish in 2 seconds, followed by Job 1 start.
562 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0));
563 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0));
564 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1));
565
566 // Test get job after finish fails.
567 EXPECT_TRUE(getJob<fail>(mClient1, 0, ""));
568
569 // Test get the remaining job 1.
570 EXPECT_TRUE(getJob(mClient1, 1, "test_file_1"));
571
572 // Cancel remaining job 1.
573 EXPECT_TRUE(cancel(mClient1, 1));
574
575 unregisterMultipleClients();
576}
577
578TEST_F(MediaTranscodingServiceTest, TestSubmitCancelWithOfflineJobs) {
579 registerMultipleClients();
580
581 // Submit some offline jobs first.
582 EXPECT_TRUE(submit(mClient1, 0, "test_file_0", TranscodingJobPriority::kUnspecified));
583 EXPECT_TRUE(submit(mClient1, 1, "test_file_1", TranscodingJobPriority::kUnspecified));
584
585 // Job 0 should start immediately.
586 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0));
587
588 // Submit more real-time jobs.
589 EXPECT_TRUE(submit(mClient1, 2, "test_file_2"));
590 EXPECT_TRUE(submit(mClient1, 3, "test_file_3"));
591
592 // Job 0 should pause immediately and job 2 should start.
593 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Pause(CLIENT(1), 0));
594 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 2));
595
596 // Job 2 should finish in 2 seconds and job 3 should start.
597 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 2));
598 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 3));
599
600 // Cancel job 3 now
601 EXPECT_TRUE(cancel(mClient1, 3));
602
603 // Job 0 should resume and finish in 2 seconds, followed by job 1 start.
604 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Resume(CLIENT(1), 0));
605 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0));
606 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1));
607
608 // Cancel remaining job 1.
609 EXPECT_TRUE(cancel(mClient1, 1));
610
611 unregisterMultipleClients();
612}
613
Chong Zhang15c192a2020-05-05 16:24:00 -0700614TEST_F(MediaTranscodingServiceTest, TestClientUseAfterUnregister) {
615 std::shared_ptr<ITranscodingClient> client;
616
617 // Register a client, then unregister.
618 Status status = mService->registerClient(mClientCallback1, kClientName, kClientOpPackageName,
619 kClientUseCallingUid, kClientUseCallingPid, &client);
620 EXPECT_TRUE(status.isOk());
621
622 status = client->unregister();
623 EXPECT_TRUE(status.isOk());
624
625 // Test various operations on the client, should fail with ERROR_DISCONNECTED.
626 TranscodingJobParcel job;
627 bool result;
628 status = client->getJobWithId(0, &job, &result);
629 EXPECT_EQ(status.getServiceSpecificError(), IMediaTranscodingService::ERROR_DISCONNECTED);
630
631 status = client->cancelJob(0, &result);
632 EXPECT_EQ(status.getServiceSpecificError(), IMediaTranscodingService::ERROR_DISCONNECTED);
633
634 TranscodingRequestParcel request;
635 status = client->submitRequest(request, &job, &result);
636 EXPECT_EQ(status.getServiceSpecificError(), IMediaTranscodingService::ERROR_DISCONNECTED);
637}
638
Chong Zhang75222182020-04-29 14:43:42 -0700639TEST_F(MediaTranscodingServiceTest, TestTranscodingUidPolicy) {
640 ALOGD("TestTranscodingUidPolicy starting...");
641
642 EXPECT_TRUE(ShellHelper::RunCmd("input keyevent KEYCODE_WAKEUP"));
643 EXPECT_TRUE(ShellHelper::RunCmd("wm dismiss-keyguard"));
644 EXPECT_TRUE(ShellHelper::Stop(kClientPackageA));
645 EXPECT_TRUE(ShellHelper::Stop(kClientPackageB));
646 EXPECT_TRUE(ShellHelper::Stop(kClientPackageC));
647
648 registerMultipleClients();
649
650 ALOGD("Moving app A to top...");
651 EXPECT_TRUE(ShellHelper::Start(kClientPackageA, kTestActivityName));
652
653 // Submit 3 requests.
654 ALOGD("Submitting job to client1 (app A) ...");
655 EXPECT_TRUE(submit(mClient1, 0, "test_file_0"));
656 EXPECT_TRUE(submit(mClient1, 1, "test_file_1"));
657 EXPECT_TRUE(submit(mClient1, 2, "test_file_2"));
658
659 // Job 0 should start immediately.
660 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 0));
661
662 ALOGD("Moving app B to top...");
663 EXPECT_TRUE(ShellHelper::Start(kClientPackageB, kTestActivityName));
664
665 // Job 0 should continue and finish in 2 seconds, then job 1 should start.
666 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 0));
667 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 1));
668
669 ALOGD("Submitting job to client2 (app B) ...");
670 EXPECT_TRUE(submit(mClient2, 0, "test_file_0"));
671
672 // Client1's job should pause, client2's job should start.
673 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Pause(CLIENT(1), 1));
674 EXPECT_EQ(mClientCallback2->pop(kPaddingUs), EventTracker::Start(CLIENT(2), 0));
675
676 ALOGD("Moving app A back to top...");
677 EXPECT_TRUE(ShellHelper::Start(kClientPackageA, kTestActivityName));
678
679 // Client2's job should pause, client1's job 1 should resume.
680 EXPECT_EQ(mClientCallback2->pop(kPaddingUs), EventTracker::Pause(CLIENT(2), 0));
681 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Resume(CLIENT(1), 1));
682
683 // Client2's job 1 should finish in 2 seconds, then its job 2 should start.
684 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 1));
685 EXPECT_EQ(mClientCallback1->pop(kPaddingUs), EventTracker::Start(CLIENT(1), 2));
686
687 // After client2's jobs finish, client1's job should resume.
688 EXPECT_EQ(mClientCallback1->pop(kJobWithPaddingUs), EventTracker::Finished(CLIENT(1), 2));
689 EXPECT_EQ(mClientCallback2->pop(kPaddingUs), EventTracker::Resume(CLIENT(2), 0));
690
691 unregisterMultipleClients();
692
693 EXPECT_TRUE(ShellHelper::Stop(kClientPackageA));
694 EXPECT_TRUE(ShellHelper::Stop(kClientPackageB));
695 EXPECT_TRUE(ShellHelper::Stop(kClientPackageC));
696
697 ALOGD("TestTranscodingUidPolicy finished.");
698}
699
hkuang9c04b8d2020-01-22 10:03:21 -0800700} // namespace media
701} // namespace android