@@ -1064,10 +1064,12 @@ def setUp(self):
10641064 "content_id" : uuid .uuid4 ().hex ,
10651065 }
10661066
1067- def serialize_to_store (self , Model , data ):
1067+ def serialize_to_store (self , Model , data , post_serialization = None ):
10681068 instance = Model (** data )
10691069 instance .calculate_uuid ()
10701070 serialized = instance .serialize ()
1071+ if post_serialization :
1072+ serialized .update (post_serialization )
10711073 Store .objects .create (
10721074 id = serialized ["id" ],
10731075 serialized = json .dumps (serialized ),
@@ -1080,11 +1082,12 @@ def serialize_to_store(self, Model, data):
10801082 model_name = instance .morango_model_name ,
10811083 )
10821084
1083- def serialize_all_to_store (self ):
1085+ def serialize_all_to_store (self , post_serialization = None ):
1086+ post_serialization = post_serialization or {}
10841087 self .serialize_to_store (Facility , self .serialized_facility )
1085- self .serialize_to_store (MyUser , self .serialized_user )
1086- self .serialize_to_store (SummaryLog , self .serialized_log1 )
1087- self .serialize_to_store (SummaryLog , self .serialized_log2 )
1088+ self .serialize_to_store (MyUser , self .serialized_user , post_serialization = post_serialization . get ( "user" , {}) )
1089+ self .serialize_to_store (SummaryLog , self .serialized_log1 , post_serialization = post_serialization . get ( "log1" , {}) )
1090+ self .serialize_to_store (SummaryLog , self .serialized_log2 , post_serialization = post_serialization . get ( "log2" , {}) )
10881091 self .serialize_to_store (ConditionalLog , self .serialized_conditional )
10891092
10901093 def assert_deserialization (
@@ -1186,15 +1189,13 @@ def test_deserialization_with_excessively_long_username(self):
11861189
11871190 def test_deserialization_with_invalid_content_id (self ):
11881191
1189- self .serialized_log1 ["content_id" ] = "invalid"
1190-
1191- self .serialize_all_to_store ()
1192+ self .serialize_all_to_store ({"log1" : {"content_id" : "invalid" }})
11921193
11931194 _deserialize_from_store (self .profile )
11941195
11951196 self .assert_deserialization (log1_deserialized = False )
11961197
1197- def test_deserialization_with_invalid_log_user_id (self ):
1198+ def test_deserialization_with_log_non_existent_user_id (self ):
11981199
11991200 self .serialized_log1 ["user_id" ] = uuid .uuid4 ().hex
12001201
0 commit comments