From 5bee9e951fb6868e0e6b7da9c0f9e6ea651a1007 Mon Sep 17 00:00:00 2001
From: Harry Ho <hh@m-labs.hk>
Date: Mon, 6 Jan 2020 10:26:28 +0800
Subject: [PATCH] wishbone: add a simple test for InterconnectShared

---
 nmigen_soc/test/test_wishbone_bus.py | 85 +++++++++++++++++++++++++++-
 1 file changed, 84 insertions(+), 1 deletion(-)

diff --git a/nmigen_soc/test/test_wishbone_bus.py b/nmigen_soc/test/test_wishbone_bus.py
index 8d7064d..7d23283 100644
--- a/nmigen_soc/test/test_wishbone_bus.py
+++ b/nmigen_soc/test/test_wishbone_bus.py
@@ -5,7 +5,7 @@ from nmigen import *
 from nmigen.hdl.rec import *
 from nmigen.back.pysim import *
 
-from ..wishbone.bus import *
+from ..wishbone import *
 
 
 class InterfaceTestCase(unittest.TestCase):
@@ -296,3 +296,86 @@ class DecoderSimulationTestCase(unittest.TestCase):
         with Simulator(m, vcd_file=open("test.vcd", "w")) as sim:
             sim.add_process(sim_test())
             sim.run()
+
+
+class InterconnectSharedSimulationTestCase(unittest.TestCase):
+    def setUp(self):
+        self.shared = Interface(addr_width=30,
+                                data_width=32,
+                                granularity=8,
+                                features={"err","cti","bte"},
+                                name="shared")
+        self.master01 = Interface(addr_width=30,
+                                  data_width=32,
+                                  granularity=8,
+                                  features={"err","cti","bte"},
+                                  name="master01")
+        self.master02 = Record([
+            ("adr",   30, DIR_FANOUT),
+            ("dat_w", 32, DIR_FANOUT),
+            ("dat_r", 32, DIR_FANIN),
+            ("sel",    4, DIR_FANOUT),
+            ("cyc",    1, DIR_FANOUT),
+            ("stb",    1, DIR_FANOUT),
+            ("ack",    1, DIR_FANIN),
+            ("we",     1, DIR_FANOUT),
+            ("cti",    3, DIR_FANOUT),
+            ("bte",    2, DIR_FANOUT),
+            ("err",    1, DIR_FANIN)
+        ])
+        self.ram = SRAM(Memory(width=32, depth=2048, init=[]),
+                        granularity=8, features={"err","cti","bte"})
+        self.sub01 = Interface(addr_width=21,
+                               data_width=32,
+                               granularity=8,
+                               features={"err","cti","bte"},
+                               name="sub01")
+        self.dut = InterconnectShared(
+            shared_bus=self.shared,
+            masters=[
+                self.master01,
+                self.master02
+            ],
+            targets=[
+                (self.ram.bus, 0), 
+                (self.sub01, (2**21) << 2)
+            ]
+        )
+
+    def test_basic(self):
+        def sim_test():
+            yield self.master01.adr.eq(0)
+            yield self.master02.adr.eq(2**21)
+            yield self.master01.we.eq(0)
+            yield self.master02.we.eq(0)
+            #
+            for _ in range(5):
+                yield self.master01.cyc.eq(1)
+                yield self.master02.cyc.eq(1)
+                yield 
+                ram_cyc = (yield self.ram.bus.cyc)
+                sub01_cyc = (yield self.sub01.cyc)
+                if ram_cyc == 1:
+                    yield self.master01.stb.eq(1)
+                    yield
+                    yield self.ram.bus.ack.eq(1)
+                    yield self.master01.stb.eq(0)
+                    yield
+                    yield self.ram.bus.ack.eq(0)
+                    yield self.master01.cyc.eq(0)
+                elif sub01_cyc == 1:
+                    yield self.master02.stb.eq(1)
+                    yield
+                    yield self.sub01.ack.eq(1)
+                    yield self.master02.stb.eq(0)
+                    yield
+                    yield self.sub01.ack.eq(0)
+                    yield self.master02.cyc.eq(0)
+                yield
+
+        m = Module()
+        m.submodules += self.dut, self.ram
+        with Simulator(m, vcd_file=open("test.vcd", "w")) as sim:
+            sim.add_clock(1e-6)
+            sim.add_sync_process(sim_test())
+            sim.run()
-- 
2.30.2