test: storing-custom-types

listings/getting-started/bytearray/src/bytearray.cairo

@@ -1,10 +1,10 @@
-// ANCHOR: contract
 #[starknet::interface]
 pub trait IMessage<TContractState> {
     fn append(ref self: TContractState, str: ByteArray);
     fn prepend(ref self: TContractState, str: ByteArray);
 }
 
+// ANCHOR: contract
 #[starknet::contract]
 pub mod MessageContract {
     #[storage]

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

@@ -3,6 +3,7 @@ pub trait IStoringCustomType<TContractState> {
     fn set_person(ref self: TContractState, person: Person);
 }
 
+// ANCHOR: contract
 // Deriving the starknet::Store trait
 // allows us to store the `Person` struct in the contract's storage.
 #[derive(Drop, Serde, Copy, starknet::Store)]
@@ -27,3 +28,25 @@ pub mod StoringCustomType {
         }
     }
 }
+// ANCHOR_END: contract
+
+#[cfg(test)]
+mod tests {
+    use super::{
+        IStoringCustomType, StoringCustomType, Person,
+        StoringCustomType::personContractMemberStateTrait
+    };
+
+    #[test]
+    fn can_call_set_person() {
+        let mut state = StoringCustomType::contract_state_for_testing();
+
+        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');
+    }
+}

listings/getting-started/storing_custom_types/src/lib.cairo

@@ -1,4 +1 @@
 mod contract;
-
-#[cfg(test)]
-mod tests;

src/ch00/basics/storing-custom-types.md

@@ -3,5 +3,5 @@
 While native types can be stored in a contract's storage without any additional work, custom types require a bit more work. This is because at compile time, the compiler does not know how to store custom types in storage. To solve this, we need to implement the `Store` trait for our custom type. Hopefully, we can just derive this trait for our custom type - unless it contains arrays or dictionaries.
 
 ```rust
-{{#include ../../../listings/getting-started/storing_custom_types/src/contract.cairo}}
+{{#rustdoc_include ../../../listings/getting-started/storing_custom_types/src/contract.cairo:contract}}
 ```