Skip to content

Commit 1cb8e40

Browse files
committed
add owner-list function
1 parent 084c368 commit 1cb8e40

File tree

1 file changed

+17
-4
lines changed

1 file changed

+17
-4
lines changed

tests/intergration/test_starfish_contracts.py

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from convex_api.exceptions import ConvexAPIError
1414

1515
CONTRACT_NAME='starfish-did-registry'
16-
CONTRACT_VERSION = '0.0.3'
16+
CONTRACT_VERSION = '0.0.4'
1717

1818
did_registry_contract = f"""
1919
(def starfish-did-registry
@@ -72,8 +72,12 @@
7272
[did (address to-account)]
7373
)
7474
)
75-
(defn dump [] (when (= creator *caller*) registry) )
76-
(export dump resolve resolve? register unregister owner owner? transfer version)
75+
(defn dump [] registry )
76+
(defn owner-list [the-owner]
77+
(assert-address the-owner)
78+
(mapcat (fn [k v] (when (= (address the-owner) (get v :owner)) [k])) (keys registry) (values registry))
79+
)
80+
(export dump resolve resolve? register unregister owner owner? owner-list transfer version)
7781
)
7882
)
7983
)
@@ -365,6 +369,15 @@ def test_contract_ddo_dump(convex, test_account, other_account):
365369
for did in did_list:
366370
assert(did in list(result['value'].keys()))
367371

372+
368373
command = f'(call {contract_address} (dump))'
369374
result = convex.query(command, other_account)
370-
assert(not result['value'])
375+
assert(result['value'])
376+
for did in did_list:
377+
assert(did in list(result['value'].keys()))
378+
379+
command = f'(call {contract_address} (owner-list "{test_account.address_api}"))'
380+
result = convex.query(command, test_account)
381+
assert(result['value'])
382+
for did in did_list:
383+
assert(did in result['value'])

0 commit comments

Comments
 (0)