@@ -758,3 +758,26 @@ def test_session_include_all_counters_after_include_single_counter_should_throw(
758758 Order ,
759759 lambda i : i .include_documents ("company" ).include_counter ("likes" ).include_all_counters (),
760760 )
761+
762+ def test_session_include_all_counters (self ):
763+ with self .store .open_session () as session :
764+ session .store (Company (name = "HR" ), "companies/1-A" )
765+ session .store (Order (company = "companies/1-A" ), "orders/1-A" )
766+ session .counters_for ("orders/1-A" ).increment ("likes" , 100 )
767+ session .counters_for ("orders/1-A" ).increment ("dislikes" , 200 )
768+ session .counters_for ("orders/1-A" ).increment ("downloads" , 300 )
769+ session .save_changes ()
770+
771+ with self .store .open_session () as session :
772+ order = session .load ("orders/1-A" , Order , lambda i : i .include_documents ("company" ).include_all_counters ())
773+
774+ company = session .load (order .company , Company )
775+ self .assertEqual ("HR" , company .name )
776+ dic = session .counters_for_entity (order ).get_all ()
777+
778+ self .assertEqual (3 , len (dic ))
779+ self .assertIn (("likes" , 100 ), dic .items ())
780+ self .assertIn (("dislikes" , 200 ), dic .items ())
781+ self .assertIn (("downloads" , 300 ), dic .items ())
782+
783+ self .assertEqual (1 , session .advanced .number_of_requests )
0 commit comments