This commit is contained in:
Nick Sweeting
2026-03-23 03:58:32 -07:00
parent 268856bcfb
commit b749b26c5d
286 changed files with 21704 additions and 13480 deletions

View File

@@ -12,25 +12,25 @@ 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',
username="api-user",
password="testpass123",
email="api@example.com",
)
def test_schedule_api_creates_schedule(self):
request = RequestFactory().post('/api/v1/cli/schedule')
request = RequestFactory().post("/api/v1/cli/schedule")
request.user = self.user
setattr(request, 'stdout', StringIO())
setattr(request, 'stderr', StringIO())
setattr(request, "stdout", StringIO())
setattr(request, "stderr", StringIO())
args = ScheduleCommandSchema(
every='daily',
import_path='https://example.com/feed.xml',
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.assertTrue(response["success"])
self.assertEqual(response["result_format"], "json")
self.assertEqual(CrawlSchedule.objects.count(), 1)
self.assertEqual(len(response['result']['created_schedule_ids']), 1)
self.assertEqual(len(response["result"]["created_schedule_ids"]), 1)