Zabbix

/*

** Zabbix

** Copyright (C) 2001-2024 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 "zbxmockdb.h"

​

/* make sure that __wrap_*() prototypes match unwrapped counterparts */

​

#define zbx_db_vselect  __wrap_zbx_db_vselect

#define zbx_db_fetch_basic  __wrap_zbx_db_fetch_basic

#define zbx_db_fetch        __wrap_zbx_db_fetch_basic

#define zbx_db_free_result  __wrap_zbx_db_free_result

#include "zbxdb.h"

#undef zbx_db_vselect

#undef zbx_db_fetch_basic

#undef zbx_db_fetch

#undef zbx_db_free_result

​

#define __zbx_db_execute        __wrap___zbx_db_execute

#define zbx_db_execute_multiple_query   __wrap_zbx_db_execute_multiple_query

#define zbx_db_begin            __wrap_zbx_db_begin

#define zbx_db_commit           __wrap_xbc_db_commit

#include "zbxdbhigh.h"

#undef __zbx_db_execute

#undef zbx_db_execute_multiple_query

#undef zbx_db_begin

#undef zbx_db_commit

​

#define ZBX_MOCK_DB_RESULT_COLUMNS_MAX  128

​

typedef struct

{

    char    *data_source;

    int num;

}

zbx_mockdb_query_t;

​

typedef struct

{

    zbx_hashset_t   queries;

}

zbx_mockdb_t;

​

static zbx_mockdb_t mockdb;

​

struct zbx_db_result

{

    zbx_db_row_t        row;

    char            *data_source;   /* for error messages */

    zbx_mock_handle_t   rows;

    int         row_to_fetch;   /* for error messages */

    int         columns;    /* to make sure that rows have identical number of columns */

};

​

zbx_db_result_t __fwd_zbx_db_select(const char *fmt, ...);

zbx_db_result_t __wrap_zbx_db_select_n_basic(const char *query, int n);

int __wrap___zbx_db_execute(const char *fmt, ...);

int __wrap_zbx_db_commit(void);

​

/* zbx_mockdb_t:queries hashset support */

static zbx_hash_t   mockdb_query_hash(const void *data)

{

    const zbx_mockdb_query_t    *query = (const zbx_mockdb_query_t *)data;

    return ZBX_DEFAULT_STRING_HASH_FUNC(query->data_source);

}

​

static int  mockdb_query_compare(const void *d1, const void *d2)

{

    const zbx_mockdb_query_t    *q1 = (const zbx_mockdb_query_t *)d1;

    const zbx_mockdb_query_t    *q2 = (const zbx_mockdb_query_t *)d2;

​

    return strcmp(q1->data_source, q2->data_source);

}

​