aboutsummaryrefslogtreecommitdiff
path: root/shadows/framework/src/main/java/org/robolectric/shadows/ShadowAmbientContextManager.java
blob: aac2e77f82efaf74de07343458693e59720d0529 (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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package org.robolectric.shadows;

import android.annotation.Nullable;
import android.app.PendingIntent;
import android.app.ambientcontext.AmbientContextEventRequest;
import android.app.ambientcontext.AmbientContextManager;
import android.os.Build.VERSION_CODES;
import com.android.internal.annotations.GuardedBy;
import java.util.Set;
import java.util.concurrent.Executor;
import java.util.function.Consumer;
import org.robolectric.annotation.Implementation;
import org.robolectric.annotation.Implements;

/** Shadow of {@link AmbientContextManager} */
@Implements(
    value = AmbientContextManager.class,
    minSdk = VERSION_CODES.TIRAMISU,
    isInAndroidSdk = false)
public class ShadowAmbientContextManager {

  private final Object lock = new Object();

  /**
   * Caches the last {@link AmbientContextEventRequest} passed into {@link
   * #registerObserver(AmbientContextEventRequest, PendingIntent, Executor, Consumer)}
   *
   * <p>If {@link #unregisterObserver()} was called after {@link
   * #registerObserver(AmbientContextEventRequest, PendingIntent, Executor, Consumer)}, this is set
   * to null.
   */
  @GuardedBy("lock")
  @Nullable
  private AmbientContextEventRequest lastRegisterObserverRequest;

  /**
   * The ambient context service status code that will be consumed by the {@code consumer} which is
   * passed in {@link #queryAmbientContextServiceStatus(Set, Executor, Consumer)} or {@link
   * #registerObserver(AmbientContextEventRequest, PendingIntent, Executor, Consumer)}.
   */
  @GuardedBy("lock")
  private Integer ambientContextServiceStatus = AmbientContextManager.STATUS_NOT_SUPPORTED;

  /** Caches the last requested event codes passed into {@link #startConsentActivity(Set)}. */
  @GuardedBy("lock")
  @Nullable
  private Set<Integer> lastRequestedEventCodesForConsentActivity;

  @Implementation
  protected void registerObserver(
      AmbientContextEventRequest request,
      PendingIntent resultPendingIntent,
      Executor executor,
      Consumer<Integer> statusConsumer) {
    synchronized (lock) {
      lastRegisterObserverRequest = request;
      statusConsumer.accept(ambientContextServiceStatus);
    }
  }

  @Implementation
  protected void unregisterObserver() {
    synchronized (lock) {
      lastRegisterObserverRequest = null;
    }
  }

  /**
   * Returns the last {@link AmbientContextEventRequest} passed into {@link
   * AmbientContextManager#registerObserver(AmbientContextEventRequest, PendingIntent, Executor,
   * Consumer)}.
   *
   * <p>Returns null if {@link AmbientContextManager#unregisterObserver()} is invoked or there is no
   * invocation of {@link AmbientContextManager#registerObserver(AmbientContextEventRequest,
   * PendingIntent, Executor, Consumer)}.
   */
  @Nullable
  public AmbientContextEventRequest getLastRegisterObserverRequest() {
    synchronized (lock) {
      return lastRegisterObserverRequest;
    }
  }

  @Implementation
  protected void queryAmbientContextServiceStatus(
      Set<Integer> eventTypes, Executor executor, Consumer<Integer> consumer) {
    synchronized (lock) {
      consumer.accept(ambientContextServiceStatus);
    }
  }

  /**
   * Sets a {@code status} that will be consumed by the {@code consumer} which is passed in {@link
   * #queryAmbientContextServiceStatus(Set, Executor, Consumer)} or {@link
   * #registerObserver(AmbientContextEventRequest, PendingIntent, Executor, Consumer)}.
   */
  public void setAmbientContextServiceStatus(Integer status) {
    synchronized (lock) {
      ambientContextServiceStatus = status;
    }
  }

  @Implementation
  protected void startConsentActivity(Set<Integer> eventTypes) {
    synchronized (lock) {
      lastRequestedEventCodesForConsentActivity = eventTypes;
    }
  }

  /**
   * Returns the last requested event codes that were passed into {@link
   * #startConsentActivity(Set)}.
   *
   * <p>If {@link #startConsentActivity(Set)} is never invoked, returns {@code null}.
   */
  @Nullable
  public Set<Integer> getLastRequestedEventCodesForConsentActivity() {
    synchronized (lock) {
      return lastRequestedEventCodesForConsentActivity;
    }
  }
}