def test_load_values(self, mock_process_missing, mock_process): mock_process.return_value = {'999999': {'onCampusAvail': '2'}} msg = update_ipeds.load_values() self.assertTrue('DRY' in msg) self.assertTrue(mock_process.call_count == 1) mock_process.return_value = { '243197': { 'onCampusAvail': '2', 'books': '.' } } msg = update_ipeds.load_values() self.assertTrue('DRY' in msg) self.assertTrue(mock_process.call_count == 2) msg = update_ipeds.load_values(dry_run=False) self.assertFalse('DRY' in msg) self.assertTrue(mock_process.call_count == 3) mock_process.return_value = {'243197': {'onCampusAvail': '1'}} msg = update_ipeds.load_values() self.assertTrue('DRY' in msg) self.assertTrue(mock_process.call_count == 4) mock_process.return_value = {'999998': {'onCampusAvail': '2'}} msg = update_ipeds.load_values(dry_run=False) self.assertFalse('DRY' in msg) self.assertTrue(mock_process.call_count == 5) self.assertTrue(mock_process_missing.call_count == 1)
def handle(self, *args, **options): if options["dry_run"].lower() == "false": msg = load_values(dry_run=False) elif options["dry_run"].lower() == "true": msg = load_values() else: msg = PARSER_HELP self.stdout.write(msg)
def handle(self, *args, **options): if options['dry_run'].lower() == 'false': msg = load_values(dry_run=False) elif options['dry_run'].lower() == 'true': msg = load_values() else: msg = PARSER_HELP self.stdout.write(msg)
def test_load_values(self, mock_process_missing, mock_process): mock_process.return_value = {'999999': {'onCampusAvail': '2'}} msg = update_ipeds.load_values() self.assertTrue('DRY' in msg) self.assertTrue(mock_process.call_count == 1) mock_process.return_value = {'243197': {'onCampusAvail': '2', 'books': '.'}} msg = update_ipeds.load_values() self.assertTrue('DRY' in msg) self.assertTrue(mock_process.call_count == 2) msg = update_ipeds.load_values(dry_run=False) self.assertFalse('DRY' in msg) self.assertTrue(mock_process.call_count == 3) mock_process.return_value = {'243197': {'onCampusAvail': '1'}} msg = update_ipeds.load_values() self.assertTrue('DRY' in msg) self.assertTrue(mock_process.call_count == 4) mock_process.return_value = {'999998': {'onCampusAvail': '2'}} msg = update_ipeds.load_values(dry_run=False) self.assertFalse('DRY' in msg) self.assertTrue(mock_process.call_count == 5) self.assertTrue(mock_process_missing.call_count == 1)