139 lines
4.0 KiB
C++
139 lines
4.0 KiB
C++
/* 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 as published by
|
|
the Free Software Foundation; version 2 of the License.
|
|
|
|
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,
|
|
51 Franklin Street, Suite 500, Boston, MA 02110-1335 USA */
|
|
|
|
#include <my_global.h>
|
|
#include <my_thread.h>
|
|
#include <pfs_instr.h>
|
|
#include <pfs_stat.h>
|
|
#include <pfs_global.h>
|
|
#include <pfs_user.h>
|
|
#include <pfs_buffer_container.h>
|
|
#include <tap.h>
|
|
|
|
#include "stub_pfs_global.h"
|
|
#include "stub_global_status_var.h"
|
|
|
|
#include <string.h> /* memset */
|
|
|
|
void test_oom()
|
|
{
|
|
PSI *psi;
|
|
PFS_global_param param;
|
|
PSI_bootstrap *boot;
|
|
|
|
memset(& param, 0xFF, sizeof(param));
|
|
param.m_enabled= true;
|
|
param.m_mutex_class_sizing= 0;
|
|
param.m_rwlock_class_sizing= 0;
|
|
param.m_cond_class_sizing= 0;
|
|
param.m_thread_class_sizing= 10;
|
|
param.m_table_share_sizing= 0;
|
|
param.m_file_class_sizing= 0;
|
|
param.m_socket_class_sizing= 0;
|
|
param.m_mutex_sizing= 0;
|
|
param.m_rwlock_sizing= 0;
|
|
param.m_cond_sizing= 0;
|
|
param.m_thread_sizing= 1000;
|
|
param.m_table_sizing= 0;
|
|
param.m_file_sizing= 0;
|
|
param.m_file_handle_sizing= 0;
|
|
param.m_socket_sizing= 0;
|
|
param.m_events_waits_history_sizing= 10;
|
|
param.m_events_waits_history_long_sizing= 0;
|
|
param.m_setup_actor_sizing= 0;
|
|
param.m_setup_object_sizing= 0;
|
|
param.m_host_sizing= 0;
|
|
param.m_user_sizing= 1000;
|
|
param.m_account_sizing= 0;
|
|
param.m_stage_class_sizing= 50;
|
|
param.m_events_stages_history_sizing= 0;
|
|
param.m_events_stages_history_long_sizing= 0;
|
|
param.m_statement_class_sizing= 50;
|
|
param.m_events_statements_history_sizing= 0;
|
|
param.m_events_statements_history_long_sizing= 0;
|
|
param.m_events_transactions_history_sizing= 0;
|
|
param.m_events_transactions_history_long_sizing= 0;
|
|
param.m_digest_sizing= 0;
|
|
param.m_session_connect_attrs_sizing= 0;
|
|
param.m_program_sizing= 0;
|
|
param.m_statement_stack_sizing= 0;
|
|
param.m_memory_class_sizing= 10;
|
|
param.m_metadata_lock_sizing= 0;
|
|
param.m_max_digest_length= 0;
|
|
param.m_max_sql_text_length= 0;
|
|
|
|
/* Setup */
|
|
|
|
stub_alloc_always_fails= false;
|
|
stub_alloc_fails_after_count= 1000;
|
|
|
|
pre_initialize_performance_schema();
|
|
boot= initialize_performance_schema(¶m);
|
|
psi= (PSI *)boot->get_interface(PSI_VERSION_1);
|
|
|
|
PSI_thread_key thread_key_1;
|
|
PSI_thread_info all_thread[]=
|
|
{
|
|
{&thread_key_1, "T-1", 0}
|
|
};
|
|
psi->register_thread("test", all_thread, 1);
|
|
|
|
PSI_thread *thread_1= psi->new_thread(thread_key_1, NULL, 0);
|
|
psi->set_thread(thread_1);
|
|
|
|
/* Tests */
|
|
|
|
int first_fail= 1;
|
|
stub_alloc_fails_after_count= first_fail;
|
|
psi->set_thread_account("user1", 5, "", 0);
|
|
ok(global_user_container.m_lost == 1, "oom (user)");
|
|
|
|
stub_alloc_fails_after_count= first_fail + 1;
|
|
psi->set_thread_account("user2", 5, "", 0);
|
|
ok(global_user_container.m_lost == 2, "oom (user waits)");
|
|
|
|
stub_alloc_fails_after_count= first_fail + 2;
|
|
psi->set_thread_account("user3", 5, "", 0);
|
|
ok(global_user_container.m_lost == 3, "oom (user stages)");
|
|
|
|
stub_alloc_fails_after_count= first_fail + 3;
|
|
psi->set_thread_account("user4", 5, "", 0);
|
|
ok(global_user_container.m_lost == 4, "oom (user statements)");
|
|
|
|
stub_alloc_fails_after_count= first_fail + 4;
|
|
psi->set_thread_account("user5", 5, "", 0);
|
|
ok(global_user_container.m_lost == 5, "oom (user transactions)");
|
|
|
|
stub_alloc_fails_after_count= first_fail + 5;
|
|
psi->set_thread_account("user6", 5, "", 0);
|
|
ok(global_user_container.m_lost == 6, "oom (user memory)");
|
|
|
|
shutdown_performance_schema();
|
|
}
|
|
|
|
void do_all_tests()
|
|
{
|
|
test_oom();
|
|
}
|
|
|
|
int main(int, char **)
|
|
{
|
|
plan(6);
|
|
MY_INIT("pfs_user-oom-t");
|
|
do_all_tests();
|
|
return (exit_status());
|
|
}
|
|
|