From 7eb368fd5dbfcbea806259a5219608df56d46d84 Mon Sep 17 00:00:00 2001
From: whitequark <whitequark@whitequark.org>
Date: Fri, 17 Mar 2017 14:47:58 +0000
Subject: [PATCH] test: add DMA test that checks the analyzer trace.

---
 artiq/test/coredevice/test_rtio.py | 31 ++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/artiq/test/coredevice/test_rtio.py b/artiq/test/coredevice/test_rtio.py
index 09f80f4eb..64c478c4d 100644
--- a/artiq/test/coredevice/test_rtio.py
+++ b/artiq/test/coredevice/test_rtio.py
@@ -8,6 +8,8 @@ from math import sqrt
 from artiq.experiment import *
 from artiq.test.hardware_testbench import ExperimentCase
 from artiq.coredevice import exceptions
+from artiq.coredevice.comm_analyzer import (StoppedMessage, OutputMessage, InputMessage,
+                                            decode_dump, get_analyzer_dump)
 
 
 artiq_low_latency = os.getenv("ARTIQ_LOW_LATENCY")
@@ -467,20 +469,21 @@ class _DMA(EnvExperiment):
     def build(self, trace_name="foobar"):
         self.setattr_device("core")
         self.setattr_device("core_dma")
-        self.setattr_device("ttl0")
+        self.setattr_device("ttl1")
         self.trace_name = trace_name
 
     @kernel
     def record(self):
         with self.core_dma.record(self.trace_name):
             delay(100*ns)
-            self.ttl0.on()
+            self.ttl1.on()
             delay(100*ns)
-            self.ttl0.off()
+            self.ttl1.off()
 
     @kernel
     def replay(self):
         self.core.break_realtime()
+        delay(100*ms)
         self.core_dma.replay(self.trace_name)
 
     @kernel
@@ -509,4 +512,24 @@ class DMATest(ExperimentCase):
         with self.assertRaises(exceptions.DMAError):
             exp.nested()
 
-    # TODO: check replay against analyzer
+    def test_dma_trace(self):
+        core_host = self.device_mgr.get_desc("comm")["arguments"]["host"]
+
+        exp = self.create(_DMA)
+        exp.record()
+        get_analyzer_dump(core_host)  # clear analyzer buffer
+        exp.replay()
+
+        dump = decode_dump(get_analyzer_dump(core_host))
+        self.assertEqual(len(dump.messages), 3)
+        self.assertIsInstance(dump.messages[-1], StoppedMessage)
+        self.assertIsInstance(dump.messages[0], OutputMessage)
+        self.assertEqual(dump.messages[0].channel, 1)
+        self.assertEqual(dump.messages[0].address, 0)
+        self.assertEqual(dump.messages[0].data, 1)
+        self.assertIsInstance(dump.messages[1], OutputMessage)
+        self.assertEqual(dump.messages[1].channel, 1)
+        self.assertEqual(dump.messages[1].address, 0)
+        self.assertEqual(dump.messages[1].data, 0)
+        self.assertEqual(dump.messages[1].timestamp -
+                         dump.messages[0].timestamp, 100)