llmessageconfig_tut.cpp

Go to the documentation of this file.

#include "linden_common.h"
#include "llmessageconfig.h"
#include "lltut.h"
#include "llsdserialize.h"
#include "llfile.h"
#include "lldir.h"
#include "lltimer.h"
#include "llframetimer.h"
#include "llsdutil.h"

namespace tut
{
        struct LLMessageConfigTestData {
                std::string mTestConfigDir;

                LLMessageConfigTestData()
                {
                        LLUUID random;
                        random.generate();
                        // generate temp dir
                        std::ostringstream oStr;
                        oStr << "/tmp/llmessage-config-test-" << random;
                        mTestConfigDir = oStr.str();
                        LLFile::mkdir(mTestConfigDir.c_str());
                        writeConfigFile(LLSD());
                        LLMessageConfig::initClass("simulator", mTestConfigDir);
                }

                ~LLMessageConfigTestData()
                {
                        // rm contents of temp dir
                        int rmfile = LLFile::remove((mTestConfigDir + "/message.xml").c_str());
                        ensure_equals("rmfile value", rmfile, 0);
                        // rm temp dir
                        int rmdir = LLFile::rmdir(mTestConfigDir.c_str());
                        ensure_equals("rmdir value", rmdir, 0);
                }

                void writeConfigFile(const LLSD& config)
                {
                        llofstream file((mTestConfigDir + "/message.xml").c_str());
                        if (file.is_open())
                        {
                                LLSDSerialize::toPrettyXML(config, file);
                        }
                        file.close();
                }
        };
        
        typedef test_group<LLMessageConfigTestData> LLMessageConfigTestGroup;
        typedef LLMessageConfigTestGroup::object LLMessageConfigTestObject;
        LLMessageConfigTestGroup llMessageConfigTestGroup("LLMessageConfig");
        
        template<> template<>
        void LLMessageConfigTestObject::test<1>()
                // tests server defaults
        {
                LLSD config;
                config["serverDefaults"]["simulator"] = "template";
                LLMessageConfig::useConfig(config);
                ensure_equals("Ensure server default is not template",
                                          LLMessageConfig::getServerDefaultFlavor(),
                                          LLMessageConfig::TEMPLATE_FLAVOR);
        }

        template<> template<>
        void LLMessageConfigTestObject::test<2>()
                // tests message flavors
        {
                LLSD config;
                config["serverDefaults"]["simulator"] = "template";
                config["messages"]["msg1"]["flavor"] = "template";
                config["messages"]["msg2"]["flavor"] = "llsd";
                LLMessageConfig::useConfig(config);
                ensure_equals("Ensure msg template flavor",
                                          LLMessageConfig::getMessageFlavor("msg1"),
                                          LLMessageConfig::TEMPLATE_FLAVOR);
                ensure_equals("Ensure msg llsd flavor",
                                          LLMessageConfig::getMessageFlavor("msg2"),
                                          LLMessageConfig::LLSD_FLAVOR);
        }

        template<> template<>
        void LLMessageConfigTestObject::test<4>()
                // tests message flavor defaults
        {
                LLSD config;
                config["serverDefaults"]["simulator"] = "llsd";
                config["messages"]["msg1"]["trusted-sender"] = true;
                LLMessageConfig::useConfig(config);
                ensure_equals("Ensure missing message gives no flavor",
                                          LLMessageConfig::getMessageFlavor("Test"),
                                          LLMessageConfig::NO_FLAVOR);
                ensure_equals("Ensure missing flavor is NO_FLAVOR even with sender trustedness set",
                                          LLMessageConfig::getMessageFlavor("msg1"),
                                          LLMessageConfig::NO_FLAVOR);
                ensure_equals("Ensure server default is llsd",
                                          LLMessageConfig::getServerDefaultFlavor(),
                                          LLMessageConfig::LLSD_FLAVOR);
        }

        template<> template<>
        void LLMessageConfigTestObject::test<3>()
                // tests trusted/untrusted senders
        {
                LLSD config;
                config["serverDefaults"]["simulator"] = "template";
                config["messages"]["msg1"]["flavor"] = "llsd";
                config["messages"]["msg1"]["trusted-sender"] = false;
                config["messages"]["msg2"]["flavor"] = "llsd";
                config["messages"]["msg2"]["trusted-sender"] = true;
                LLMessageConfig::useConfig(config);
                ensure_equals("Ensure untrusted is untrusted",
                                          LLMessageConfig::getSenderTrustedness("msg1"),
                                          LLMessageConfig::UNTRUSTED);
                ensure_equals("Ensure trusted is trusted",
                                          LLMessageConfig::getSenderTrustedness("msg2"),
                                          LLMessageConfig::TRUSTED);
                ensure_equals("Ensure missing trustedness is NOT_SET",
                                          LLMessageConfig::getSenderTrustedness("msg3"),
                                          LLMessageConfig::NOT_SET);
        }
        
        template<> template<>
        void LLMessageConfigTestObject::test<5>()
                // tests trusted/untrusted without flag, only flavor
        {
                LLSD config;
                config["serverDefaults"]["simulator"] = "template";
                config["messages"]["msg1"]["flavor"] = "llsd";
                LLMessageConfig::useConfig(config);
                ensure_equals("Ensure msg1 exists, has llsd flavor",
                                          LLMessageConfig::getMessageFlavor("msg1"),
                                          LLMessageConfig::LLSD_FLAVOR);
                ensure_equals("Ensure missing trusted is not set",
                                          LLMessageConfig::getSenderTrustedness("msg1"),
                                          LLMessageConfig::NOT_SET);
        }

        template<> template<>
        void LLMessageConfigTestObject::test<6>()
        {
                LLSD config;
                config["capBans"]["MapLayer"] = true;
                config["capBans"]["MapLayerGod"] = false;
                LLMessageConfig::useConfig(config);
                ensure_equals("Ensure cap ban true MapLayer",
                                          LLMessageConfig::isCapBanned("MapLayer"),
                                          true);
                ensure_equals("Ensure cap ban false",
                                          LLMessageConfig::isCapBanned("MapLayerGod"),
                                          false);
        }

        template<> template<>
        void LLMessageConfigTestObject::test<7>()
                // tests that config changes are picked up/refreshed periodically
        {
                LLSD config;
                config["serverDefaults"]["simulator"] = "llsd";
                writeConfigFile(config);

                // wait for it to reload after N seconds
                ms_sleep(6*1000);
                LLFrameTimer::updateFrameTime();
                ensure_equals("Ensure reload after 6 seconds",
                                          LLMessageConfig::getServerDefaultFlavor(),
                                          LLMessageConfig::LLSD_FLAVOR);
        }
}

---