Function TEST_F

test/test_python_plugin_manager_impl.cpp:153–271

}

TEST_F(TestPythonPluginManagerImpl, PythonPluginManagerImpl) {
    using namespace fma_common;
    using namespace lgraph;
    std::string plugin_dir = "./python_plugin";
    std::string db_dir = "./testdb";
    int n_processes = 3;
    int max_idle_seconds = 2;
    int max_plugin_lifetime_seconds = 10 * max_idle_seconds;
    int argc = _ut_argc;
    char** argv = _ut_argv;
    Configuration config;
    config.Add(db_dir, "db", true).Comment("DB dir");
    config.Add(plugin_dir, "plugin", true).Comment("Plugin dir");
    config.Add(n_processes, "n,processes", true).Comment("Number of processes");
    config.Add(max_idle_seconds, "i,idle", true).Comment("Max idle seconds for python process");
    config.Add(max_plugin_lifetime_seconds, "lifetime", true)
        .Comment("Max life time for python process to be reused");
    config.ParseAndFinalize(argc, argv);
    fma_common::file_system::RemoveDir(db_dir);

    Galaxy galaxy(db_dir);
    std::string user = "admin";
    AccessControlledDB db = galaxy.OpenGraph(user, "default");

    {
        UT_WARN() << "Testing load plugin";
        Tester manager(db_dir, (size_t)1 << 30, plugin_dir, max_idle_seconds,
                       max_plugin_lifetime_seconds);
        fma_common::FileSystem::GetFileSystem("./").Mkdir(plugin_dir);
        WriteEchoPlugin(plugin_dir + "/echo.so");
        auto* pinfo = manager.CreatePluginInfo();
        pinfo->read_only = true;
        manager.LoadPlugin(user, "echo", pinfo);
        // current implementation allows repeated load
        manager.LoadPlugin(user, "echo", pinfo);

        UT_LOG() << "Testing call plugin";
        std::string output;

        manager.DoCall(nullptr, user, &db, "echo", pinfo, "hello", 0, true, output);
        UT_EXPECT_EQ(output, "hello");

        // currently delete has no effect, just return success
        UT_LOG() << "Testing del plugin";
        manager.UnloadPlugin(user, "echo", pinfo);

        // since delete has no effect, this should work
        UT_LOG() << "Testing call plugin";
        manager.DoCall(nullptr, user, &db, "echo", pinfo, "hello", 0, true, output);
        UT_EXPECT_EQ(output, "hello");

        // test timeout
        UT_LOG() << "Testing timeout with sleep";
        delete pinfo;
        pinfo = manager.CreatePluginInfo();
        WriteSleepPlugin(plugin_dir + "/sleep.so");
        manager.LoadPlugin(user, "sleep", pinfo);
        UT_LOG() << "Calling sleep for 0.1";