| /* |
| * Copyright 2019 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 androidx.work.testing |
| |
| import android.content.Context |
| import android.net.Uri |
| import androidx.test.core.app.ApplicationProvider |
| import androidx.test.ext.junit.runners.AndroidJUnit4 |
| import androidx.test.filters.MediumTest |
| import androidx.test.filters.SmallTest |
| import androidx.work.ListenableWorker.Result |
| import androidx.work.OneTimeWorkRequestBuilder |
| import androidx.work.WorkerFactory |
| import androidx.work.WorkerParameters |
| import androidx.work.await |
| import androidx.work.testing.workers.TestListenableWorker |
| import androidx.work.testing.workers.TestWorker |
| import kotlinx.coroutines.runBlocking |
| import org.hamcrest.CoreMatchers.`is` |
| import org.hamcrest.CoreMatchers.hasItems |
| import org.hamcrest.MatcherAssert.assertThat |
| import org.hamcrest.Matchers.containsInAnyOrder |
| import org.hamcrest.Matchers.notNullValue |
| import org.junit.Before |
| import org.junit.Test |
| import org.junit.runner.RunWith |
| import org.mockito.ArgumentMatchers.any |
| import org.mockito.ArgumentMatchers.anyString |
| import org.mockito.Mockito.mock |
| import org.mockito.Mockito.times |
| import org.mockito.Mockito.verify |
| import java.util.concurrent.ExecutionException |
| import java.util.concurrent.Executor |
| import java.util.concurrent.Executors |
| |
| @RunWith(AndroidJUnit4::class) |
| class TestWorkerBuilderTest { |
| |
| private lateinit var context: Context |
| private lateinit var executor: Executor |
| |
| @Before |
| fun setUp() { |
| context = ApplicationProvider.getApplicationContext() |
| executor = SynchronousExecutor() |
| } |
| |
| @Test |
| @SmallTest |
| @Throws(InterruptedException::class, ExecutionException::class) |
| fun testListenableWorkerBuilder_buildsWorker() { |
| val request = OneTimeWorkRequestBuilder<TestWorker>().build() |
| val worker = TestListenableWorkerBuilder.from(context, request).build() |
| val result = worker.startWork().get() |
| assertThat(result, `is`(Result.success())) |
| } |
| |
| @Test |
| @SmallTest |
| fun testWorkerBuilder_buildsWorker() { |
| val request = OneTimeWorkRequestBuilder<TestWorker>().build() |
| val worker = TestWorkerBuilder.from(context, request, executor).build() |
| val result = worker.doWork() |
| assertThat(result, `is`(Result.success())) |
| } |
| |
| @Test(expected = IllegalArgumentException::class) |
| @SmallTest |
| fun testWorkerBuilder_invalidWorker() { |
| val request = OneTimeWorkRequestBuilder<TestListenableWorker>().build() |
| TestWorkerBuilder.from(context, request, executor).build() |
| } |
| |
| @Test |
| @SmallTest |
| fun testBuilder() { |
| val request = OneTimeWorkRequestBuilder<TestWorker>() |
| .addTag("test") |
| .build() |
| |
| val contentUris = arrayOf(Uri.parse("android.test://1")) |
| val authorities = arrayOf("android.test") |
| |
| val worker = TestListenableWorkerBuilder.from(context, request) |
| .setRunAttemptCount(2) |
| .setTriggeredContentAuthorities(authorities.toList()) |
| .setTriggeredContentUris(contentUris.toList()) |
| .build() |
| |
| assertThat(worker.tags, hasItems("test")) |
| assertThat(worker.id, `is`(request.id)) |
| assertThat(worker.runAttemptCount, `is`(2)) |
| assertThat(worker.triggeredContentAuthorities, containsInAnyOrder(*authorities)) |
| assertThat(worker.triggeredContentUris, containsInAnyOrder(*contentUris)) |
| } |
| |
| @Test |
| @SmallTest |
| fun testWorkerBuilder_usesSingleThreadedExecutor() { |
| val request = OneTimeWorkRequestBuilder<TestWorker>().build() |
| val singleThreadedExecutor = Executors.newSingleThreadExecutor() |
| val worker = TestWorkerBuilder.from(context, request, singleThreadedExecutor).build() |
| val result = worker.doWork() |
| assertThat(result, `is`(Result.success())) |
| } |
| |
| @Test |
| @SmallTest |
| fun testWorkerBuilder_returnsExpectedType() { |
| val listenableWorker: TestListenableWorker = |
| TestListenableWorkerBuilder.from(context, TestListenableWorker::class.java).build() |
| |
| val worker: TestWorker = |
| TestWorkerBuilder.from(context, TestWorker::class.java, executor).build() |
| |
| assertThat(listenableWorker, notNullValue()) |
| assertThat(worker, notNullValue()) |
| } |
| |
| @Test |
| @SmallTest |
| fun testListenableWorkerBuilder_usesExtension() { |
| val worker = TestListenableWorkerBuilder<TestWorker>(context).build() |
| val result = worker.doWork() |
| assertThat(result, `is`(Result.success())) |
| } |
| |
| @Test |
| @SmallTest |
| fun testWorkerBuilder_usesExtension() { |
| val singleThreadedExecutor = Executors.newSingleThreadExecutor() |
| val worker = TestWorkerBuilder<TestWorker>(context, singleThreadedExecutor).build() |
| val result = worker.doWork() |
| assertThat(result, `is`(Result.success())) |
| } |
| |
| @Test |
| @MediumTest |
| fun testWorkerBuilder_usesWorkerFactory() { |
| val workerFactory = mock(WorkerFactory::class.java) |
| val worker = TestListenableWorkerBuilder<TestWorker>(context) |
| .setWorkerFactory(workerFactory) |
| .build() |
| |
| runBlocking { |
| val result = worker.startWork().await() |
| verify(workerFactory, times(1)) |
| .createWorker( |
| any(Context::class.java), anyString(), any(WorkerParameters::class.java) |
| ) |
| assertThat(result, `is`(Result.success())) |
| } |
| } |
| } |
