# -*- coding: utf-8 -*-
"""ATA over Ethernet ATA command"""
from __future__ import print_function
from __future__ import absolute_import

from . import dpkt

ATA_DEVICE_IDENTIFY = 0xec


class AOEATA(dpkt.Packet):
    """ATA over Ethernet ATA command.

    See more about the AOEATA on
    https://en.wikipedia.org/wiki/ATA_over_Ethernet

    Attributes:
        __hdr__: Header fields of AOEATA.
        data: Message data.
    """

    __hdr__ = (
        ('aflags', 'B', 0),
        ('errfeat', 'B', 0),
        ('scnt', 'B', 0),
        ('cmdstat', 'B', ATA_DEVICE_IDENTIFY),
        ('lba0', 'B', 0),
        ('lba1', 'B', 0),
        ('lba2', 'B', 0),
        ('lba3', 'B', 0),
        ('lba4', 'B', 0),
        ('lba5', 'B', 0),
        ('res', 'H', 0),
    )

    # XXX: in unpack, switch on ATA command like icmp does on type


def test_aoeata():
    s = (b'\x03\x0a\x6b\x19\x00\x00\x00\x00\x45\x00\x00\x28\x94\x1f\x00\x00\xe3\x06\x99\xb4\x23\x2b'
         b'\x24\x00\xde\x8e\x84\x42\xab\xd1\x00\x50\x00\x35\xe1\x29\x20\xd9\x00\x00\x00\x22\x9b\xf0\xe2\x04\x65\x6b')
    aoeata = AOEATA(s)
    assert (bytes(aoeata) == s)


if __name__ == '__main__':
    test_aoeata()
    print('Tests Successful...')
