From 7b1334ade193f4ffabfb235f99329f2f70e7bb7c Mon Sep 17 00:00:00 2001
From: Andinus <andinus@nand.sh>
Date: Sat, 25 Sep 2021 16:49:01 +0530
Subject: Move records to big_map, add update and delete functions, add tests

---
 contract/memoir.py | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/contract/memoir.py b/contract/memoir.py
index 792274b..c33c155 100644
--- a/contract/memoir.py
+++ b/contract/memoir.py
@@ -3,14 +3,23 @@ import smartpy as sp
 class Memoir(sp.Contract):
     def __init__(self):
         self.init(
-            dates = sp.list([]),
-            records = sp.list([])
+            records = sp.big_map(
+                tkey = sp.TInt,
+                tvalue = sp.TString
+            ),
         )
 
     @sp.entry_point
     def add_record(self, params):
-        self.data.dates.push(sp.now)
-        self.data.records.push(params.record)
+        self.data.records[params.id] = params.record
+
+    @sp.entry_point
+    def update_record(self, params):
+        self.data.records = sp.update_map(self.data.records, params.id, sp.some(params.record))
+
+    @sp.entry_point
+    def delete_record(self, params):
+        del self.data.records[params.id]
 
 @sp.add_test(name="MyMedMemoir")
 def test():
@@ -19,5 +28,8 @@ def test():
 
     c1 = Memoir()
     scenario += c1
-    scenario += c1.add_record(record = '{record #1}')
-    scenario += c1.add_record(record = '{record #2}')
+    scenario += c1.add_record(id = 0, record = '{record #0}')
+    scenario += c1.add_record(id = 1, record = '{record #1}')
+    scenario += c1.add_record(id = 2, record = '{record #2}')
+    scenario += c1.update_record(id = 1, record = '{record #1 - updated}')
+    scenario += c1.delete_record(id = 2)
-- 
cgit 1.4.1-2-gfad0