test_zone_events.c 2.68 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*  Copyright (C) 2014 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 <http://www.gnu.org/licenses/>.
*/

#include <tap/basic.h>

19
#include "knot/common/evsched.h"
Jan Včelák's avatar
Jan Včelák committed
20
#include "knot/worker/pool.h"
21
#include "knot/events/events.h"
Jan Včelák's avatar
Jan Včelák committed
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
#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

37
	zone_events_schedule_at(zone, ZONE_EVENT_EXPIRE, now + offset);
Jan Včelák's avatar
Jan Včelák committed
38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
	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");

54
	zone_events_schedule_at(zone, ZONE_EVENT_FLUSH, 0);
Jan Včelák's avatar
Jan Včelák committed
55 56 57 58

	timestamp = zone_events_get_next(zone, &event);
	ok(timestamp >= now + offset && event == ZONE_EVENT_EXPIRE, "expire is next");

59
	zone_events_schedule_at(zone, ZONE_EVENT_EXPIRE, 0);
Jan Včelák's avatar
Jan Včelák committed
60 61 62 63 64 65 66 67 68 69

	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
}

70 71 72 73
int main(void)
{
	plan_lazy();

Jan Včelák's avatar
Jan Včelák committed
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
	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");

89
	r = zone_events_setup(&zone, pool, &sched, NULL);
Jan Včelák's avatar
Jan Včelák committed
90 91 92 93 94 95 96
	ok(r == KNOT_EOK, "zone events setup");

	test_scheduling(&zone);

	zone_events_deinit(&zone);
	worker_pool_destroy(pool);
	evsched_deinit(&sched);
97 98 99

	return 0;
}