def test_get_channels_categories_is_up(self, client, initial_data):
channel = initial_data['channel']
categories = ['Books', 'National Literature', 'Science Fiction']
Category.create_channel_categories(channel=channel,
categories=categories)
url = reverse('core:category_list',
kwargs={'channel': initial_data['channel'].pk})
response = client.get(url)
assert response.status_code == 200
def handle(self, *args, **options):
channel, _ = Channel.objects.get_or_create(name=options['channel'])
Category.objects.filter(channel=channel).delete()
with open(options['file']) as categories_file:
categories_reader = csv.reader(categories_file)
next(categories_reader)
for row in categories_reader:
count = Category.create_channel_categories(channel, row[0].split(' / '))
self.stdout.write('Processed {} Categories'.format(count))
self.stdout.write(self.style.SUCCESS('Categories imported!'))
def test_create_categories_from_list(self, initial_data):
channel = initial_data['channel']
categories = ['Books', 'National Literature', 'Science Fiction']
created = Category.create_channel_categories(channel=channel,
categories=categories)
assert created == len(categories)