129 lines
3.3 KiB
C++
129 lines
3.3 KiB
C++
#include <catch_amalgamated.hpp>
|
|
#include <cstdint>
|
|
|
|
#include "../client/client.h"
|
|
#include "../client/config.h"
|
|
#include "../server/server.h"
|
|
#include "assistant.h"
|
|
|
|
std::shared_ptr<TransmServer> server;
|
|
std::shared_ptr<TransmClient> clientA;
|
|
std::shared_ptr<TransmClient> clientB;
|
|
std::shared_ptr<ClientConfig> config;
|
|
|
|
asio::io_context server_context;
|
|
|
|
constexpr char* ip = "127.0.0.1";
|
|
constexpr unsigned short port = 9897;
|
|
bool server_suc = false;
|
|
constexpr unsigned int max_wait = 3000;
|
|
constexpr unsigned int wait_interval = 100;
|
|
|
|
std::string str_id_a;
|
|
int ida_in_b = -1;
|
|
std::thread server_th;
|
|
|
|
bool test_ls();
|
|
|
|
void server_run()
|
|
{
|
|
server = std::make_shared<TransmServer>(server_context);
|
|
if (!server->start(port)) {
|
|
server_suc = false;
|
|
return;
|
|
}
|
|
server_suc = true;
|
|
server_context.run();
|
|
}
|
|
|
|
bool base_connect()
|
|
{
|
|
config = std::make_shared<ClientConfig>();
|
|
if (!config->baseInit()) {
|
|
return false;
|
|
}
|
|
|
|
server_th = std::thread(server_run);
|
|
if (value_wait<bool>([]() -> bool { return server_suc; }, true, std::equal_to<bool>(), max_wait,
|
|
wait_interval) == false) {
|
|
return false;
|
|
}
|
|
|
|
clientA = std::make_shared<TransmClient>();
|
|
if (clientA->connect_for_test(ip, std::to_string(port), config->get_config_dir()) == false) {
|
|
return false;
|
|
}
|
|
|
|
clientB = std::make_shared<TransmClient>();
|
|
if (clientB->connect_for_test(ip, std::to_string(port), config->get_config_dir()) == false) {
|
|
return false;
|
|
}
|
|
|
|
if (value_wait<std::string>([]() -> std::string { return clientA->test_get_own_id(); }, std::string(),
|
|
std::not_equal_to<std::string>(), max_wait, wait_interval) == false) {
|
|
return false;
|
|
}
|
|
|
|
if (value_wait<std::string>([]() -> std::string { return clientB->test_get_own_id(); }, std::string(),
|
|
std::not_equal_to<std::string>(), max_wait, wait_interval) == false) {
|
|
return false;
|
|
}
|
|
|
|
str_id_a = clientA->test_get_own_id();
|
|
std::cout << "clientA id: " << str_id_a << std::endl;
|
|
if (value_wait<int>([]() -> int { return clientB->test_index_by_id(str_id_a); }, -1,
|
|
std::not_equal_to<int>(), max_wait, wait_interval) == false) {
|
|
return false;
|
|
}
|
|
ida_in_b = clientB->test_index_by_id(str_id_a);
|
|
std::cout << "clientA index In B: " << ida_in_b << std::endl;
|
|
return true;
|
|
}
|
|
|
|
bool main_test()
|
|
{
|
|
if (!base_connect()) {
|
|
return false;
|
|
}
|
|
|
|
std::shared_ptr<int> deleter(new int(), [](int* p) {
|
|
if (clientA) {
|
|
clientA->disconnect_for_test();
|
|
}
|
|
if (clientB) {
|
|
clientB->disconnect_for_test();
|
|
}
|
|
if (server) {
|
|
server->stop();
|
|
}
|
|
server_context.stop();
|
|
if (server_th.joinable()) {
|
|
server_th.join();
|
|
}
|
|
delete p;
|
|
});
|
|
|
|
if (!test_ls()) {
|
|
return false;
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::seconds(10));
|
|
return true;
|
|
}
|
|
|
|
// 测试 Ls
|
|
bool test_ls()
|
|
{
|
|
std::string cmd = std::to_string(ida_in_b) + " .";
|
|
if (!clientB->cmd_ls(cmd)) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
TEST_CASE("transm cmd part", "[cmd]")
|
|
{
|
|
SECTION("correctness of cmd")
|
|
{
|
|
REQUIRE(main_test() == true);
|
|
}
|
|
} |