UnifiedAlchemyMagicMock does not work with .label()

[hozher]

Hello,

I have this query in my function to be tested:

query = session.query(
            whitelist.whitelistEmailId,
            whitelist.areaId,
            whitelist.businessNameId,
            whitelist.email,
            whitelist.identificationNumber,
            whitelist.isRegistered,
            whitelist.serviceTypeId,
            area.name.label('area'),
            business.name.label('businessName'),
            service.name.label('serviceType')
            ).\
        join(area, whitelist.areaId == area.areaId).\
        join(business, whitelist.businessNameId == business.businessNameId).\
        join(service, whitelist.serviceTypeId == service.serviceTypeId)

I am trying to mock it out like this:

return UnifiedAlchemyMagicMock(data=[
        (
            [mock.call.query(
                WhitelistedEmail.whitelistEmailId,
                WhitelistedEmail.areaId,
                WhitelistedEmail.businessNameId,
                WhitelistedEmail.email,
                WhitelistedEmail.identificationNumber,
                WhitelistedEmail.isRegistered,
                WhitelistedEmail.serviceTypeId,
                Area.name.label('area'),
                BusinessName.name.label('businessName'),
                ServiceType.name.label('serviceType')
            )], Factory.get_list()
        )
    ])

After Ive added these three guys, that make use of .label() does not work anymore.

    Area.name.label('area'),
    BusinessName.name.label('businessName'),
    ServiceType.name.label('serviceType')

Is this a known issue? or might be am I doing something wrong. any clue is more than welcome.

[miki725]

dont think tested with labels. could be the comparison logic does not correctly compare labels. Can you try using

alchemy-mock/alchemy_mock/comparison.py

Line 60 in e8a9d61

class ExpressionMatcher(PrettyExpression):

to manually compare labels?
so something like:

ExpressionMatcher(Area.name.label('area')) == Area.name.label('area')

[hozher]

@miki725 Thanks for answering.

The problem is that when query gets called is not returning any value (not getting mocked) I did the ExpressionMatcher whiteout success as follows:

return UnifiedAlchemyMagicMock(data=[
        (
            [mock.call.query(
                WhitelistedEmail.whitelistEmailId,
                WhitelistedEmail.areaId,
                WhitelistedEmail.businessNameId,
                WhitelistedEmail.email,
                WhitelistedEmail.identificationNumber,
                WhitelistedEmail.isRegistered,
                WhitelistedEmail.serviceTypeId,
                ExpressionMatcher(Area.name.label('area')) == Area.name.label('area'),
                ExpressionMatcher(BusinessName.name.label('businessName')) == BusinessName.name.label('businessName'),
                ExpressionMatcher(ServiceType.name.label('serviceType')) == ServiceType.name.label('serviceType'),
            )], Factory.get_list()
        )
    ])

[miki725]

I meant can you post results running ExpressionMatcher by itself. I dont know if its capable of comparing label objects

[hozher]

alright, i got this:

 print(ExpressionMatcher(Area.name.label('area')) == Area.name.label('area'))
    print(ExpressionMatcher(BusinessName.name.label('businessName')) == BusinessName.name.label('businessName'))
    print(ExpressionMatcher(ServiceType.name.label('serviceType')) == ServiceType.name.label('serviceType'))

"Area".name = "Area".name
"BusinessName".name = "BusinessName".name
"ServiceType".name = "ServiceType".name

[miki725]

yep thats not supported. should of gotten a boolean if it was. Ill try to add support at some point or feel free to submit a PR!

[miki725]

#25 adds labels support. can you test that branch?

[hozher]

It works pretty well 😃

>>> from alchemy_mock.comparison import ExpressionMatcher
>>> from model import Area
>>> ExpressionMatcher(Area.name.label('area')) == Area.name.label('area')
True

Thanks a lot! merge, might be?

[miki725]

just released 0.4.2. should be able to install from pypi. let me know if any further issues