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_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 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, '')
def dataset_badge_labelizer(label, kind):
return DATASET_BADGE_KINDS.get(kind, '')
