aboutsummaryrefslogtreecommitdiff
path: root/robolectric/src/test/java/org/robolectric/shadows/ShadowPausedAsyncTaskLoaderTest.java
blob: b524f382d3cef2ff2ba579dc766ce6042944a10a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package org.robolectric.shadows;

import static com.google.common.truth.Truth.assertThat;

import android.content.AsyncTaskLoader;
import androidx.test.core.app.ApplicationProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import java.util.ArrayList;
import java.util.List;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.android.util.concurrent.PausedExecutorService;
import org.robolectric.shadow.api.Shadow;

/** Tests for {@link ShadowPausedAsyncTaskLoader}. */
@RunWith(AndroidJUnit4.class)
public class ShadowPausedAsyncTaskLoaderTest {
  private final List<String> taskRecord = new ArrayList<>();
  private TestLoader testLoader;
  private PausedExecutorService pausedBackgroundExecutor;

  @Before
  public void setUp() {
    pausedBackgroundExecutor = new PausedExecutorService();
    testLoader = new TestLoader(42);
    ShadowPausedAsyncTaskLoader<Integer> shadowLoader = Shadow.extract(testLoader);
    shadowLoader.setExecutor(pausedBackgroundExecutor);
  }

  @Test
  public void forceLoad_enqueuesWork() {
    testLoader.forceLoad();
    assertThat(taskRecord).isEmpty();

    pausedBackgroundExecutor.runAll();
    assertThat(taskRecord).containsExactly("loadInBackground");
    taskRecord.clear();

    ShadowLooper.idleMainLooper();
    assertThat(taskRecord).containsExactly("deliverResult 42");
  }

  @Test
  public void forceLoad_multipleLoads() {
    testLoader.forceLoad();
    assertThat(taskRecord).isEmpty();

    pausedBackgroundExecutor.runAll();
    assertThat(taskRecord).containsExactly("loadInBackground");
    taskRecord.clear();

    ShadowLooper.idleMainLooper();
    assertThat(taskRecord).containsExactly("deliverResult 42");

    testLoader.setData(43);
    taskRecord.clear();
    testLoader.forceLoad();

    pausedBackgroundExecutor.runAll();
    assertThat(taskRecord).containsExactly("loadInBackground");
    taskRecord.clear();

    ShadowLooper.idleMainLooper();
    assertThat(taskRecord).containsExactly("deliverResult 43");
  }

  class TestLoader extends AsyncTaskLoader<Integer> {
    private Integer data;

    public TestLoader(Integer data) {
      super(ApplicationProvider.getApplicationContext());
      this.data = data;
    }

    @Override
    public Integer loadInBackground() {
      taskRecord.add("loadInBackground");
      return data;
    }

    @Override
    public void deliverResult(Integer data) {
      taskRecord.add("deliverResult " + data);
    }

    public void setData(int newData) {
      this.data = newData;
    }
  }
}