@@ -1063,9 +1063,11 @@ def setUp(self):
10631063 "content_id" : uuid .uuid4 ().hex ,
10641064 }
10651065
1066- def serialize_to_store (self , Model , data ):
1066+ def serialize_to_store (self , Model , data , post_serialization = None ):
10671067 instance = Model (** data )
10681068 serialized = instance .serialize ()
1069+ if post_serialization :
1070+ serialized .update (post_serialization )
10691071 Store .objects .create (
10701072 id = serialized ["id" ],
10711073 serialized = json .dumps (serialized ),
@@ -1078,10 +1080,10 @@ def serialize_to_store(self, Model, data):
10781080 model_name = instance .morango_model_name ,
10791081 )
10801082
1081- def serialize_all_to_store (self ):
1082- self .serialize_to_store (MyUser , self .serialized_user )
1083- self .serialize_to_store (SummaryLog , self .serialized_log1 )
1084- self .serialize_to_store (SummaryLog , self .serialized_log2 )
1083+ def serialize_all_to_store (self , post_serialization = None ):
1084+ self .serialize_to_store (MyUser , self .serialized_user , post_serialization = getattr ( post_serialization , "user" , {}) )
1085+ self .serialize_to_store (SummaryLog , self .serialized_log1 , post_serialization = getattr ( post_serialization , "log1" , {}) )
1086+ self .serialize_to_store (SummaryLog , self .serialized_log2 , post_serialization = getattr ( post_serialization , "log2" , {}) )
10851087
10861088 def assert_deserialization (self , user_deserialized = True , log1_deserialized = True , log2_deserialized = True ):
10871089 assert MyUser .objects .filter (id = self .serialized_user ["id" ]).exists () == user_deserialized
@@ -1121,15 +1123,13 @@ def test_deserialization_with_excessively_long_username(self):
11211123
11221124 def test_deserialization_with_invalid_content_id (self ):
11231125
1124- self .serialized_log1 ["content_id" ] = "invalid"
1125-
1126- self .serialize_all_to_store ()
1126+ self .serialize_all_to_store ({"log1" : {"content_id" : "invalid" }})
11271127
11281128 _deserialize_from_store (self .profile )
11291129
11301130 self .assert_deserialization (log1_deserialized = False )
11311131
1132- def test_deserialization_with_invalid_log_user_id (self ):
1132+ def test_deserialization_with_log_non_existent_user_id (self ):
11331133
11341134 self .serialized_log1 ["user_id" ] = uuid .uuid4 ().hex
11351135
0 commit comments