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.
118 lines
3.5 KiB
118 lines
3.5 KiB
/* Copyright (c) 2011, 2017, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
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, version 2.0, 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 St, Fifth Floor, Boston, MA 02110-1301 USA */
|
|
|
|
#ifndef TEST_UTILS_INCLUDED
|
|
#define TEST_UTILS_INCLUDED
|
|
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "my_compiler.h"
|
|
#include "my_sys.h"
|
|
#include "mysql/components/services/mysql_mutex_bits.h"
|
|
#include "nullable.h"
|
|
#include "sql/error_handler.h"
|
|
#include "sql/sql_error.h"
|
|
|
|
class THD;
|
|
class my_decimal;
|
|
struct MEM_ROOT;
|
|
|
|
extern thread_local THD *current_thd;
|
|
extern thread_local MEM_ROOT **THR_MALLOC;
|
|
extern mysql_mutex_t LOCK_open;
|
|
extern uint opt_debug_sync_timeout;
|
|
extern "C" void sql_alloc_error_handler(void);
|
|
|
|
namespace my_testing {
|
|
|
|
inline int native_compare(size_t *length, unsigned char **a,
|
|
unsigned char **b) {
|
|
return memcmp(*a, *b, *length);
|
|
}
|
|
|
|
inline qsort2_cmp get_ptr_compare(size_t size MY_ATTRIBUTE((unused))) {
|
|
return (qsort2_cmp)native_compare;
|
|
}
|
|
|
|
void setup_server_for_unit_tests();
|
|
void teardown_server_for_unit_tests();
|
|
int chars_2_decimal(const char *chars, my_decimal *to);
|
|
|
|
/*
|
|
A class which wraps the necessary setup/teardown logic for
|
|
unit tests which depend on a working THD environment.
|
|
*/
|
|
class Server_initializer {
|
|
public:
|
|
Server_initializer() : m_thd(NULL) {}
|
|
|
|
// Invoke these from corresponding functions in test fixture classes.
|
|
void SetUp();
|
|
void TearDown();
|
|
|
|
// Sets expected error for error_handler_hook.
|
|
static void set_expected_error(uint val);
|
|
|
|
THD *thd() const { return m_thd; }
|
|
|
|
private:
|
|
THD *m_thd;
|
|
};
|
|
|
|
/**
|
|
A mock error handler which registers itself with the THD in the CTOR,
|
|
and unregisters in the DTOR. The function handle_condition() will
|
|
verify that it is called with the expected error number.
|
|
The DTOR will verify that handle_condition() has actually been called.
|
|
*/
|
|
class Mock_error_handler : public Internal_error_handler {
|
|
public:
|
|
Mock_error_handler(THD *thd, uint expected_error);
|
|
virtual ~Mock_error_handler();
|
|
|
|
virtual bool handle_condition(THD *thd, uint sql_errno, const char *sqlstate,
|
|
Sql_condition::enum_severity_level *level,
|
|
const char *msg);
|
|
|
|
int handle_called() const { return m_handle_called; }
|
|
|
|
private:
|
|
THD *m_thd;
|
|
uint m_expected_error;
|
|
int m_handle_called;
|
|
};
|
|
|
|
/*
|
|
A class which wraps the necessary setup/teardown logic for
|
|
Data Dictionary.
|
|
*/
|
|
class DD_initializer {
|
|
public:
|
|
static void SetUp();
|
|
static void TearDown();
|
|
};
|
|
|
|
} // namespace my_testing
|
|
|
|
#endif // TEST_UTILS_INCLUDED
|
|
|