/*
|
* =====================================================================================
|
*
|
* Filename: simple_tests.cpp
|
*
|
* Description:
|
*
|
* Version: 1.0
|
* Created: 2021年03月30日 11时29分40秒
|
* Revision: none
|
* Compiler: gcc
|
*
|
* Author: YOUR NAME (),
|
* Organization:
|
*
|
* =====================================================================================
|
*/
|
|
#include "util.h"
|
|
struct s1000 {
|
char a[1000];
|
};
|
|
BOOST_AUTO_TEST_CASE(BasicTest)
|
{
|
const std::string shm_name("basic");
|
ShmRemover auto_remove(shm_name);
|
SharedMemory shm(shm_name, 1024 * 1024 * 10);
|
auto Avail = [&]() { return shm.get_free_memory(); };
|
|
auto init_avail = Avail();
|
|
auto BasicTest = [&](int tid, int nloop) {
|
auto Code = [&](int id) {
|
typedef NamedShmObject<s1000> Int;
|
std::string name = std::to_string(id);
|
auto a0 = Avail();
|
Int i1(shm, name, eOpenOrCreate);
|
auto a1 = Avail();
|
BOOST_CHECK_LT(a1, a0);
|
printf("s1000 size: %ld\n", a0 - a1);
|
i1->a[0] = 5;
|
Int i2(shm, name, eOpenOrCreate);
|
auto a2 = Avail();
|
BOOST_CHECK_EQUAL(a1, a2);
|
BOOST_CHECK_EQUAL(i1.data(), i2.data());
|
int i = i1.Remove();
|
BOOST_CHECK_EQUAL(Avail(), a0);
|
|
{
|
auto old = Avail();
|
void *p = shm.Alloc(1024);
|
shm.Dealloc(p);
|
BOOST_CHECK_EQUAL(old, Avail());
|
}
|
|
bool r = shared_memory_object::remove(shm_name.c_str());
|
BOOST_CHECK(r);
|
};
|
for (int i = 0; i < nloop; ++i) {
|
Code(i + tid * nloop);
|
}
|
};
|
|
// boost::timer::auto_cpu_timer timer;
|
ThreadManager threads;
|
int nthread = 1;
|
int nloop = 1;
|
for (int i = 0; i < nthread; ++i) {
|
threads.Launch(BasicTest, i, nloop);
|
}
|
BOOST_CHECK_EQUAL(init_avail, Avail());
|
}
|
|
BOOST_AUTO_TEST_CASE(ForkTest)
|
{
|
ProcessManager procs;
|
const int nproc = 10;
|
|
printf("Testing fork:\n");
|
|
auto child = [&](int id) {
|
std::this_thread::sleep_for(100ms * id);
|
printf("child id: %3d/%d ends\r", id, nproc);
|
};
|
|
for (int i = 0; i < nproc; ++i) {
|
procs.Launch(child, i + 1);
|
}
|
}
|
|
BOOST_AUTO_TEST_CASE(RefCountTest)
|
{
|
SharedMemory &shm = TestShm();
|
typedef MsgI Msg;
|
GlobalInit(shm);
|
|
Msg m0(1000, shm);
|
BOOST_CHECK(!m0.valid());
|
m0.Make(100);
|
BOOST_CHECK_EQUAL(m0.Count(), 1);
|
Msg m1 = m0;
|
BOOST_CHECK(m1.valid());
|
BOOST_CHECK_EQUAL(m1.AddRef(), 2);
|
BOOST_CHECK_EQUAL(m0.AddRef(), 3);
|
BOOST_CHECK_EQUAL(m0.Release(), 2);
|
BOOST_CHECK_EQUAL(m0.Release(), 1);
|
BOOST_CHECK_EQUAL(m1.Release(), 0);
|
BOOST_CHECK(!m1.valid());
|
}
|
