Test storing custom types

listings/ch00-getting-started/storing_custom_types/src/contract.cairo

@@ -1,29 +1,29 @@
+#[starknet::interface]
+trait IStoringCustomType<TContractState> {
+    fn set_person(ref self: TContractState, person: Person);
+}
+
+// Deriving the starknet::Store trait
+// allows us to store the `Person` struct in the contract's storage.
+#[derive(Drop, Serde, Copy, starknet::Store)]
+struct Person {
+    age: u8,
+    name: felt252
+}
+
 #[starknet::contract]
 mod StoringCustomType {
+    use super::Person;
+
     #[storage]
     struct Storage {
         person: Person
     }
 
-    // Deriving the starknet::Store trait
-    // allows us to store the `Person` struct in the contract's storage.
-    #[derive(Drop, starknet::Store)]
-    struct Person {
-        age: u8,
-        name: felt252
-    }
-
-    #[abi(per_item)]
-    #[generate_trait]
-    impl StoringCustomType of IStoringCustomType {
-        #[external(v0)]
-        fn set_person(ref self: ContractState) {
-            self.person.write(Person { age: 10, name: 'Joe' });
-        }
-
-        #[external(v0)]
-        fn get_person(self: @ContractState) {
-            self.person.read();
+    #[abi(embed_v0)]
+    impl StoringCustomType of super::IStoringCustomType<ContractState> {
+        fn set_person(ref self: ContractState, person: Person) {
+            self.person.write(person);
         }
     }
 }

listings/ch00-getting-started/storing_custom_types/src/tests.cairo

@@ -1,2 +1,32 @@
-mod tests { // TODO
+// The purpose of these tests is to demonstrate the capability to store custom types in the contract's state.
+
+mod tests {
+    use storing_custom_types::{
+        contract::{
+            IStoringCustomType, StoringCustomType,
+            StoringCustomType::{Person, person::InternalContractMemberStateTrait}
+        }
+    };
+
+    use starknet::{ContractAddress, contract_address_const, testing::{set_contract_address}};
+
+    fn setup() -> StoringCustomType::ContractState {
+        let mut state = StoringCustomType::contract_state_for_testing();
+        let contract_address = contract_address_const::<0x1>();
+        set_contract_address(contract_address);
+        state
+    }
+
+    #[test]
+    #[available_gas(2000000000)]
+    fn can_call_set_person() {
+        let mut state = setup();
+        let person = Person { age: 10, name: 'Joe' };
+
+        state.set_person(person);
+        let read_person = state.person.read();
+
+        assert(person.age == read_person.age, 'wrong age');
+        assert(person.name == read_person.name, 'wrong name');
+    }
 }