Skip to content

Commit

Permalink
feat: add more test cases of failing to start graph (#529)
Browse files Browse the repository at this point in the history
  • Loading branch information
halajohn authored Jan 9, 2025
1 parent d8c4ffa commit e43cd9f
Show file tree
Hide file tree
Showing 21 changed files with 298 additions and 24 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ static void on_addon_create_extension_done(ten_env_t *ten_env,
} else {
// Failed to create the specified extension.

TEN_LOGI(
TEN_LOGE(
"Failed to create extension %s",
ten_string_get_raw_str(&create_extension_done_ctx->extension_name));

Expand Down
3 changes: 3 additions & 0 deletions core/src/ten_runtime/extension_thread/extension_thread.c
Original file line number Diff line number Diff line change
Expand Up @@ -507,6 +507,9 @@ static void ten_engine_on_all_extension_threads_are_ready(

ten_shared_ptr_t *cmd_result = NULL;
if (error_occurred) {
TEN_LOGE("[%s] Failed to start the graph successfully, shutting it down.",
ten_engine_get_id(self, true));

cmd_result = ten_cmd_result_create_from_cmd(TEN_STATUS_CODE_ERROR,
state_requester_cmd);
} else {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(predefined_graph_basic_1__predefined_graph,

} // namespace

TEST(ExtensionTest, PredefinedGraphBasic1) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphBasic1) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -177,7 +177,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(predefined_graph_basic_2__normal_extension,

} // namespace

TEST(ExtensionTest, PredefinedGraphBasic2) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphBasic2) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, PredefinedGraphEngineOwnEventloop) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphEngineOwnEventloop) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,102 @@
//
// Copyright © 2025 Agora
// This file is part of TEN Framework, an open source project.
// Licensed under the Apache License, Version 2.0, with certain conditions.
// Refer to the "LICENSE" file in the root directory for more information.
//
#include "gtest/gtest.h"
#include "include_internal/ten_runtime/binding/cpp/ten.h"
#include "tests/common/client/cpp/msgpack_tcp.h"
#include "tests/ten_runtime/smoke/util/binding/cpp/check.h"

namespace {

class test_predefined_graph : public ten::extension_t {
public:
explicit test_predefined_graph(const char *name) : ten::extension_t(name) {}

void on_cmd(ten::ten_env_t &ten_env,
std::unique_ptr<ten::cmd_t> cmd) override {
nlohmann::json const detail = {{"id", 1}, {"name", "a"}};

auto cmd_result = ten::cmd_result_t::create(TEN_STATUS_CODE_OK);
cmd_result->set_property_from_json("detail", detail.dump().c_str());
ten_env.return_result(std::move(cmd_result), std::move(cmd));
}
};

class test_app : public ten::app_t {
public:
void on_configure(ten::ten_env_t &ten_env) override {
bool rc = ten::ten_env_internal_accessor_t::init_manifest_from_json(
ten_env,
// clang-format off
R"({
"type": "app",
"name": "test_app",
"version": "0.1.0"
})"
// clang-format on
);
ASSERT_EQ(rc, true);

rc = ten_env.init_property_from_json(
// clang-format off
R"###({
"_ten": {
"uri": "msgpack://127.0.0.1:8001/",
"log_level": 2,
"predefined_graphs": [{
"name": "default",
"auto_start": true,
"singleton": true,
"nodes": [{
"type": "extension",
"name": "predefined_graph",
"addon": "predefined_graph_incorrect_1__predefined_graph",
"extension_group": "predefined_graph_group"
}]
}]
}
})###"
// clang-format on
);
ASSERT_EQ(rc, true);

ten_env.on_configure_done();
}
};

void *app_thread_main(TEN_UNUSED void *args) {
auto *app = new test_app();
app->run();
delete app;

return nullptr;
}

TEN_CPP_REGISTER_ADDON_AS_EXTENSION(
predefined_graph_incorrect_1__predefined_graph, test_predefined_graph);

} // namespace

TEST(PredefinedGraphTest, PredefinedGraphIncorrect1) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
auto *client = new ten::msgpack_tcp_client_t("msgpack://127.0.0.1:8001/");

// Do not need to send 'start_graph' command first.
// The 'graph_id' MUST be "default" if we want to send the request to
// predefined graph.
auto test_cmd = ten::cmd_t::create("test");
test_cmd->set_dest("msgpack://127.0.0.1:8001/", "default",
"predefined_graph_group", "predefined_graph");
auto cmd_result = client->send_cmd_and_recv_result(std::move(test_cmd));
ten_test::check_status_code(cmd_result, TEN_STATUS_CODE_OK);
ten_test::check_detail_with_json(cmd_result, R"({"id": 1, "name": "a"})");

delete client;

ten_thread_join(app_thread, -1);
}
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(predefined_graph_multi_app__extension_2,

} // namespace

TEST(ExtensionTest, PredefinedGraphMultiApp) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphMultiApp) { // NOLINT
// Start app.
auto *app_1_thread =
ten_thread_create("app thread 1", app_thread_1_main, nullptr);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, PredefinedGraphMultiExtension1) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphMultiExtension1) { // NOLINT
// Start app.
auto *app_thread =
ten_thread_create("app thread", test_app_thread_main, nullptr);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, PredefinedGraphMultiExtension2) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphMultiExtension2) { // NOLINT
// Start app.
auto *app_thread =
ten_thread_create("app thread", test_app_thread_main, nullptr);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, PredefinedGraphMultiExtension3) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphMultiExtension3) { // NOLINT
// Start app.
auto *app_thread =
ten_thread_create("app thread", test_app_thread_main, nullptr);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, PredefinedGraphMultiExtension4) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphMultiExtension4) { // NOLINT
// Start app.
auto *app_thread =
ten_thread_create("app thread", test_app_thread_main, nullptr);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, PredefinedGraphNoAutoStart) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphNoAutoStart) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(prebuild_two_extensions_1,

} // namespace

TEST(ExtensionTest, PredefinedGraphTwoStandaloneExtensions1) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphTwoStandaloneExtensions1) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(prebuild_two_extensions_2,

} // namespace

TEST(ExtensionTest, PredefinedGraphTwoStandaloneExtensions2) { // NOLINT
TEST(PredefinedGraphTest, PredefinedGraphTwoStandaloneExtensions2) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, StartGraphAndCommunication) { // NOLINT
TEST(StartGraphTest, StartGraphAndCommunication) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ TEN_CPP_REGISTER_ADDON_AS_EXTENSION(

} // namespace

TEST(ExtensionTest, StartGraphFromExtension) { // NOLINT
TEST(StartGraphTest, StartGraphFromExtension) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ class test_predefined_graph : public ten::extension_t {
"name": "normal_extension",
"addon": "not_existed_extension_addon",
"app": "msgpack://127.0.0.1:8001/",
"extension_group": "start_incorrect_graph__normal_extension_group"
"extension_group": "start_incorrect_graph_1__normal_extension_group"
}]
})"_json.dump()
.c_str());
Expand Down Expand Up @@ -111,8 +111,8 @@ class test_app : public ten::app_t {
"nodes": [{
"type": "extension",
"name": "predefined_graph",
"addon": "start_incorrect_graph__predefined_graph_extension",
"extension_group": "start_incorrect_graph__predefined_graph_group"
"addon": "start_incorrect_graph_1__predefined_graph_extension",
"extension_group": "start_incorrect_graph_1__predefined_graph_group"
}]
}]
}
Expand All @@ -134,13 +134,13 @@ void *app_thread_main(TEN_UNUSED void *args) {
}

TEN_CPP_REGISTER_ADDON_AS_EXTENSION(
start_incorrect_graph__predefined_graph_extension, test_predefined_graph);
TEN_CPP_REGISTER_ADDON_AS_EXTENSION(start_incorrect_graph__normal_extension,
start_incorrect_graph_1__predefined_graph_extension, test_predefined_graph);
TEN_CPP_REGISTER_ADDON_AS_EXTENSION(start_incorrect_graph_1__normal_extension,
test_normal_extension);

} // namespace

TEST(ExtensionTest, StartIncorrectGraph) { // NOLINT
TEST(StartGraphTest, StartIncorrectGraph1) { // NOLINT
auto *app_thread = ten_thread_create("app thread", app_thread_main, nullptr);

// Create a client and connect to the app.
Expand All @@ -151,7 +151,7 @@ TEST(ExtensionTest, StartIncorrectGraph) { // NOLINT
// request to predefined graph.
auto test_cmd = ten::cmd_t::create("test");
test_cmd->set_dest("msgpack://127.0.0.1:8001/", "default",
"start_incorrect_graph__predefined_graph_group",
"start_incorrect_graph_1__predefined_graph_group",
"predefined_graph");
auto cmd_result = client->send_cmd_and_recv_result(std::move(test_cmd));
ten_test::check_status_code(cmd_result, TEN_STATUS_CODE_OK);
Expand Down
Loading

0 comments on commit e43cd9f

Please sign in to comment.