from io import StringIO

from django.contrib.auth import get_user_model
from django.test import RequestFactory, TestCase

from archivebox.api.v1_cli import ScheduleCommandSchema, cli_schedule
from archivebox.crawls.models import CrawlSchedule

User = get_user_model()


class CLIScheduleAPITests(TestCase):
    def setUp(self):
        self.user = User.objects.create_user(
            username='api-user',
            password='testpass123',
            email='api@example.com',
        )

    def test_schedule_api_creates_schedule(self):
        request = RequestFactory().post('/api/v1/cli/schedule')
        request.user = self.user
        setattr(request, 'stdout', StringIO())
        setattr(request, 'stderr', StringIO())
        args = ScheduleCommandSchema(
            every='daily',
            import_path='https://example.com/feed.xml',
            quiet=True,
        )

        response = cli_schedule(request, args)

        self.assertTrue(response['success'])
        self.assertEqual(response['result_format'], 'json')
        self.assertEqual(CrawlSchedule.objects.count(), 1)
        self.assertEqual(len(response['result']['created_schedule_ids']), 1)