def test_list(self): response = self.get(url_for('api.available_dataset_badges')) self.assertStatus(response, 200) self.assertEqual(len(response.json), len(DATASET_BADGE_KINDS)) for kind, label in DATASET_BADGE_KINDS.items(): self.assertIn(kind, response.json) self.assertEqual(response.json[kind], label)
def test_create_2nd(self): # Explicitely setting the kind to avoid collisions given the # small number of choices for kinds. kinds_keys = DATASET_BADGE_KINDS.keys() self.dataset.badges.append(DatasetBadgeFactory(kind=kinds_keys[0])) self.dataset.save() data = DatasetBadgeFactory.attributes() data['kind'] = kinds_keys[1] with self.api_user(): response = self.post( url_for('api.dataset_badges', dataset=self.dataset), data) self.assertStatus(response, 201) self.dataset.reload() self.assertEqual(len(self.dataset.badges), 2)
def dataset_badge_labelizer(label, kind): return DATASET_BADGE_KINDS.get(kind, '')