mirror of
https://github.com/mutouyun/cpp-ipc.git
synced 2025-12-06 16:56:45 +08:00
115 lines
2.5 KiB
C++
115 lines
2.5 KiB
C++
#include <cstring>
|
|
#include <cstdint>
|
|
#include <thread>
|
|
|
|
#include "shm.h"
|
|
#include "test.h"
|
|
|
|
using namespace ipc::shm;
|
|
|
|
namespace {
|
|
|
|
class Unit : public TestSuite {
|
|
Q_OBJECT
|
|
|
|
const char* name() const {
|
|
return "test_shm";
|
|
}
|
|
|
|
private slots:
|
|
void cleanupTestCase();
|
|
|
|
void test_acquire();
|
|
void test_release();
|
|
void test_get();
|
|
void test_hello();
|
|
void test_mt();
|
|
} unit__;
|
|
|
|
#include "test_shm.moc"
|
|
|
|
handle shm_hd__;
|
|
|
|
void Unit::cleanupTestCase() {
|
|
shm_hd__.release();
|
|
}
|
|
|
|
void Unit::test_acquire() {
|
|
QVERIFY(!shm_hd__.valid());
|
|
|
|
QVERIFY(shm_hd__.acquire("my-test-1", 1024));
|
|
QVERIFY(shm_hd__.size() == 1024);
|
|
QCOMPARE(shm_hd__.name(), "my-test-1");
|
|
|
|
QVERIFY(shm_hd__.acquire("my-test-2", 2048));
|
|
QVERIFY(shm_hd__.size() == 2048);
|
|
QCOMPARE(shm_hd__.name(), "my-test-2");
|
|
|
|
QVERIFY(shm_hd__.acquire("my-test-3", 4096));
|
|
QVERIFY(shm_hd__.size() == 4096);
|
|
QCOMPARE(shm_hd__.name(), "my-test-3");
|
|
}
|
|
|
|
void Unit::test_release() {
|
|
QVERIFY(shm_hd__.valid());
|
|
shm_hd__.release();
|
|
QVERIFY(!shm_hd__.valid());
|
|
}
|
|
|
|
void Unit::test_get() {
|
|
QVERIFY(shm_hd__.get() == nullptr);
|
|
QVERIFY(shm_hd__.acquire("my-test", 1024));
|
|
|
|
auto mem = shm_hd__.get();
|
|
QVERIFY(mem != nullptr);
|
|
QVERIFY(mem == shm_hd__.get());
|
|
|
|
std::uint8_t buf[1024] = {};
|
|
QVERIFY(memcmp(mem, buf, sizeof(buf)) == 0);
|
|
|
|
handle shm_other(shm_hd__.name(), shm_hd__.size());
|
|
QVERIFY(shm_other.get() != shm_hd__.get());
|
|
}
|
|
|
|
void Unit::test_hello() {
|
|
auto mem = shm_hd__.get();
|
|
QVERIFY(mem != nullptr);
|
|
|
|
constexpr char hello[] = "hello!";
|
|
std::memcpy(mem, hello, sizeof(hello));
|
|
QCOMPARE((char*)shm_hd__.get(), hello);
|
|
|
|
shm_hd__.release();
|
|
QVERIFY(shm_hd__.get() == nullptr);
|
|
QVERIFY(shm_hd__.acquire("my-test", 1024));
|
|
|
|
mem = shm_hd__.get();
|
|
QVERIFY(mem != nullptr);
|
|
std::uint8_t buf[1024] = {};
|
|
QVERIFY(memcmp(mem, buf, sizeof(buf)) == 0);
|
|
|
|
std::memcpy(mem, hello, sizeof(hello));
|
|
QCOMPARE((char*)shm_hd__.get(), hello);
|
|
}
|
|
|
|
void Unit::test_mt() {
|
|
std::thread {
|
|
[] {
|
|
handle shm_mt(shm_hd__.name(), shm_hd__.size());
|
|
|
|
shm_hd__.release();
|
|
|
|
constexpr char hello[] = "hello!";
|
|
QCOMPARE((char*)shm_mt.get(), hello);
|
|
}
|
|
}.join();
|
|
QVERIFY(shm_hd__.get() == nullptr);
|
|
QVERIFY(!shm_hd__.valid());
|
|
|
|
QVERIFY(shm_hd__.acquire("my-test", 1024));
|
|
std::uint8_t buf[1024] = {};
|
|
QVERIFY(memcmp(shm_hd__.get(), buf, sizeof(buf)) == 0);
|
|
}
|
|
|
|
} // internal-linkage
|