From 577674bff230c2eb2da58dbbb9cc779647e81efe Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Fri, 4 Aug 2023 17:51:07 +0200
Subject: [PATCH] test: Add minimal test_spi_mmap with simulation of SPIMaster.

---
 test/test_spi_mmap.py | 103 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 test/test_spi_mmap.py

diff --git a/test/test_spi_mmap.py b/test/test_spi_mmap.py
new file mode 100644
index 000000000..8915eef23
--- /dev/null
+++ b/test/test_spi_mmap.py
@@ -0,0 +1,103 @@
+#
+# This file is part of LiteX.
+#
+# Copyright (c) 2022-2023 MoTeC
+# Copyright (c) 2022-2023 Florent Kermarrec <florent@enjoy-digital.fr>
+# SPDX-License-Identifier: BSD-2-Clause
+
+import unittest
+import random
+
+from migen import *
+
+from litex.gen.sim import *
+
+from litex.soc.cores.spi.spi_mmap import SPIMaster
+
+class TestSPIMMAP(unittest.TestCase):
+    def test_spi_master(self):
+        pads = Record([("clk", 1), ("cs_n", 4), ("mosi", 1), ("miso", 1)])
+        dut  = SPIMaster(pads=pads, data_width=32, sys_clk_freq=int(100e6))
+        def generator(dut):
+            data = [
+                0x12345678,
+                0xdeadbeef,
+            ]
+            #data = [
+            #    0x80000001,
+            #    0x80000001,
+            #]
+
+            # Config: Mode0, Loopback, Sys-Clk/4
+            yield dut.loopback.eq(1)
+            yield dut.clk_divider.eq(4)
+            yield dut.mode.eq(0)
+            yield
+            yield dut.mosi.eq(data[0])
+            yield dut.cs.eq(0b0001)
+            yield dut.length.eq(32)
+            yield dut.start.eq(1)
+            yield
+            yield dut.start.eq(0)
+            while (yield dut.done) == 0b0:
+                yield
+            yield dut.cs.eq(0b0000)
+            for i in range(16):
+                yield
+            print(f"mosi_data : {(yield dut.miso):08x}")
+
+            # Config: Mode3, Loopback, Sys-Clk/4.
+            yield dut.loopback.eq(1)
+            yield dut.clk_divider.eq(4)
+            yield dut.mode.eq(3)
+            yield
+            yield dut.mosi.eq(data[0])
+            yield dut.cs.eq(0b0001)
+            yield dut.length.eq(32)
+            yield dut.start.eq(1)
+            yield
+            yield dut.start.eq(0)
+            while (yield dut.done) == 0b0:
+                yield
+            yield dut.cs.eq(0b0000)
+            for i in range(16):
+                yield
+            print(f"mosi_data : {(yield dut.miso):08x}")
+
+            # Config: Mode0, Loopback, Sys-Clk/8.
+            yield dut.loopback.eq(1)
+            yield dut.clk_divider.eq(8)
+            yield dut.mode.eq(0)
+            yield
+            yield dut.mosi.eq(data[1])
+            yield dut.cs.eq(0b0001)
+            yield dut.length.eq(32)
+            yield dut.start.eq(1)
+            yield
+            yield dut.start.eq(0)
+            while (yield dut.done) == 0b0:
+                yield
+            yield dut.cs.eq(0b0000)
+            for i in range(16):
+                yield
+            print(f"mosi_data : {(yield dut.miso):08x}")
+
+            # Config: Mode3, Loopback, Sys-Clk/8.
+            yield dut.loopback.eq(1)
+            yield dut.clk_divider.eq(8)
+            yield dut.mode.eq(3)
+            yield
+            yield dut.mosi.eq(data[1])
+            yield dut.cs.eq(0b0001)
+            yield dut.length.eq(32)
+            yield dut.start.eq(1)
+            yield
+            yield dut.start.eq(0)
+            while (yield dut.done) == 0b0:
+                yield
+            yield dut.cs.eq(0b0000)
+            for i in range(16):
+                yield
+            print(f"mosi_data : {(yield dut.miso):08x}")
+
+        run_simulation(dut, generator(dut), vcd_name="sim.vcd")