def getConfirmationModeQuery(mode):
return {
'$or': [{
'corrected_mode': mode
}, {
'$and': [{
'corrected_mode': {
'$exists': False
}
}, {
'confirmed_mode': mode
}]
}, {
'$and': [{
'corrected_mode': {
'$exists': False
}
}, {
'confirmed_mode': {
'$exists': False
}
}] + Client.getClientConfirmedModeQueries(mode)
}]
}
def testClientConfirmedModeQueries(self):
queryDict = Client.getClientConfirmedModeQueries(4)[0]
self.assertTrue('$or' in queryDict)
queryList = queryDict['$or']
self.assertIn({'test_auto_confirmed.mode': 4}, queryList)