@@ -550,6 +550,26 @@ def test_unwrap_filters_dict(self):
550550 self .carepoint ._unwrap_filters (model_obj , filters )
551551 mk .assert_has_calls (expect , any_order = True )
552552
553+ @mock .patch ('carepoint.db.carepoint.or_' )
554+ def test_unwrap_filters_list (self , or_ ):
555+ """ It should create a proper or query for lists. """
556+ model_obj = self .__get_model_obj ()
557+ col_expect = 'test_col'
558+ query_expect = ['test_1' , 'test_2' ]
559+ filters = {
560+ col_expect : query_expect
561+ }
562+ expect = [
563+ mock .call (model_obj , col_expect , '==' , query_expect [0 ]),
564+ mock .call (model_obj , col_expect , '==' , query_expect [1 ]),
565+ ]
566+ criterion_return = 'Criterion'
567+ with mock .patch .object (self .carepoint , '_create_criterion' ) as mk :
568+ mk .return_value = criterion_return
569+ self .carepoint ._unwrap_filters (model_obj , filters )
570+ mk .assert_has_calls (expect , any_order = True )
571+ or_ .assert_called_once_with (criterion_return , criterion_return )
572+
553573 def test_unwrap_filters_str (self ):
554574 model_obj = self .__get_model_obj ()
555575 col_expect = 'test_col'
0 commit comments