Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

How to use apply_filters method in avocado

Best Python code snippet using avocado_python

test_filters.py

Source:test_filters.py

...69class TestFiltersNotApplied:70    def test_no_filters_provided(self, session):71        stmt = select(Bar)72        filters = []73        filtered_query = apply_filters(stmt, filters)74        assert stmt == filtered_query75    @pytest.mark.parametrize('filter_', ['some text', 1, ''])76    def test_wrong_filters_format(self, session, filter_):77        stmt = select(Bar)78        filters = [filter_]79        with pytest.raises(BadFilterFormat) as err:80            apply_filters(stmt, filters)81        expected_error = 'Filter spec `{}` should be a dictionary.'.format(82            filter_83        )84        assert expected_error == err.value.args[0]85    def test_invalid_operator(self, session):86        stmt = select(Bar)87        filters = [{'field': 'name', 'op': 'op_not_valid', 'value': 'name_1'}]88        with pytest.raises(BadFilterFormat) as err:89            apply_filters(stmt, filters)90        assert 'Operator `op_not_valid` not valid.' == err.value.args[0]91    @pytest.mark.usefixtures('multiple_bars_inserted')92    def test_no_operator_provided(self, session):93        stmt = select(Bar)94        filters = [{'field': 'name', 'value': 'name_1'}]95        filtered_query = apply_filters(stmt, filters)96        result = session.execute(filtered_query).scalars().all()97        assert len(result) == 298        assert result[0].id == 199        assert result[1].id == 3100    def test_no_field_provided(self, session):101        stmt = select(Bar)102        filters = [{'op': '==', 'value': 'name_1'}]103        with pytest.raises(BadFilterFormat) as err:104            apply_filters(stmt, filters)105        expected_error = '`field` is a mandatory filter attribute.'106        assert expected_error == err.value.args[0]107    # TODO: replace this test once we add the option to compare against108    # another field109    def test_no_value_provided(self, session):110        stmt = select(Bar)111        filters = [{'field': 'name', 'op': '==', }]112        with pytest.raises(BadFilterFormat) as err:113            apply_filters(stmt, filters)114        assert '`value` must be provided.' == err.value.args[0]115    def test_invalid_field(self, session):116        stmt = select(Bar)117        filters = [{'field': 'invalid_field', 'op': '==', 'value': 'name_1'}]118        with pytest.raises(FieldNotFound) as err:119            apply_filters(stmt, filters)120        expected_error = (121            "Model <class 'test.models.Bar'> has no column `invalid_field`."122        )123        assert expected_error == err.value.args[0]124    @pytest.mark.parametrize('attr_name', [125        'metadata',  # model attribute126        'foos',  # model relationship127    ])128    def test_invalid_field_but_valid_model_attribute(self, session, attr_name):129        stmt = select(Bar)130        filters = [{'field': attr_name, 'op': '==', 'value': 'name_1'}]131        with pytest.raises(FieldNotFound) as err:132            apply_filters(stmt, filters)133        expected_error = (134            "Model <class 'test.models.Bar'> has no column `{}`.".format(135                attr_name136            )137        )138        assert expected_error == err.value.args[0]139class TestMultipleModels:140    # TODO: multi-model should be tested for each filter type141    @pytest.mark.usefixtures('multiple_bars_inserted')142    @pytest.mark.usefixtures('multiple_quxs_inserted')143    def test_multiple_models(self, session):144        stmt = select(Bar, Qux)145        filters = [146            {'model': 'Bar', 'field': 'name', 'op': '==', 'value': 'name_1'},147            {'model': 'Qux', 'field': 'name', 'op': '==', 'value': 'name_1'},148        ]149        filtered_query = apply_filters(stmt, filters)150        result = session.execute(filtered_query).unique().all()151        assert len(result) == 4152        bars, quxs = zip(*result)153        assert set(map(type, bars)) == {Bar}154        assert {bar.id for bar in bars} == {1, 3}155        assert {bar.name for bar in bars} == {"name_1"}156        assert set(map(type, quxs)) == {Qux}157        assert {qux.id for qux in quxs} == {1, 3}158        assert {qux.name for qux in quxs} == {"name_1"}159class TestAutoJoin:160    @pytest.mark.usefixtures('multiple_foos_inserted')161    def test_auto_join(self, session):162        stmt = select(Foo)163        filters = [164            {'field': 'name', 'op': '==', 'value': 'name_1'},165            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},166        ]167        filtered_query = apply_filters(stmt, filters)168        result = session.execute(filtered_query).scalars().all()169        assert len(result) == 1170        assert result[0].id == 3171        assert result[0].bar_id == 3172        assert result[0].bar.count is None173    @pytest.mark.usefixtures('multiple_foos_inserted')174    def test_do_not_auto_join(self, session):175        stmt = select(Foo)176        filters = [177            {'field': 'name', 'op': '==', 'value': 'name_1'},178            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},179        ]180        with pytest.raises(BadSpec) as exc:181            apply_filters(stmt, filters, do_auto_join=False)182        assert 'The query does not contain model `Bar`' in str(exc)183    @pytest.mark.usefixtures('multiple_foos_inserted')184    def test_noop_if_query_contains_named_models(self, session):185        stmt = select(Foo).join(Bar)186        filters = [187            {'model': 'Foo', 'field': 'name', 'op': '==', 'value': 'name_1'},188            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},189        ]190        filtered_query = apply_filters(stmt, filters)191        result = session.execute(filtered_query).scalars().all()192        assert len(result) == 1193        assert result[0].id == 3194        assert result[0].bar_id == 3195        assert result[0].bar.count is None196    @pytest.mark.usefixtures('multiple_foos_inserted')197    def test_auto_join_to_invalid_model(self, session):198        stmt = select(Foo)199        filters = [200            {'field': 'name', 'op': '==', 'value': 'name_1'},201            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},202            {'model': 'Qux', 'field': 'created_at', 'op': 'is_not_null'}203        ]204        with pytest.raises(BadSpec) as err:205            apply_filters(stmt, filters)206        assert 'The query does not contain model `Qux`.' == err.value.args[0]207    @pytest.mark.usefixtures('multiple_foos_inserted')208    def test_ambiguous_query(self, session):209        stmt = select(Foo).join(Bar)210        filters = [211            {'field': 'name', 'op': '==', 'value': 'name_1'},  # ambiguous212            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},213        ]214        with pytest.raises(BadSpec) as err:215            apply_filters(stmt, filters)216        assert 'Ambiguous spec. Please specify a model.' == err.value.args[0]217    @pytest.mark.usefixtures('multiple_foos_inserted')218    def test_eager_load(self, session):219        # behaves as if the joinedload wasn't present220        stmt = select(Foo).options(joinedload(Foo.bar))221        filters = [222            {'field': 'name', 'op': '==', 'value': 'name_1'},223            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},224        ]225        filtered_query = apply_filters(stmt, filters)226        result = session.execute(filtered_query).scalars().all()227        assert len(result) == 1228        assert result[0].id == 3229        assert result[0].bar_id == 3230        assert result[0].bar.count is None231class TestApplyIsNullFilter:232    @pytest.mark.usefixtures('multiple_bars_inserted')233    def test_filter_field_with_null_values(self, session):234        stmt = select(Bar)235        filters = [{'field': 'count', 'op': 'is_null'}]236        filtered_query = apply_filters(stmt, filters)237        result = session.execute(filtered_query).scalars().all()238        assert len(result) == 1239        assert result[0].id == 3240    @pytest.mark.usefixtures('multiple_bars_inserted')241    def test_filter_field_with_no_null_values(self, session):242        stmt = select(Bar)243        filters = [{'field': 'name', 'op': 'is_null'}]244        filtered_query = apply_filters(stmt, filters)245        result = session.execute(filtered_query).scalars().all()246        assert len(result) == 0247class TestApplyIsNotNullFilter:248    @pytest.mark.usefixtures('multiple_bars_inserted')249    def test_filter_field_with_null_values(self, session):250        stmt = select(Bar)251        filters = [{'field': 'count', 'op': 'is_not_null'}]252        filtered_query = apply_filters(stmt, filters)253        result = session.execute(filtered_query).scalars().all()254        assert len(result) == 3255        assert result[0].id == 1256        assert result[1].id == 2257        assert result[2].id == 4258    @pytest.mark.usefixtures('multiple_bars_inserted')259    def test_filter_field_with_no_null_values(self, session):260        stmt = select(Bar)261        filters = [{'field': 'name', 'op': 'is_not_null'}]262        filtered_query = apply_filters(stmt, filters)263        result = session.execute(filtered_query).scalars().all()264        assert len(result) == 4265        assert result[0].id == 1266        assert result[1].id == 2267        assert result[2].id == 3268        assert result[3].id == 4269class TestApplyFiltersMultipleTimes:270    @pytest.mark.usefixtures('multiple_bars_inserted')271    def test_concatenate_queries(self, session):272        stmt = select(Bar)273        filters = [{'field': 'name', 'op': '==', 'value': 'name_1'}]274        filtered_query = apply_filters(stmt, filters)275        result = session.execute(filtered_query).scalars().all()276        assert len(result) == 2277        assert result[0].id == 1278        assert result[0].name == 'name_1'279        assert result[1].id == 3280        assert result[1].name == 'name_1'281        filters = [{'field': 'id', 'op': '==', 'value': 3}]282        filtered_query = apply_filters(filtered_query, filters)283        result = session.execute(filtered_query).scalars().all()284        assert len(result) == 1285        assert result[0].id == 3286        assert result[0].name == 'name_1'287class TestApplyFilterWithoutList:288    @pytest.mark.usefixtures('multiple_bars_inserted')289    def test_a_single_dict_can_be_supplied_as_filters(self, session):290        stmt = select(Bar)291        filters = {'field': 'name', 'op': '==', 'value': 'name_1'}292        filtered_query = apply_filters(stmt, filters)293        result = session.execute(filtered_query).scalars().all()294        assert len(result) == 2295        assert result[0].id == 1296        assert result[0].name == 'name_1'297        assert result[1].id == 3298        assert result[1].name == 'name_1'299class TestApplyFilterOnFieldBasedQuery:300    @pytest.mark.usefixtures('multiple_bars_inserted')301    def test_apply_filter_on_single_field_query(self, session):302        stmt = select(Bar.id)303        filters = [{'field': 'name', 'op': '==', 'value': 'name_1'}]304        filtered_query = apply_filters(stmt, filters)305        result = session.execute(filtered_query).all()306        assert len(result) == 2307        assert result[0] == (1,)308        assert result[1] == (3,)309    @pytest.mark.usefixtures('multiple_bars_inserted')310    def test_apply_filter_on_aggregate_query(self, session):311        stmt = select(func.count(Bar.id))312        filters = [{'field': 'name', 'op': '==', 'value': 'name_1'}]313        filtered_query = apply_filters(stmt, filters)314        result = session.execute(filtered_query).all()315        assert len(result) == 1316        assert result[0] == (2,)317class TestApplyEqualToFilter:318    @pytest.mark.parametrize('operator', ['==', 'eq'])319    @pytest.mark.usefixtures('multiple_bars_inserted')320    def test_one_filter_applied_to_a_single_model(self, session, operator):321        stmt = select(Bar)322        filters = [{'field': 'name', 'op': operator, 'value': 'name_1'}]323        filtered_query = apply_filters(stmt, filters)324        result = session.execute(filtered_query).scalars().all()325        assert len(result) == 2326        assert result[0].id == 1327        assert result[0].name == 'name_1'328        assert result[1].id == 3329        assert result[1].name == 'name_1'330    @pytest.mark.parametrize(331        'filters', [332            [  # filters using `==` in a list333                {'field': 'name', 'op': '==', 'value': 'name_1'},334                {'field': 'id', 'op': '==', 'value': 3}335            ],336            (  # filters using `eq` in a tuple337                {'field': 'name', 'op': 'eq', 'value': 'name_1'},338                {'field': 'id', 'op': 'eq', 'value': 3}339            )340        ]341    )342    @pytest.mark.usefixtures('multiple_bars_inserted')343    def test_multiple_filters_applied_to_a_single_model(344        self, session, filters345    ):346        stmt = select(Bar)347        filtered_query = apply_filters(stmt, filters)348        result = session.execute(filtered_query).scalars().all()349        assert len(result) == 1350        assert result[0].id == 3351        assert result[0].name == 'name_1'352class TestApplyNotEqualToFilter:353    @pytest.mark.parametrize('operator', ['!=', 'ne'])354    @pytest.mark.usefixtures('multiple_bars_inserted')355    def test_one_filter_applied_to_a_single_model(self, session, operator):356        stmt = select(Bar)357        filters = [{'field': 'name', 'op': operator, 'value': 'name_1'}]358        filtered_query = apply_filters(stmt, filters)359        result = session.execute(filtered_query).scalars().all()360        assert len(result) == 2361        assert result[0].id == 2362        assert result[0].name == 'name_2'363        assert result[1].id == 4364        assert result[1].name == 'name_4'365    @pytest.mark.parametrize('operator', ['!=', 'ne'])366    @pytest.mark.usefixtures('multiple_bars_inserted')367    def test_multiple_filters_applied_to_a_single_model(368        self, session, operator369    ):370        stmt = select(Bar)371        filters = [372            {'field': 'name', 'op': operator, 'value': 'name_2'},373            {'field': 'id', 'op': operator, 'value': 3}374        ]375        filtered_query = apply_filters(stmt, filters)376        result = session.execute(filtered_query).scalars().all()377        assert len(result) == 2378        assert result[0].id == 1379        assert result[0].name == 'name_1'380        assert result[1].id == 4381        assert result[1].name == 'name_4'382class TestApplyGreaterThanFilter:383    @pytest.mark.parametrize('operator', ['>', 'gt'])384    @pytest.mark.usefixtures('multiple_bars_inserted')385    def test_one_filter_applied_to_a_single_model(self, session, operator):386        stmt = select(Bar)387        filters = [{'field': 'count', 'op': operator, 'value': '5'}]388        filtered_query = apply_filters(stmt, filters)389        result = session.execute(filtered_query).scalars().all()390        assert len(result) == 2391        assert result[0].id == 2392        assert result[1].id == 4393    @pytest.mark.parametrize('operator', ['>', 'gt'])394    @pytest.mark.usefixtures('multiple_bars_inserted')395    def test_multiple_filters_applied_to_a_single_model(396        self, session, operator397    ):398        stmt = select(Bar)399        filters = [400            {'field': 'count', 'op': operator, 'value': '5'},401            {'field': 'id', 'op': operator, 'value': 2},402        ]403        filtered_query = apply_filters(stmt, filters)404        result = session.execute(filtered_query).scalars().all()405        assert len(result) == 1406        assert result[0].id == 4407class TestApplyLessThanFilter:408    @pytest.mark.parametrize('operator', ['<', 'lt'])409    @pytest.mark.usefixtures('multiple_bars_inserted')410    def test_one_filter_applied_to_a_single_model(self, session, operator):411        stmt = select(Bar)412        filters = [{'field': 'count', 'op': operator, 'value': '7'}]413        filtered_query = apply_filters(stmt, filters)414        result = session.execute(filtered_query).scalars().all()415        assert len(result) == 1416        assert result[0].id == 1417    @pytest.mark.parametrize('operator', ['<', 'lt'])418    @pytest.mark.usefixtures('multiple_bars_inserted')419    def test_multiple_filters_applied_to_a_single_model(420        self, session, operator421    ):422        stmt = select(Bar)423        filters = [424            {'field': 'count', 'op': operator, 'value': '7'},425            {'field': 'id', 'op': operator, 'value': 1},426        ]427        filtered_query = apply_filters(stmt, filters)428        result = session.execute(filtered_query).scalars().all()429        assert len(result) == 0430class TestApplyGreaterOrEqualThanFilter:431    @pytest.mark.parametrize('operator', ['>=', 'ge'])432    @pytest.mark.usefixtures('multiple_bars_inserted')433    def test_one_filter_applied_to_a_single_model(self, session, operator):434        stmt = select(Bar)435        filters = [{'field': 'count', 'op': operator, 'value': '5'}]436        filtered_query = apply_filters(stmt, filters)437        result = session.execute(filtered_query).scalars().all()438        assert len(result) == 3439        assert result[0].id == 1440        assert result[1].id == 2441        assert result[2].id == 4442    @pytest.mark.parametrize('operator', ['>=', 'ge'])443    @pytest.mark.usefixtures('multiple_bars_inserted')444    def test_multiple_filters_applied_to_a_single_model(445        self, session, operator446    ):447        stmt = select(Bar)448        filters = [449            {'field': 'count', 'op': operator, 'value': '5'},450            {'field': 'id', 'op': operator, 'value': 4},451        ]452        filtered_query = apply_filters(stmt, filters)453        result = session.execute(filtered_query).scalars().all()454        assert len(result) == 1455        assert result[0].id == 4456class TestApplyLessOrEqualThanFilter:457    @pytest.mark.parametrize('operator', ['<=', 'le'])458    @pytest.mark.usefixtures('multiple_bars_inserted')459    def test_one_filter_applied_to_a_single_model(self, session, operator):460        stmt = select(Bar)461        filters = [{'field': 'count', 'op': operator, 'value': '15'}]462        filtered_query = apply_filters(stmt, filters)463        result = session.execute(filtered_query).scalars().all()464        assert len(result) == 3465        assert result[0].id == 1466        assert result[1].id == 2467        assert result[2].id == 4468    @pytest.mark.parametrize('operator', ['<=', 'le'])469    @pytest.mark.usefixtures('multiple_bars_inserted')470    def test_multiple_filters_applied_to_a_single_model(471        self, session, operator472    ):473        stmt = select(Bar)474        filters = [475            {'field': 'count', 'op': operator, 'value': '15'},476            {'field': 'id', 'op': operator, 'value': 1},477        ]478        filtered_query = apply_filters(stmt, filters)479        result = session.execute(filtered_query).scalars().all()480        assert len(result) == 1481        assert result[0].id == 1482class TestApplyLikeFilter:483    @pytest.mark.usefixtures('multiple_bars_inserted')484    def test_one_filter_applied_to_a_single_model(self, session):485        stmt = select(Bar)486        filters = [{'field': 'name', 'op': 'like', 'value': '%me_1'}]487        filtered_query = apply_filters(stmt, filters)488        result = session.execute(filtered_query).scalars().all()489        assert len(result) == 2490        assert result[0].id == 1491        assert result[1].id == 3492class TestApplyILikeFilter:493    @pytest.mark.usefixtures('multiple_bars_inserted')494    def test_one_filter_applied_to_a_single_model(self, session):495        stmt = session.query(Bar)496        filters = [{'field': 'name', 'op': 'ilike', 'value': '%ME_1'}]497        filtered_query = apply_filters(stmt, filters)498        result = session.execute(filtered_query).scalars().all()499        assert len(result) == 2500        assert result[0].id == 1501        assert result[1].id == 3502class TestApplyNotILikeFilter:503    @pytest.mark.usefixtures('multiple_bars_inserted')504    def test_one_filter_applied_to_a_single_model(self, session):505        stmt = select(Bar)506        filters = [{'field': 'name', 'op': 'not_ilike', 'value': '%ME_1'}]507        filtered_query = apply_filters(stmt, filters)508        result = session.execute(filtered_query).scalars().all()509        assert len(result) == 2510        assert result[0].id == 2511        assert result[1].id == 4512class TestApplyInFilter:513    @pytest.mark.usefixtures('multiple_bars_inserted')514    def test_field_not_in_value_list(self, session):515        stmt = select(Bar)516        filters = [{'field': 'count', 'op': 'in', 'value': [1, 2, 3]}]517        filtered_query = apply_filters(stmt, filters)518        result = session.execute(filtered_query).all()519        assert len(result) == 0520    @pytest.mark.usefixtures('multiple_bars_inserted')521    def test_field_in_value_list(self, session):522        stmt = select(Bar)523        filters = [{'field': 'count', 'op': 'in', 'value': [15, 2, 3]}]524        filtered_query = apply_filters(stmt, filters)525        result = session.execute(filtered_query).scalars().all()526        assert len(result) == 1527        assert result[0].id == 4528class TestApplyNotInFilter:529    @pytest.mark.usefixtures('multiple_bars_inserted')530    def test_field_not_in_value_list(self, session):531        stmt = select(Bar)532        filters = [{'field': 'count', 'op': 'not_in', 'value': [1, 2, 3]}]533        filtered_query = apply_filters(stmt, filters)534        result = session.execute(filtered_query).scalars().all()535        assert len(result) == 3536        assert result[0].id == 1537        assert result[1].id == 2538        assert result[2].id == 4539    @pytest.mark.usefixtures('multiple_bars_inserted')540    def test_field_in_value_list(self, session):541        stmt = select(Bar)542        filters = [{'field': 'count', 'op': 'not_in', 'value': [15, 2, 3]}]543        filtered_query = apply_filters(stmt, filters)544        result = session.execute(filtered_query).scalars().all()545        assert len(result) == 2546        assert result[0].id == 1547        assert result[1].id == 2548class TestDateFields:549    @pytest.mark.parametrize(550        'value',551        [552            datetime.date(2016, 7, 14),553            datetime.date(2016, 7, 14).isoformat()554        ]555    )556    @pytest.mark.usefixtures('multiple_quxs_inserted')557    def test_filter_date_equality(self, session, value):558        stmt = select(Qux)559        filters = [{560            'field': 'created_at',561            'op': '==',562            'value': value563        }]564        filtered_query = apply_filters(stmt, filters)565        result = session.execute(filtered_query).scalars().all()566        assert len(result) == 1567        assert result[0].created_at == datetime.date(2016, 7, 14)568    @pytest.mark.parametrize(569        'value',570        [571            datetime.date(2016, 7, 13),572            datetime.date(2016, 7, 13).isoformat()573        ]574    )575    @pytest.mark.usefixtures('multiple_quxs_inserted')576    def test_filter_multiple_dates(self, session, value):577        stmt = select(Qux)578        filters = [{579            'field': 'created_at',580            'op': '>=',581            'value': value582        }]583        filtered_query = apply_filters(stmt, filters)584        result = session.execute(filtered_query).scalars().all()585        assert len(result) == 2586        assert result[0].created_at == datetime.date(2016, 7, 13)587        assert result[1].created_at == datetime.date(2016, 7, 14)588    @pytest.mark.usefixtures('multiple_quxs_inserted')589    def test_null_date(self, session):590        stmt = session.query(Qux)591        filters = [{'field': 'created_at', 'op': 'is_null'}]592        filtered_query = apply_filters(stmt, filters)593        result = session.execute(filtered_query).scalars().all()594        assert len(result) == 1595        assert result[0].created_at is None596class TestTimeFields:597    @pytest.mark.parametrize(598        'value',599        [600            datetime.time(3, 5, 9),601            datetime.time(3, 5, 9).isoformat()  # '03:05:09'602        ]603    )604    @pytest.mark.usefixtures('multiple_quxs_inserted')605    def test_filter_time_equality(self, session, is_sqlite, value):606        if isinstance(value, str) and is_sqlite:607            pytest.skip(STRING_DATE_TIME_NOT_SUPPORTED)608        stmt = select(Qux)609        filters = [{'field': 'expiration_time', 'op': '==', 'value': value}]610        filtered_query = apply_filters(stmt, filters)611        result = session.execute(filtered_query).scalars().all()612        assert len(result) == 1613        assert result[0].expiration_time == datetime.time(3, 5, 9)614    @pytest.mark.parametrize(615        'value',616        [617            datetime.time(2, 5, 9),618            datetime.time(2, 5, 9).isoformat()  # '02:05:09'619        ]620    )621    @pytest.mark.usefixtures('multiple_quxs_inserted')622    def test_filter_multiple_times(self, session, is_sqlite, value):623        if isinstance(value, str) and is_sqlite:624            pytest.skip(STRING_DATE_TIME_NOT_SUPPORTED)625        stmt = select(Qux)626        filters = [{627            'field': 'expiration_time',628            'op': '>=',629            'value': value630        }]631        filtered_query = apply_filters(stmt, filters)632        result = session.execute(filtered_query).scalars().all()633        assert len(result) == 2634        assert result[0].expiration_time == datetime.time(2, 5, 9)635        assert result[1].expiration_time == datetime.time(3, 5, 9)636    @pytest.mark.usefixtures('multiple_quxs_inserted')637    def test_null_time(self, session):638        stmt = select(Qux)639        filters = [{'field': 'expiration_time', 'op': 'is_null'}]640        filtered_query = apply_filters(stmt, filters)641        result = session.execute(filtered_query).scalars().all()642        assert len(result) == 1643        assert result[0].expiration_time is None644class TestDateTimeFields:645    @pytest.mark.parametrize(646        'value',647        [648            datetime.datetime(2016, 7, 14, 3, 5, 9),649            # '2016-07-14T03:05:09'650            datetime.datetime(2016, 7, 14, 3, 5, 9).isoformat()651        ]652    )653    @pytest.mark.usefixtures('multiple_quxs_inserted')654    def test_filter_datetime_equality(self, session, is_sqlite, value):655        if isinstance(value, str) and is_sqlite:656            pytest.skip(STRING_DATE_TIME_NOT_SUPPORTED)657        stmt = select(Qux)658        filters = [{659            'field': 'execution_time',660            'op': '==',661            'value': value662        }]663        filtered_query = apply_filters(stmt, filters)664        result = session.execute(filtered_query).scalars().all()665        assert len(result) == 1666        assert result[0].execution_time == datetime.datetime(667            2016, 7, 14, 3, 5, 9668        )669    @pytest.mark.parametrize(670        'value',671        [672            datetime.datetime(2016, 7, 13, 2, 5, 9),673            # '2016-07-13T02:05:09'674            datetime.datetime(2016, 7, 13, 2, 5, 9).isoformat()675        ]676    )677    @pytest.mark.usefixtures('multiple_quxs_inserted')678    def test_filter_multiple_datetimes(self, session, is_sqlite, value):679        if isinstance(value, str) and is_sqlite:680            pytest.skip(STRING_DATE_TIME_NOT_SUPPORTED)681        stmt = select(Qux)682        filters = [{683            'field': 'execution_time',684            'op': '>=',685            'value': value686        }]687        filtered_query = apply_filters(stmt, filters)688        result = session.execute(filtered_query).scalars().all()689        assert len(result) == 2690        assert result[0].execution_time == datetime.datetime(691            2016, 7, 13, 2, 5, 9692        )693        assert result[1].execution_time == datetime.datetime(694            2016, 7, 14, 3, 5, 9695        )696    @pytest.mark.usefixtures('multiple_quxs_inserted')697    def test_null_datetime(self, session):698        stmt = select(Qux)699        filters = [{'field': 'execution_time', 'op': 'is_null'}]700        filtered_query = apply_filters(stmt, filters)701        result = session.execute(filtered_query).scalars().all()702        assert len(result) == 1703        assert result[0].execution_time is None704class TestApplyBooleanFunctions:705    @pytest.mark.usefixtures('multiple_bars_inserted')706    def test_or(self, session):707        stmt = select(Bar)708        filters = [709            {'or': [710                {'field': 'id', 'op': '==', 'value': 1},711                {'field': 'id', 'op': '==', 'value': 3},712            ]},713        ]714        filtered_query = apply_filters(stmt, filters)715        result = session.execute(filtered_query).scalars().all()716        assert len(result) == 2717        assert result[0].id == 1718        assert result[1].id == 3719    @pytest.mark.usefixtures('multiple_bars_inserted')720    def test_or_with_one_arg(self, session):721        stmt = select(Bar)722        filters = [723            {'or': [724                {'field': 'id', 'op': '==', 'value': 1},725            ]},726        ]727        filtered_query = apply_filters(stmt, filters)728        result = session.execute(filtered_query).scalars().all()729        assert len(result) == 1730        assert result[0].id == 1731    @pytest.mark.usefixtures('multiple_bars_inserted')732    def test_or_with_three_args(self, session):733        stmt = select(Bar)734        filters = [735            {'or': [736                {'field': 'id', 'op': '==', 'value': 1},737                {'field': 'id', 'op': '==', 'value': 3},738                {'field': 'id', 'op': '==', 'value': 4},739            ]},740        ]741        filtered_query = apply_filters(stmt, filters)742        result = session.execute(filtered_query).scalars().all()743        assert len(result) == 3744        assert result[0].id == 1745        assert result[1].id == 3746        assert result[2].id == 4747    @pytest.mark.parametrize(748        ('or_args', 'expected_error'), [749            (750                [],751                '`or` must have one or more arguments'752            ),753            (754                {},755                '`or` value must be an iterable across the function arguments'756            ),757            (758                'hello',759                '`or` value must be an iterable across the function arguments'760            ),761        ]762    )763    @pytest.mark.usefixtures('multiple_bars_inserted')764    def test_or_with_bad_format(self, session, or_args, expected_error):765        stmt = select(Bar)766        filters = [{'or': or_args}]767        with pytest.raises(BadFilterFormat) as exc:768            apply_filters(stmt, filters)769        assert expected_error in str(exc)770    @pytest.mark.usefixtures('multiple_bars_inserted')771    def test_and(self, session):772        stmt = select(Bar)773        filters = [774            {'and': [775                {'field': 'id', 'op': '<=', 'value': 2},776                {'field': 'count', 'op': '>=', 'value': 6},777            ]},778        ]779        filtered_query = apply_filters(stmt, filters)780        result = session.execute(filtered_query).scalars().all()781        assert len(result) == 1782        assert result[0].id == 2783    @pytest.mark.usefixtures('multiple_bars_inserted')784    def test_and_with_one_arg(self, session):785        stmt = select(Bar)786        filters = [787            {'and': [788                {'field': 'id', 'op': '==', 'value': 3},789            ]},790        ]791        filtered_query = apply_filters(stmt, filters)792        result = session.execute(filtered_query).scalars().all()793        assert len(result) == 1794        assert result[0].id == 3795    @pytest.mark.usefixtures('multiple_bars_inserted')796    def test_and_with_three_args(self, session):797        stmt = select(Bar)798        filters = [799            {'and': [800                {'field': 'id', 'op': '<=', 'value': 3},801                {'field': 'name', 'op': '==', 'value': 'name_1'},802                {'field': 'count', 'op': 'is_not_null'},803            ]},804        ]805        filtered_query = apply_filters(stmt, filters)806        result = session.execute(filtered_query).scalars().all()807        assert len(result) == 1808        assert result[0].id == 1809    @pytest.mark.parametrize(810        ('and_args', 'expected_error'), [811            (812                [],813                '`and` must have one or more arguments'814            ),815            (816                {},817                '`and` value must be an iterable across the function arguments'818            ),819            (820                'hello',821                '`and` value must be an iterable across the function arguments'822            ),823        ]824    )825    @pytest.mark.usefixtures('multiple_bars_inserted')826    def test_and_with_bad_format(self, session, and_args, expected_error):827        stmt = select(Bar)828        filters = [{'and': and_args}]829        with pytest.raises(BadFilterFormat) as exc:830            apply_filters(stmt, filters)831        assert expected_error in str(exc)832    @pytest.mark.usefixtures('multiple_bars_inserted')833    def test_not(self, session):834        stmt = select(Bar)835        filters = [836            {'not': [837                {'field': 'id', 'op': '==', 'value': 3},838            ]},839        ]840        filtered_query = apply_filters(stmt, filters)841        result = session.execute(filtered_query).scalars().all()842        assert len(result) == 3843        assert result[0].id == 1844        assert result[1].id == 2845        assert result[2].id == 4846    @pytest.mark.parametrize(847        ('not_args', 'expected_error'), [848            (849                [{'field': 'id', 'op': '==', 'value': 1},850                    {'field': 'id', 'op': '==', 'value': 2}],851                '`not` must have one argument'852            ),853            (854                [],855                '`not` must have one argument'856            ),857            (858                {},859                '`not` value must be an iterable across the function arguments'860            ),861            (862                'hello',863                '`not` value must be an iterable across the function arguments'864            ),865        ]866    )867    @pytest.mark.usefixtures('multiple_bars_inserted')868    def test_not_with_bad_format(self, session, not_args, expected_error):869        stmt = select(Bar)870        filters = [{'not': not_args}]871        with pytest.raises(BadFilterFormat) as exc:872            apply_filters(stmt, filters)873        assert expected_error in str(exc)874    @pytest.mark.usefixtures('multiple_bars_inserted')875    def test_complex(self, session):876        stmt = select(Bar)877        filters = [878            {879                'and': [880                    {881                        'or': [882                            {'field': 'id', 'op': '==', 'value': 2},883                            {'field': 'id', 'op': '==', 'value': 3},884                        ]885                    },886                    {887                        'not': [888                            {'field': 'name', 'op': '==', 'value': 'name_2'}889                        ]890                    },891                ],892            }893        ]894        filtered_query = apply_filters(stmt, filters)895        result = session.execute(filtered_query).scalars().all()896        assert len(result) == 1897        assert result[0].id == 3898    @pytest.mark.usefixtures('multiple_bars_inserted')899    def test_complex_using_tuples(self, session):900        stmt = select(Bar)901        filters = (902            {903                'and': (904                    {905                        'or': (906                            {'field': 'id', 'op': '==', 'value': 2},907                            {'field': 'id', 'op': '==', 'value': 3},908                        )909                    },910                    {911                        'not': (912                            {'field': 'name', 'op': '==', 'value': 'name_2'},913                        )914                    },915                ),916            },917        )918        filtered_query = apply_filters(stmt, filters)919        result = session.execute(filtered_query).scalars().all()920        assert len(result) == 1921        assert result[0].id == 3922class TestApplyArrayFilters:923    @pytest.mark.usefixtures('multiple_corges_inserted')924    def test_any_value_in_array(self, session, is_postgresql):925        if not is_postgresql:926            pytest.skip(ARRAY_NOT_SUPPORTED)927        stmt = select(Corge)928        filters = [{'field': 'tags', 'op': 'any', 'value': 'foo'}]929        filtered_query = apply_filters(stmt, filters)930        result = session.execute(filtered_query).scalars().all()931        assert len(result) == 2932        assert result[0].id == 2933        assert result[1].id == 3934    @pytest.mark.usefixtures('multiple_corges_inserted')935    def test_not_any_values_in_array(self, session, is_postgresql):936        if not is_postgresql:937            pytest.skip(ARRAY_NOT_SUPPORTED)938        stmt = select(Corge)939        filters = [{'field': 'tags', 'op': 'not_any', 'value': 'foo'}]940        filtered_query = apply_filters(stmt, filters)941        result = session.execute(filtered_query).scalars().all()942        assert len(result) == 2943        assert result[0].id == 1944        assert result[1].id == 4945class TestHybridAttributes:946    @pytest.mark.usefixtures('multiple_bars_inserted')947    @pytest.mark.parametrize(948        ('field, expected_error'),949        [950            ('foos', "Model <class 'test.models.Bar'> has no column `foos`."),951            (952                '__mapper__',953                "Model <class 'test.models.Bar'> has no column `__mapper__`.",954            ),955            (956                'not_valid',957                "Model <class 'test.models.Bar'> has no column `not_valid`.",958            ),959        ]960    )961    def test_orm_descriptors_not_valid_hybrid_attributes(962        self, session, field, expected_error963    ):964        stmt = select(Bar)965        filters = [966            {967                'model': 'Bar',968                'field': field,969                'op': '==',970                'value': 100971            }972        ]973        with pytest.raises(FieldNotFound) as exc:974            apply_filters(stmt, filters)975        assert expected_error in str(exc)976    @pytest.mark.usefixtures('multiple_bars_inserted')977    @pytest.mark.usefixtures('multiple_quxs_inserted')978    def test_filter_by_hybrid_properties(self, session):979        stmt = select(Bar, Qux)980        filters = [981            {982                'model': 'Bar',983                'field': 'count_square',984                'op': '==',985                'value': 100986            },987            {988                'model': 'Qux',989                'field': 'count_square',990                'op': '>=',991                'value': 26992            },993        ]994        filtered_query = apply_filters(stmt, filters)995        result = session.execute(filtered_query).all()996        assert len(result) == 2997        bars, quxs = zip(*result)998        assert set(map(type, bars)) == {Bar}999        assert {bar.id for bar in bars} == {2}1000        assert {bar.count_square for bar in bars} == {100}1001        assert set(map(type, quxs)) == {Qux}1002        assert {qux.id for qux in quxs} == {2, 4}1003        assert {qux.count_square for qux in quxs} == {100, 225}1004    @pytest.mark.usefixtures('multiple_bars_inserted')1005    @pytest.mark.usefixtures('multiple_quxs_inserted')1006    def test_filter_by_hybrid_methods(self, session):1007        stmt = select(Bar, Qux)1008        filters = [1009            {1010                'model': 'Bar',1011                'field': 'three_times_count',1012                'op': '==',1013                'value': 301014            },1015            {1016                'model': 'Qux',1017                'field': 'three_times_count',1018                'op': '>=',1019                'value': 311020            },1021        ]1022        filtered_query = apply_filters(stmt, filters)1023        result = session.execute(filtered_query).all()1024        assert len(result) == 11025        bars, quxs = zip(*result)1026        assert set(map(type, bars)) == {Bar}1027        assert {bar.id for bar in bars} == {2}1028        assert {bar.three_times_count() for bar in bars} == {30}1029        assert set(map(type, quxs)) == {Qux}1030        assert {qux.id for qux in quxs} == {4}1031        assert {qux.three_times_count() for qux in quxs} == {45}1032class TestQueryObject:1033    @pytest.mark.usefixtures('multiple_foos_inserted')1034    def test_auto_join(self, session):1035        query = session.query(Foo)1036        filters = [1037            {'field': 'name', 'op': '==', 'value': 'name_1'},1038            {'model': 'Bar', 'field': 'count', 'op': 'is_null'},1039        ]1040        filtered_query = apply_filters(query, filters)1041        result = filtered_query.all()1042        assert len(result) == 11043        assert result[0].id == 31044        assert result[0].bar_id == 3...

test_tools.py

Source:test_tools.py

...23    assert extract_filters({'filter_foo': 'bar'}) == ({'foo': 'bar'}, {})24    assert extract_filters({'exclude_foo': 'bar'}) == ({}, {'foo': 'bar'})25    assert extract_filters({'filter_foo': ''}) == ({'foo': ''}, {})26    assert extract_filters({'exclude_foo': ''}) == ({}, {'foo': ''})27def test_apply_filters():28    assert apply_filters({}, {}, {}) == {}29    assert apply_filters({'foo': 'bar'}, {}, {}) == {'foo': 'bar'}30    assert apply_filters({'foo': 'bar'}, {'foo': 'bar'}, {}) == {'foo': 'bar'}31    assert apply_filters({'foo': 'bar'}, {}, {'foo': 'bar'}) is None32    assert apply_filters({'foo': 'bar'}, {'baz': 'quux'}, {}) is None33    assert apply_filters({'foo': 'bar'}, {'foo': 'quux'}, {}) is None34    assert apply_filters({'foo': 'bar'}, {}, {'foo': 'quux'}) == {'foo': 'bar'}35    assert apply_filters({'foo': 'bar'}, {}, {'baz': 'quux'}) == {'foo': 'bar'}36def test_apply_filters_with_domains():37    assert apply_filters(38        {'checked-url': 'http://example.com/'}, {'domain': 'example.org'},39        {}) is None40    input = {'checked-url': 'http://example.org/'}41    assert apply_filters(input, {'domain': 'example.org'}, {}) == input42    assert apply_filters(input, {}, {'domain': 'example.org'}) is None43    input = {'checked-url': 'http://example.org/'}44    assert apply_filters(input, {'domain': 'example.com'}, {}) is None45    assert apply_filters(input, {}, {'domain': 'example.com'}) == input46    input = {47        'checked-url': 'http://example.com/',48        'status': '200',49    }50    assert apply_filters(51        input, {'domain': 'example.org', 'status': '200'}, {}) is None52    assert apply_filters(53        input, {'domain': 'example.org'}, {'status': '500'}) is None54    input = {55        'checked-url': 'http://example.org/',56        'status': '200',57    }58    assert apply_filters(59        input, {'domain': 'example.org', 'status': '200'}, {}) == input60    assert apply_filters(61        input, {'domain': 'example.org', 'status': '400'}, {}) is None62    assert apply_filters(63        input, {'domain': 'example.org'}, {'status': '200'}) is None64    input = {65        'checked-url': 'http://example.org/',66        'status': '200',67        'content-type': 'text/html',68    }69    assert apply_filters(70        input, {71            'domain': 'example.org',72            'status': '200',73            'content-type': 'text/html'74        }, {}) == input75    assert apply_filters(76        input,77        {'domain': 'example.org', 'status': '200'},78        {'content-type': 'text/html'}) is None79    assert apply_filters(80        input,81        {'domain': 'example.org', 'status': '500'},82        {'content-type': 'application/csv'}) is None83    assert apply_filters(84        input,85        {'domain': 'example.org'},86        {'content-type': 'application/csv', 'status': '500'}) == input87    assert apply_filters(88        input,89        {'domain': 'example.org', 'status': '200'},90        {'content-type': 'application/csv'}) == input91    assert apply_filters(92        input, {93            'status': '200',94            'content-type': 'text/html'95        }, {'domain': 'example.com'}) == input96    assert apply_filters(97        input,98        {'status': '200'},99        {'domain': 'example.org', 'content-type': 'application/csv'}) is None100    assert apply_filters(101        input,102        {'status': '500'},103        {'domain': 'example.com', 'content-type': 'application/csv'}) is None104    assert apply_filters(105        input,106        {}, {107            'domain': 'example.com',108            'content-type': 'application/csv',109            'status': '500'110        }) == input111    assert apply_filters(112        input,113        {'status': '200'},...

filters.py

Source:filters.py

...45        session[key]['user'] = '*'46    if 'paid' in filters:47        session[key]['paid'] = '*'48    session.save()49def apply_filters(request, key, filters = {'date'}):50    if key not in request.session or 'date_from' not in request.session[key] or 'date_to' not in request.session[key]:51        save_default_values(request, key, filters)52    for k in filters:53        if k != 'date' and k not in request.session[key]:54            save_default_values(request, key, filters)55            break56    if request.method == 'POST' and request.POST.get('date_filter_button', '') == 'add_filter':57        apply_filters = {}58        if 'date' in filters:59            date_from = request.POST.get('date_from', '').strip()60            date_to = request.POST.get('date_to', '').strip()61            try:62                datetime.strptime(date_from, date_filter_format())63                datetime.strptime(date_to, date_filter_format())...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.