def testConfirmationModeQueryCorrectedManualAndAuto(self): from dao.client import Client (user, dummySection, dummyPredModeMap) = self.setupClientTest() clientSetQuery = Client(user.getFirstStudy()).clientSpecificSetters(user.uuid, dummySection, dummyPredModeMap) # Apply the change get_section_db().update({'_id': dummySection['_id']}, clientSetQuery) retrievedSection = get_section_db().find_one({'_id': dummySection['_id']}) self.assertEqual(retrievedSection['test_auto_confirmed']['mode'], 1) get_section_db().update({'_id': dummySection['_id']}, {'$set': {'confirmed_mode': 4}}) get_section_db().update({'_id': dummySection['_id']}, {'$set': {'corrected_mode': 9}}) retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(1)) for entry in retrieveByQuery: print entry self.assertEqual(retrieveByQuery.count(), 0) retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(4)) for entry in retrieveByQuery: print entry self.assertEqual(retrieveByQuery.count(), 0) retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(9)) for entry in retrieveByQuery: print entry self.assertEqual(retrieveByQuery.count(), 1)
def testConfirmationModeQueryNeither(self): from dao.client import Client (user, dummySection, dummyPredModeMap) = self.setupClientTest() retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(1)) self.assertEqual(retrieveByQuery.count(), 0) retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(4)) self.assertEqual(retrieveByQuery.count(), 0)
def testConfirmationModeQueryManualNotAuto(self): from dao.client import Client (user, dummySection, dummyPredModeMap) = self.setupClientTest() get_section_db().update({'_id': dummySection['_id']}, {'$set': {'confirmed_mode': 4}}) retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(1)) self.assertEqual(retrieveByQuery.count(), 0) retrieveByQuery = get_section_db().find(common.getConfirmationModeQuery(4)) self.assertEqual(retrieveByQuery.count(), 1)