mirror of
https://gitlab.nic.cz/knot/knot-dns.git
synced 2026-03-06 23:40:36 -05:00
99 lines
2.7 KiB
C
99 lines
2.7 KiB
C
/* Copyright (C) 2019 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <tap/basic.h>
|
|
|
|
#include "knot/common/evsched.h"
|
|
#include "knot/worker/pool.h"
|
|
#include "knot/events/events.h"
|
|
#include "knot/zone/zone.h"
|
|
|
|
static void test_scheduling(zone_t *zone)
|
|
{
|
|
const time_t now = time(NULL);
|
|
const unsigned offset = 1000;
|
|
|
|
time_t timestamp = 0;
|
|
zone_event_type_t event = 0;
|
|
|
|
timestamp = zone_events_get_next(zone, &event);
|
|
ok(timestamp < 0 && event == ZONE_EVENT_INVALID, "nothing planned");
|
|
|
|
// scheduling
|
|
|
|
zone_events_schedule_at(zone, ZONE_EVENT_EXPIRE, now + offset);
|
|
zone_events_schedule_at(zone, ZONE_EVENT_FLUSH, now + (offset / 2));
|
|
|
|
for (int i = 0; i < ZONE_EVENT_COUNT; i++) {
|
|
time_t t = zone_events_get_time(zone, i);
|
|
bool scheduled = i == ZONE_EVENT_EXPIRE || i == ZONE_EVENT_FLUSH;
|
|
const char *name = zone_events_get_name(i);
|
|
|
|
ok((t > 0) == scheduled && name, "event %s (%s)", name,
|
|
scheduled ? "scheduled" : "not scheduled");
|
|
}
|
|
|
|
// queuing
|
|
|
|
timestamp = zone_events_get_next(zone, &event);
|
|
ok(timestamp >= now + (offset / 2) && event == ZONE_EVENT_FLUSH, "flush is next");
|
|
|
|
zone_events_schedule_at(zone, ZONE_EVENT_FLUSH, (time_t)0);
|
|
|
|
timestamp = zone_events_get_next(zone, &event);
|
|
ok(timestamp >= now + offset && event == ZONE_EVENT_EXPIRE, "expire is next");
|
|
|
|
zone_events_schedule_at(zone, ZONE_EVENT_EXPIRE, (time_t)0);
|
|
|
|
timestamp = zone_events_get_next(zone, &event);
|
|
ok(timestamp < 0 && event == ZONE_EVENT_INVALID, "nothing planned");
|
|
|
|
// zone_events_enqueue
|
|
|
|
// zone_events_freeze
|
|
// zone_events_start
|
|
}
|
|
|
|
int main(void)
|
|
{
|
|
plan_lazy();
|
|
|
|
int r;
|
|
|
|
evsched_t sched = { 0 };
|
|
worker_pool_t *pool = NULL;
|
|
zone_t zone = { 0 };
|
|
|
|
r = evsched_init(&sched, NULL);
|
|
ok(r == KNOT_EOK, "create scheduler");
|
|
|
|
pool = worker_pool_create(1);
|
|
ok(pool != NULL, "create worker pool");
|
|
|
|
r = zone_events_init(&zone);
|
|
ok(r == KNOT_EOK, "zone events init");
|
|
|
|
r = zone_events_setup(&zone, pool, &sched);
|
|
ok(r == KNOT_EOK, "zone events setup");
|
|
|
|
test_scheduling(&zone);
|
|
|
|
zone_events_deinit(&zone);
|
|
worker_pool_destroy(pool);
|
|
evsched_deinit(&sched);
|
|
|
|
return 0;
|
|
}
|