package com.mapbox.mapboxsdk.location;

import android.hardware.Sensor;
import android.hardware.SensorEventListener;
import android.hardware.SensorManager;
import android.view.WindowManager;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;

import static com.mapbox.mapboxsdk.location.LocationComponentCompassEngine.SENSOR_DELAY_MICROS;
import static junit.framework.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class CompassEngineTest {

  private LocationComponentCompassEngine compassEngine;

  @Mock
  private WindowManager windowManager;

  @Mock
  private SensorManager sensorManager;

  @Mock
  private Sensor compassSensor;

  @Mock
  private CompassListener compassListener;

  @Before
  public void setUp() throws Exception {
    when(sensorManager.getDefaultSensor(Sensor.TYPE_ROTATION_VECTOR)).thenReturn(compassSensor);
    compassEngine = new LocationComponentCompassEngine(windowManager, sensorManager);
  }

  @Test
  public void lastKnownCompassBearingAccuracyDefault() {
    assertEquals("Last accuracy should match", compassEngine.getLastAccuracySensorStatus(), 0);
  }

  @Test
  public void lastKnownCompassAccuracyStatusValue() {
    Sensor sensor = mock(Sensor.class);
    compassEngine.onAccuracyChanged(sensor, 2);
    assertEquals("Last accuracy should match", compassEngine.getLastAccuracySensorStatus(), 2);
  }

  @Test
  public void whenGyroscopeIsNull_fallbackToGravity() {
    SensorManager sensorManager = mock(SensorManager.class);
    when(sensorManager.getDefaultSensor(Sensor.TYPE_GYROSCOPE)).thenReturn(null);
    new LocationComponentCompassEngine(windowManager, sensorManager);

    verify(sensorManager, times(1)).getDefaultSensor(Sensor.TYPE_ACCELEROMETER);
  }

  @Test
  public void whenGyroscopeIsNull_fallbackToMagneticField() {
    SensorManager sensorManager = mock(SensorManager.class);
    when(sensorManager.getDefaultSensor(Sensor.TYPE_GYROSCOPE)).thenReturn(null);
    new LocationComponentCompassEngine(windowManager, sensorManager);

    verify(sensorManager, times(1)).getDefaultSensor(Sensor.TYPE_MAGNETIC_FIELD);
  }

  @Test
  public void listener_registerOnAdd() {
    compassEngine.addCompassListener(compassListener);
    verify(sensorManager)
      .registerListener(any(SensorEventListener.class), eq(compassSensor), eq(SENSOR_DELAY_MICROS));
  }

  @Test
  public void listener_unregisterOnRemove() {
    compassEngine.addCompassListener(compassListener);
    compassEngine.removeCompassListener(compassListener);
    verify(sensorManager).unregisterListener(any(SensorEventListener.class), eq(compassSensor));
  }
}