# Copyright 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import common
from autotest_lib.client.cros.cellular.mbim_compliance import mbim_constants
from autotest_lib.client.cros.cellular.mbim_compliance import mbim_data_transfer
from autotest_lib.client.cros.cellular.mbim_compliance import mbim_dts_test_base
from autotest_lib.client.cros.cellular.mbim_compliance import mbim_errors
from autotest_lib.client.cros.cellular.mbim_compliance.sequences \
        import loopback_sequence


class cellular_MbimComplianceDTS0410(mbim_dts_test_base.MbimDtsTestBase):
    """
    Validation of wSequence after function reset.

    This test verifies that function reset properly re-initializes the sequence
    number.

    Reference:
        [1] Universal Serial Bus Communication Class MBIM Compliance Testing: 28
        http://www.usb.org/developers/docs/devclass_docs/MBIM-Compliance-1.0.pdf

    """
    version = 1

    def run_internal(self, ntb_format):
        """
        Run DTS_04/DTS_10 test.

        @param ntb_format: Whether to send/receive an NTB16 or NTB32 frame.
                Possible values: NTB_FORMAT_16, NTB_FORMAT_32 (mbim_constants)

        """
        # Precondition
        _, open_sequence, connect_sequence = self.run_precondition(ntb_format)

        # Step 1
        loopback = loopback_sequence.LoopbackSequence(self.device_context)
        _, _, _, _ = loopback.run(ntb_format=ntb_format)

        # Step 2
        open_sequence.run(ntb_format=ntb_format)
        connect_sequence.run()
        mbim_data_transfer.MBIMNtb.reset_sequence_number()

        # Step 3
        nth, _, _, _ = loopback.run(ntb_format=ntb_format)

        # Step 4
        if ntb_format == mbim_constants.NTB_FORMAT_16:
            if nth.sequence_number != 0:
                mbim_errors.log_and_raise(
                        mbim_errors.MBIMComplianceAssertionError,
                        'ncm1.0:3.2.1#3')
        else:
            if nth.sequence_number != 0:
                mbim_errors.log_and_raise(
                        mbim_errors.MBIMComplianceAssertionError,
                        'ncm1.0:3.2.2#3')