3_usb_to_can.py

import sys
import time
import pytest
from flaky import flaky

from panda import Panda
from panda.tests.hitl.conftest import SPEED_NORMAL, SPEED_GMLAN, PandaGroup
from panda.tests.hitl.helpers import time_many_sends

def test_can_loopback(p):
  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)
  p.set_can_loopback(True)

  for bus in (0, 1, 2):
    # set bus 0 speed to 5000
    p.set_can_speed_kbps(bus, 500)

    # send a message on bus 0
    p.can_send(0x1aa, b"message", bus)

    # confirm receive both on loopback and send receipt
    time.sleep(0.05)
    r = p.can_recv()
    sr = [x for x in r if x[3] == 0x80 | bus]
    lb = [x for x in r if x[3] == bus]
    assert len(sr) == 1
    assert len(lb) == 1

    # confirm data is correct
    assert 0x1aa == sr[0][0] == lb[0][0]
    assert b"message" == sr[0][2] == lb[0][2]

def test_reliability(p):
  MSG_COUNT = 100

  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)
  p.set_can_loopback(True)
  p.set_can_speed_kbps(0, 1000)

  addrs = list(range(100, 100 + MSG_COUNT))
  ts = [(j, 0, b"\xaa" * 8, 0) for j in addrs]

  for _ in range(100):
    st = time.monotonic()

    p.can_send_many(ts)

    r = []
    while len(r) < 200 and (time.monotonic() - st) < 0.5:
      r.extend(p.can_recv())

    sent_echo = [x for x in r if x[3] == 0x80]
    loopback_resp = [x for x in r if x[3] == 0]

    assert sorted([x[0] for x in loopback_resp]) == addrs
    assert sorted([x[0] for x in sent_echo]) == addrs
    assert len(r) == 200

    # take sub 20ms
    et = (time.monotonic() - st) * 1000.0
    assert et < 20

    sys.stdout.write("P")
    sys.stdout.flush()

@flaky(max_runs=6, min_passes=1)
def test_throughput(p):
  # enable output mode
  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)

  # enable CAN loopback mode
  p.set_can_loopback(True)

  for speed in [10, 20, 50, 100, 125, 250, 500, 1000]:
    # set bus 0 speed to speed
    p.set_can_speed_kbps(0, speed)
    time.sleep(0.05)

    comp_kbps = time_many_sends(p, 0)

    # bit count from https://en.wikipedia.org/wiki/CAN_bus
    saturation_pct = (comp_kbps / speed) * 100.0
    assert saturation_pct > 80
    assert saturation_pct < 100

    print("loopback 100 messages at speed %d, comp speed is %.2f, percent %.2f" % (speed, comp_kbps, saturation_pct))

@pytest.mark.test_panda_types(PandaGroup.GMLAN)
def test_gmlan(p):
  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)
  p.set_can_loopback(True)

  p.set_can_speed_kbps(1, SPEED_NORMAL)
  p.set_can_speed_kbps(2, SPEED_NORMAL)
  p.set_can_speed_kbps(3, SPEED_GMLAN)

  # set gmlan on CAN2
  for bus in [Panda.GMLAN_CAN2, Panda.GMLAN_CAN3, Panda.GMLAN_CAN2, Panda.GMLAN_CAN3]:
    p.set_gmlan(bus)
    comp_kbps_gmlan = time_many_sends(p, 3)
    assert comp_kbps_gmlan > (0.8 * SPEED_GMLAN)
    assert comp_kbps_gmlan < (1.0 * SPEED_GMLAN)

    p.set_gmlan(None)
    comp_kbps_normal = time_many_sends(p, bus)
    assert comp_kbps_normal > (0.8 * SPEED_NORMAL)
    assert comp_kbps_normal < (1.0 * SPEED_NORMAL)

    print("%d: %.2f kbps vs %.2f kbps" % (bus, comp_kbps_gmlan, comp_kbps_normal))

@pytest.mark.test_panda_types(PandaGroup.GMLAN)
def test_gmlan_bad_toggle(p):
  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)
  p.set_can_loopback(True)

  # GMLAN_CAN2
  for bus in [Panda.GMLAN_CAN2, Panda.GMLAN_CAN3]:
    p.set_gmlan(bus)
    comp_kbps_gmlan = time_many_sends(p, 3)
    assert comp_kbps_gmlan > (0.6 * SPEED_GMLAN)
    assert comp_kbps_gmlan < (1.0 * SPEED_GMLAN)

  # normal
  for bus in [Panda.GMLAN_CAN2, Panda.GMLAN_CAN3]:
    p.set_gmlan(None)
    comp_kbps_normal = time_many_sends(p, bus)
    assert comp_kbps_normal > (0.6 * SPEED_NORMAL)
    assert comp_kbps_normal < (1.0 * SPEED_NORMAL)


# this will fail if you have hardware serial connected
def test_serial_debug(p):
  _ = p.serial_read(Panda.SERIAL_DEBUG)  # junk
  p.call_control_api(0x01)
  assert p.serial_read(Panda.SERIAL_DEBUG).startswith(b"NO HANDLER")