From 737dc0159f3653a0768c68ba079ae63808247ce5 Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <stick@gk2.sk>
Date: Wed, 27 Sep 2017 13:32:55 +0200
Subject: [PATCH] tests: implement otp read/write in production test

---
 tests/production_tests/main.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/tests/production_tests/main.py b/tests/production_tests/main.py
index 65134992f5..db51957313 100644
--- a/tests/production_tests/main.py
+++ b/tests/production_tests/main.py
@@ -25,6 +25,7 @@ usb.add(usb_vcp)
 usb.open()
 
 d = ui.Display()
+otp = io.FlashOTP()
 sd = io.SDCard()
 sbu = io.SBU()
 
@@ -112,14 +113,19 @@ def test_sbu(v):
     print('OK')
 
 
-def test_otp_read(v):
-    # FIXME: really read
-    otp = '00000'
-    print('OK', otp)
+def test_otp_read():
+    data = bytearray(32)
+    otp.read(0, 0, data)
+    data = bytes(data).rstrip(b'\x00\xff').decode()
+    print('OK', data)
 
 
 def test_otp_write(v):
-    # FIXME: really write
+    if len(v) < 32:
+        v = v + '\x00' * (32 - len(v))
+    data = v[:32].encode()
+    otp.write(0, 0, data)
+    otp.lock(0)
     print('OK')
 
 
@@ -147,7 +153,7 @@ while True:
             test_sbu(line[4:])
 
         elif line.startswith('OTP READ'):
-            test_otp_read(line[8:])
+            test_otp_read()
 
         elif line.startswith('OTP WRITE '):
             test_otp_write(line[10:])