You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
3.6 KiB
117 lines
3.6 KiB
1 year ago
|
/*
|
||
|
|
** Zabbix
|
||
|
|
** Copyright (C) 2001-2023 Zabbix SIA
|
||
|
|
**
|
||
|
|
** 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 2 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, write to the Free Software
|
||
|
|
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||
|
|
**/
|
||
|
|
|
||
|
|
#include "zbxmocktest.h"
|
||
|
|
#include "zbxmockdata.h"
|
||
|
|
#include "zbxmockassert.h"
|
||
|
|
#include "zbxmockutil.h"
|
||
|
|
|
||
|
|
#include "zbxlog.h"
|
||
|
|
#include "zbxexpr.h"
|
||
|
|
|
||
|
|
static unsigned char mock_get_cycle(const char *path)
|
||
|
|
{
|
||
|
|
const char *value;
|
||
|
|
|
||
|
|
value = zbx_mock_get_parameter_string(path);
|
||
|
|
|
||
|
|
if (0 == strcmp(value, "ZBX_REPORT_CYCLE_DAILY"))
|
||
|
|
return ZBX_REPORT_CYCLE_DAILY;
|
||
|
|
if (0 == strcmp(value, "ZBX_REPORT_CYCLE_WEEKLY"))
|
||
|
|
return ZBX_REPORT_CYCLE_WEEKLY;
|
||
|
|
if (0 == strcmp(value, "ZBX_REPORT_CYCLE_MONTHLY"))
|
||
|
|
return ZBX_REPORT_CYCLE_MONTHLY;
|
||
|
|
if (0 == strcmp(value, "ZBX_REPORT_CYCLE_YEARLY"))
|
||
|
|
return ZBX_REPORT_CYCLE_YEARLY;
|
||
|
|
|
||
|
|
fail_msg("Unsupported report cycle: %s", value);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static unsigned char mock_get_weekdays(const char *path)
|
||
|
|
{
|
||
|
|
zbx_mock_handle_t hweekdays, hvalue;
|
||
|
|
zbx_mock_error_t err;
|
||
|
|
unsigned char weekdays = 0;
|
||
|
|
|
||
|
|
hweekdays = zbx_mock_get_parameter_handle(path);
|
||
|
|
while (ZBX_MOCK_END_OF_VECTOR != (err = (zbx_mock_vector_element(hweekdays, &hvalue))))
|
||
|
|
{
|
||
|
|
const char *value;
|
||
|
|
|
||
|
|
if (ZBX_MOCK_SUCCESS != err || ZBX_MOCK_SUCCESS != (err = zbx_mock_string(hvalue, &value)))
|
||
|
|
fail_msg("Cannot read weekday flag");
|
||
|
|
|
||
|
|
weekdays |= (1 << atoi(value));
|
||
|
|
}
|
||
|
|
|
||
|
|
return weekdays;
|
||
|
|
}
|
||
|
|
|
||
|
|
void zbx_mock_test_entry(void **state)
|
||
|
|
{
|
||
|
|
unsigned char cycle, weekdays;
|
||
|
|
zbx_timespec_t ts;
|
||
|
|
time_t now, nextcheck;
|
||
|
|
int start_time, step = 1;
|
||
|
|
zbx_mock_handle_t htimes, htime;
|
||
|
|
zbx_mock_error_t err;
|
||
|
|
char buf[MAX_STRING_LEN];
|
||
|
|
const char *report_tz;
|
||
|
|
|
||
|
|
ZBX_UNUSED(state);
|
||
|
|
|
||
|
|
report_tz = zbx_mock_get_parameter_string("in.timezone");
|
||
|
|
if (0 != setenv("TZ", report_tz, 1))
|
||
|
|
fail_msg("Cannot set 'TZ' environment variable: %s", zbx_strerror(errno));
|
||
|
|
|
||
|
|
tzset();
|
||
|
|
|
||
|
|
cycle = mock_get_cycle("in.cycle");
|
||
|
|
weekdays = mock_get_weekdays("in.weekdays");
|
||
|
|
|
||
|
|
if (ZBX_MOCK_SUCCESS != zbx_strtime_to_timespec(zbx_mock_get_parameter_string("in.now"), &ts))
|
||
|
|
fail_msg("Invalid time format for 'now' parameter");
|
||
|
|
now = ts.sec;
|
||
|
|
|
||
|
|
if (ZBX_MOCK_SUCCESS != zbx_strtime_to_timespec(zbx_mock_get_parameter_string("in.start_time"), &ts))
|
||
|
|
fail_msg("Invalid time format for 'start_time' parameter");
|
||
|
|
start_time = ts.sec;
|
||
|
|
|
||
|
|
htimes = zbx_mock_get_parameter_handle("out.reports");
|
||
|
|
while (ZBX_MOCK_END_OF_VECTOR != (err = (zbx_mock_vector_element(htimes, &htime))))
|
||
|
|
{
|
||
|
|
const char *value;
|
||
|
|
|
||
|
|
if (ZBX_MOCK_SUCCESS != err || ZBX_MOCK_SUCCESS != (err = zbx_mock_string(htime, &value)))
|
||
|
|
fail_msg("[%d] cannot read nextcheck timestamp", step);
|
||
|
|
|
||
|
|
if (ZBX_MOCK_SUCCESS != (err = zbx_strtime_to_timespec(value, &ts)))
|
||
|
|
fail_msg("[%d] cannot parse nextcheck timestamp: %s", step, zbx_mock_error_string(err));
|
||
|
|
|
||
|
|
if (FAIL == (nextcheck = zbx_get_report_nextcheck(now, cycle, weekdays, start_time, report_tz)))
|
||
|
|
fail_msg("[%d] cannot calculate report nextcheck", step);
|
||
|
|
|
||
|
|
zbx_snprintf(buf, sizeof(buf), "[%d] invalid report nextchek value", step++);
|
||
|
|
zbx_mock_assert_time_eq(buf, ts.sec, nextcheck);
|
||
|
|
|
||
|
|
now = nextcheck;
|
||
|
|
}
|
||
|
|
}
|