1 files changed, 35 insertions, 5 deletions
diff --git a/src/tb/tb_keywrap_mkmif.v b/src/tb/tb_keywrap_mkmif.v
index 8bd3a05..e917c2c 100644
--- a/src/tb/tb_keywrap_mkmif.v
+++ b/src/tb/tb_keywrap_mkmif.v
@@ -182,9 +182,9 @@ module tb_keywrap_mkm();
       tc_ctr = tc_ctr + 1;
 
       $display("TEST AUTO-LOAD START");
-      $display("Check that correct ICV is generated for a single block message.");
+      $display("Check that key and key status is read automatically after reset.");
 
-      // Observe SPI for a number of cycles. Reset the DUT during monitoring.
+      // Observe SPI for a number of cycles. Reset the DUT during observation.
       show_spi = 1;
       #(10 * CLK_PERIOD);
       reset_dut();
@@ -197,21 +197,51 @@ module tb_keywrap_mkm();
   endtask // auto_load
 
 
+  //----------------------------------------------------------------
+  // test_write_status
+  //----------------------------------------------------------------
+  task test_write_status;
+    begin
+      tc_ctr = tc_ctr + 1;
+
+      $display("TEST WRITE STATUS START");
+      $display("Check that we can write the key status word.");
+
+      // Observe SPI for a number of cycles. Reset the DUT during observation.
+      show_spi = 1;
+      #(10 * CLK_PERIOD);
+
+      tb_key_status = 1'h0;
+      tb_write      = 1'h1;
+      tb_wr_status  = 32'hdeadbeef;
+
+      #(CLK_PERIOD);
+      tb_write      = 1'h0;
+
+      #(100 * CLK_PERIOD);
+      show_spi = 0;
+
+      $display("TEST WRITE STATUS END");
+      $display("");
+    end
+  endtask // test_write_status
+
 
   //----------------------------------------------------------------
   // main
   //----------------------------------------------------------------
   initial
     begin : main
-      $display("   -= Testbench for Keywrap mkmif wrapper started =-");
-      $display("    ================================================");
+      $display("   -= Testbench for Keywrap mkmif integration started =-");
+      $display("    ====================================================");
       $display("");
 
       init_sim();
       test_auto_load();
+      test_write_status();
 
       $display("");
-      $display("*** Keywrap mkmif wrapper testbench done. ***");
+      $display("*** Keywrap mkmif integration testbench done. ***");
       $finish;
     end // main