import pytest
import pynmea2

import datetime

from decimal import Decimal

def test_GGA():
    data = "$GPGGA,184353.07,1929.045,S,02410.506,E,1,04,2.6,100.00,M,-33.9,M,,0000*6D"
    msg = pynmea2.parse(data)
    assert msg.talker == 'GP'
    assert msg.sentence_type == 'GGA'
    assert isinstance(msg, pynmea2.GGA)

    # Timestamp
    assert msg.timestamp        == datetime.time(18, 43, 53, 70000, tzinfo=datetime.timezone.utc)
    # Latitude
    assert msg.lat              == '1929.045'
    # Latitude Direction
    assert msg.lat_dir          == 'S'
    # Longitude
    assert msg.lon              == '02410.506'
    # Longitude Direction
    assert msg.lon_dir          == 'E'
    # GPS Quality Indicator
    assert msg.gps_qual         == 1
    # Number of Satellites in use
    assert msg.num_sats         == '04'
    # Horizontal Dilution of Precision
    assert msg.horizontal_dil   == '2.6'
    # Antenna Alt above sea level (mean)
    assert msg.altitude         == 100.0
    # Units of altitude (meters)
    assert msg.altitude_units   == 'M'
    # Geoidal Separation
    assert msg.geo_sep          == '-33.9'
    # Units of Geoidal Separation (meters)
    assert msg.geo_sep_units    == 'M'
    # Age of Differential GPS Data (secs)
    assert msg.age_gps_data     == ''
    # Differential Reference Station ID
    assert msg.ref_station_id   == '0000'
    assert msg.is_valid == True

    msg.altitude = 200.0
    assert str(msg) == "$GPGGA,184353.07,1929.045,S,02410.506,E,1,04,2.6,200.0,M,-33.9,M,,0000*5E"

def test_RTE():
    data = "$GPRTE,2,1,c,0,PBRCPK,PBRTO,PTELGR,PPLAND,PYAMBU,PPFAIR,PWARRN,PMORTL,PLISMR*73"
    msg = pynmea2.parse(data)
    assert msg.talker == 'GP'
    assert msg.sentence_type == 'RTE'
    assert "2" == msg.num_in_seq
    assert "1" == msg.sen_num
    assert "c" == msg.start_type
    assert "0" == msg.active_route_id
    assert msg.waypoint_list == [
        "PBRCPK", "PBRTO", "PTELGR", "PPLAND", "PYAMBU",
        "PPFAIR", "PWARRN", "PMORTL", "PLISMR"]

    msg.waypoint_list = ['ABC', 'DEF']
    assert str(msg) == "$GPRTE,2,1,c,0,ABC,DEF*03"

def test_R00():
    data = "$GPR00,A,B,C*29"
    msg = pynmea2.parse(data)
    assert msg.talker == 'GP'
    assert msg.sentence_type == 'R00'
    assert msg.waypoint_list == ['A', 'B', 'C']

    msg.waypoint_list = ['ABC', 'DEF']
    assert str(msg) == "$GPR00,ABC,DEF*42"

def test_MWV():
    data = "$IIMWV,271.0,R,000.2,N,A*3B"
    msg = pynmea2.parse(data)

    assert msg.talker == 'II'
    assert msg.sentence_type == 'MWV'

    # Wind angle in degrees
    assert msg.wind_angle == Decimal('271.0')

    # Reference type
    assert msg.reference == 'R'

    # Wind speed
    assert msg.wind_speed == Decimal('0.2')

    # Wind speed units
    assert msg.wind_speed_units == 'N'

    # Device status
    assert msg.status == 'A'
    assert msg.render() == data


def test_GST():
    data = "$GPGST,172814.0,0.006,0.023,0.020,273.6,0.023,0.020,0.031*6A"
    msg = pynmea2.parse(data)
    assert isinstance(msg, pynmea2.GST)
    assert msg.timestamp == datetime.time(hour=17, minute=28, second=14, tzinfo=datetime.timezone.utc)
    assert msg.rms == 0.006
    assert msg.std_dev_major == 0.023
    assert msg.std_dev_minor == 0.020
    assert msg.orientation == 273.6
    assert msg.std_dev_latitude == 0.023
    assert msg.std_dev_longitude == 0.020
    assert msg.std_dev_altitude == 0.031
    assert msg.render() == data


def test_RMC():
    data = '''$GPRMC,225446,A,4916.45,N,12311.12,W,000.5,054.7,191194,020.3,E*68'''
    msg = pynmea2.parse(data)
    assert isinstance(msg, pynmea2.RMC)
    assert msg.timestamp == datetime.time(hour=22, minute=54, second=46, tzinfo=datetime.timezone.utc)
    assert msg.datestamp == datetime.date(1994, 11, 19)
    assert msg.latitude == 49.274166666666666
    assert msg.longitude == -123.18533333333333
    assert msg.datetime == datetime.datetime(1994, 11, 19, 22, 54, 46, tzinfo=datetime.timezone.utc)
    assert msg.is_valid == True
    assert msg.render() == data


def test_RMC_valid():
    '''The RMC mode indicator and navigation status values are optional.
    Test that when supplied the whole message must be valid. When not supplied
    only test validation against supplied values.

    Supplied means that a `,` exists it does NOT mean that a value had to be
    supplied in the space provided. See

    https://orolia.com/manuals/VSP/Content/NC_and_SS/Com/Topics/APPENDIX/NMEA_RMCmess.htm

    for more information about the RMC Message additions.
    '''
    msgs = [
        # Original
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,*33',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,*24',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,*72',

        # RMC Timing Messages
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,S*4C',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,N*51',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,*1F',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,S*5B',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,N*46',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,*08',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,S*0D',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,N*10',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,*5E',

        # RMC Nav Messags
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,S,S*33',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,S,V*36',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,S,*60',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,N,A*3C',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,N,V*2B',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,N,*7D',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,,A*72',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,,V*65',
        '$GPRMC,123519.00,A,4807.038,N,01131.000,E,,,230394,,,,*33',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,S,A*36',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,S,V*21',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,S,*77',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,N,A*2B',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,N,V*3C',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,N,*6A',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,,A*65',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,,V*72',
        '$GPRMC,123519.00,V,4807.038,N,01131.000,E,,,230394,,,,*24',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,S,A*60',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,S,V*77',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,S,*21',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,N,A*7D',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,N,V*6A',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,N,*3C',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,,A*33',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,,V*24',
        '$GPRMC,123519.00,,4807.038,N,01131.000,E,,,230394,,,,*72',
    ]

    # only the first of each section is valid
    expected = [False] * 39
    expected[0] = True
    expected[3] = True
    expected[12] = True

    for i, msg in enumerate(msgs):
        parsed = pynmea2.parse(msg)
        assert expected[i] == parsed.is_valid


def test_TXT():
    data = '$GNTXT,01,01,02,ROM BASE 2.01 (75331) Oct 29 2013 13:28:17*44'
    msg = pynmea2.parse(data)
    assert type(msg) == pynmea2.talker.TXT
    assert msg.text == 'ROM BASE 2.01 (75331) Oct 29 2013 13:28:17'


def test_ZDA():
    data = '''$GPZDA,010203.05,06,07,2008,-08,30'''
    msg = pynmea2.parse(data)
    assert isinstance(msg, pynmea2.ZDA)
    assert msg.timestamp == datetime.time(hour=1, minute=2, second=3, microsecond=50000, tzinfo=datetime.timezone.utc)
    assert msg.day == 6
    assert msg.month == 7
    assert msg.year == 2008
    assert msg.tzinfo.utcoffset(0) == datetime.timedelta(hours=-8, minutes=30)
    assert msg.local_zone == -8
    assert msg.local_zone_minutes == 30
    assert msg.datestamp == datetime.date(2008, 7, 6)
    assert msg.datetime == datetime.datetime(2008, 7, 6, 1, 2, 3, 50000, tzinfo=datetime.timezone.utc)
    assert msg.localdatetime == datetime.datetime(2008, 7, 5, 17, 32, 3, 50000, tzinfo=msg.tzinfo)

def test_VPW():
    data = "$XXVPW,1.2,N,3.4,M"
    msg = pynmea2.parse(data)
    assert isinstance(msg, pynmea2.VPW)
    assert msg.talker == 'XX'
    assert msg.speed_kn == 1.2
    assert msg.unit_knots == 'N'
    assert msg.speed_ms == 3.4
    assert msg.unit_ms == 'M'

def test_BOD():
    data = "XXBOD,045.,T,023.,M,DEST,START"
    msg = pynmea2.parse(data)
    assert isinstance(msg, pynmea2.BOD)
    assert msg.talker == 'XX'


def test_XDR():
    data = "$YXXDR,A,-64.437,M,N,A,054.454,D,E,C,17.09,C,T-N1052*46"
    msg = pynmea2.parse(data)
    assert isinstance(msg, pynmea2.XDR)
    assert msg.talker == 'YX'
    assert msg.type == 'A'
    assert msg.value == '-64.437'
    assert msg.units == 'M'
    assert msg.id == 'N'

    assert msg.num_transducers == 3

    t0 = msg.get_transducer(0)
    assert t0.type == 'A'
    assert t0.value == '-64.437'
    assert t0.units == 'M'
    assert t0.id == 'N'

    t1 = msg.get_transducer(1)
    assert t1.type == 'A'
    assert t1.value == '054.454'
    assert t1.units == 'D'
    assert t1.id == 'E'

    t2 = msg.get_transducer(2)
    assert t2.type == 'C'
    assert t2.value == '17.09'
    assert t2.units == 'C'
    assert t2.id == 'T-N1052'

def test_GLL():
    data = "$GPGLL,4916.45,N,12311.12,W,225444,A,*1D"
    msg = pynmea2.parse(data)
    assert msg.is_valid == True
    assert msg.render() == data

def test_GSA():
    data = "$GPGSA,A,3,02,,,07,,09,24,26,,,,,1.6,1.6,1.0*3D"
    msg = pynmea2.parse(data)
    assert msg.is_valid == True
    assert msg.render() == data

def test_VBW():
    data = "XXVBW,1.2,3.4,A,5.6,7.8,A"
    msg = pynmea2.parse(data)
    assert msg.is_valid == True
    assert msg.render(checksum=False, dollar=False) == data

def test_STALK():
    data = "$STALK,9C,C1,2A,E5*4A"
    msg = pynmea2.parse(data)
    assert msg.render() == data
    assert msg.command_name == 'Compass heading and Rudder position'

def test_STALK_unidentified_command():
    data = "$STALK,AA,C1,2A,E5*30"
    msg = pynmea2.parse(data)
    assert msg.render() == data
    assert msg.command_name == 'Unknown Command'

def test_GRS():
    data = "$GNGRS,162047.00,1,0.6,0.1,-16.6,-0.8,-0.1,0.5,,,,,,*41"
    msg = pynmea2.parse(data)
    assert msg.render() == data
    assert msg.talker == 'GN'
    assert msg.sentence_type == 'GRS'
    assert msg.residuals_mode == 1
    assert msg.sv_res_01 == 0.6
    assert msg.sv_res_02 == 0.1
    assert msg.sv_res_03 == -16.6
    assert msg.sv_res_04 == -0.8
    assert msg.sv_res_05 == -0.1
    assert msg.sv_res_06 == 0.5
    assert msg.sv_res_07 == None