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)