//------------------------------------------------------------------------------
//      GetAppInfoTester.cpp
//
//------------------------------------------------------------------------------

// Standard Includes -----------------------------------------------------------
#include <stdio.h>

// System Includes -------------------------------------------------------------
#include <Message.h>
#include <OS.h>
#include <Handler.h>
#include <Looper.h>
#include <Roster.h>
#include <String.h>

// Project Includes ------------------------------------------------------------
#include <TestShell.h>
#include <TestUtils.h>
#include <cppunit/TestAssert.h>

// Local Includes --------------------------------------------------------------
#include "AppRunner.h"
#include "GetAppInfoTester.h"

// Local Defines ---------------------------------------------------------------

// Globals ---------------------------------------------------------------------

//------------------------------------------------------------------------------

// check_app_info
static
bool
check_app_info(app_info &info, AppRunner &runner, const char *signature,
                           uint32 flags)
{
        team_id team = runner.Team();
        // get the thread
        thread_id thread = -1;
        int32 cookie = 0;
        thread_info threadInfo;
        while (get_next_thread_info(team, &cookie, &threadInfo) == B_OK) {
                if (thread < 0 || threadInfo.thread < thread)
                        thread = threadInfo.thread;
        }
        // get port and ref
        port_id port = runner.AppLooperPort();
        entry_ref ref;
        runner.GetRef(&ref);
        // compare
//printf("check_app_info(): "
//"  thread:    %ld vs %ld\n"
//"  team:      %ld vs %ld\n"
//"  port:      %ld vs %ld\n"
//"  flags:     %lx  vs %lx\n"
//"  signature: `%s' vs `%s'\n", info.thread, thread, info.team, team,
//info.port, port, info.flags, flags, info.signature, signature);
//printf("  ref:       (%ld, %lld, `%s') vs (%ld, %lld, `%s')\n",
//info.ref.device, info.ref.directory, info.ref.name,
//ref.device, ref.directory, ref.name);
        return (info.thread == thread && info.team == team && info.port == port
                        && info.flags == flags && info.ref == ref
                        && !strncmp(info.signature, signature, B_MIME_TYPE_LENGTH));
}

/*
        status_t GetAppInfo(const char *signature, app_info *info) const
        @case 1                 signature is NULL or info is NULL
        @results                Should return B_BAD_VALUE.
*/
void GetAppInfoTester::GetAppInfoTestA1()
{
        BRoster roster;
        app_info info;
        CHK(roster.GetAppInfo((const char*)NULL, NULL) == B_BAD_VALUE);
        CHK(roster.GetAppInfo((const char*)NULL, &info) == B_BAD_VALUE);
// R5: crashes when passing a NULL app_info
#ifndef TEST_R5
        CHK(roster.GetAppInfo("application/x-vnd.obos-app-run-testapp1",
                                                  NULL) == B_BAD_VALUE);
#endif
}

/*
        status_t GetAppInfo(const char *signature, app_info *info) const
        @case 2                 signature/info are not NULL, but no app with this
                                        signature is running
        @results                Should return B_ERROR.
*/
void GetAppInfoTester::GetAppInfoTestA2()
{
        BRoster roster;
        app_info info;
        CHK(roster.GetAppInfo("application/x-vnd.obos-app-run-testapp1", &info)
                == B_ERROR);
}

/*
        status_t GetAppInfo(const char *signature, app_info *info) const
        @case 3                 signature/info are not NULL and an (two) app(s) with this
                                        signature is (are) running; quit one; quit the second one
        @results                Should
                                        - fill the app info with the data of one of the apps and
                                          return B_OK;
                                        - fill the app info with the data of the second apps and
                                          return B_OK;
                                        - return B_ERROR.
*/
void GetAppInfoTester::GetAppInfoTestA3()
{
        const char *signature = "application/x-vnd.obos-app-run-testapp1";
        uint32 flags = B_MULTIPLE_LAUNCH | B_ARGV_ONLY;
        // run the remote apps
        AppRunner runner1(true);
        AppRunner runner2(true);
        CHK(runner1.Run("AppRunTestApp1") == B_OK);
        CHK(runner2.Run("AppRunTestApp1") == B_OK);
        // create the BRoster and perform the tests
        BRoster roster;
        app_info info1;
        CHK(roster.GetAppInfo(signature, &info1) == B_OK);
        CHK(check_app_info(info1, runner1, signature, flags)
                || check_app_info(info1, runner2, signature, flags));
        // quit app 1
        runner1.WaitFor(true);
        app_info info2;
        CHK(roster.GetAppInfo(signature, &info2) == B_OK);
        CHK(check_app_info(info2, runner2, signature, flags));
        // quit app 2
        runner2.WaitFor(true);
        CHK(roster.GetAppInfo(signature, &info1) == B_ERROR);
}

/*
        status_t GetAppInfo(entry_ref *ref, app_info *info) const
        @case 1                 ref is NULL or info is NULL
        @results                Should return B_BAD_VALUE.
*/
void GetAppInfoTester::GetAppInfoTestB1()
{
        BRoster roster;
        entry_ref ref;
        CHK(find_test_app("AppRunTestApp1", &ref) == B_OK);
        app_info info;
        CHK(roster.GetAppInfo((entry_ref*)NULL, NULL) == B_BAD_VALUE);
        CHK(roster.GetAppInfo((entry_ref*)NULL, &info) == B_BAD_VALUE);
// R5: crashes when passing a NULL app_info
#ifndef TEST_R5
        CHK(roster.GetAppInfo(&ref, NULL) == B_BAD_VALUE);
#endif
}

/*
        status_t GetAppInfo(entry_ref *ref, app_info *info) const
        @case 2                 ref/info are not NULL, but no app with this ref is running
        @results                Should return B_ERROR.
*/
void GetAppInfoTester::GetAppInfoTestB2()
{
        BRoster roster;
        entry_ref ref;
        CHK(find_test_app("AppRunTestApp1", &ref) == B_OK);
        app_info info;
        CHK(roster.GetAppInfo(&ref, &info) == B_ERROR);
}

/*
        status_t GetAppInfo(entry_ref *ref, app_info *info) const
        @case 3                 ref/info are not NULL and an (two) app(s) with this ref
                                        is (are) running; quit one; quit the second one
        @results                Should
                                        - fill the app info with the data of one of the apps and
                                          return B_OK;
                                        - fill the app info with the data of the second apps and
                                          return B_OK;
                                        - return B_ERROR.
*/
void GetAppInfoTester::GetAppInfoTestB3()
{
        const char *signature = "application/x-vnd.obos-app-run-testapp1";
        uint32 flags = B_MULTIPLE_LAUNCH | B_ARGV_ONLY;
        entry_ref ref;
        CHK(find_test_app("AppRunTestApp1", &ref) == B_OK);
        // run the remote apps
        AppRunner runner1(true);
        AppRunner runner2(true);
        CHK(runner1.Run("AppRunTestApp1") == B_OK);
        CHK(runner2.Run("AppRunTestApp1") == B_OK);
        // create the BRoster and perform the tests
        BRoster roster;
        app_info info1;
        CHK(roster.GetAppInfo(&ref, &info1) == B_OK);
        CHK(check_app_info(info1, runner1, signature, flags)
                || check_app_info(info1, runner2, signature, flags));
        // quit app 1
        runner1.WaitFor(true);
        app_info info2;
        CHK(roster.GetAppInfo(&ref, &info2) == B_OK);
        CHK(check_app_info(info2, runner2, signature, flags));
        // quit app 2
        runner2.WaitFor(true);
        CHK(roster.GetAppInfo(&ref, &info1) == B_ERROR);
}

/*
        status_t GetRunningAppInfo(team_id team, app_info *info) const
        @case 1                 info is NULL
        @results                Should return B_BAD_VALUE.
*/
void GetAppInfoTester::GetRunningAppInfoTest1()
{
// R5: crashes when passing a NULL app_info
#ifndef TEST_R5
        BRoster roster;
        // invalid team ID
        CHK(roster.GetRunningAppInfo(-1, NULL) == B_BAD_VALUE);
        // valid team ID
        AppRunner runner(true);
        CHK(runner.Run("AppRunTestApp1") == B_OK);
        CHK(roster.GetRunningAppInfo(runner.Team(), NULL) == B_BAD_VALUE);
        runner.WaitFor(true);
#endif
}

/*
        status_t GetRunningAppInfo(team_id team, app_info *info) const
        @case 2                 info is not NULL, but no app with the team ID is running
        @results                Should return B_BAD_TEAM_ID.
*/
void GetAppInfoTester::GetRunningAppInfoTest2()
{
        BRoster roster;
        // invalid team ID
        app_info info;
#ifdef TEST_R5
        CHK(roster.GetRunningAppInfo(-1, &info) == B_ERROR);
#else
        CHK(roster.GetRunningAppInfo(-1, &info) == B_BAD_TEAM_ID);
#endif
        CHK(roster.GetRunningAppInfo(-2, &info) == B_BAD_TEAM_ID);
        // originally valid team ID -- app terminates before call
        AppRunner runner(true);
        CHK(runner.Run("AppRunTestApp1") == B_OK);
        team_id team = runner.Team();
        runner.WaitFor(true);
        CHK(roster.GetRunningAppInfo(team, &info) == B_BAD_TEAM_ID);
}

/*
        status_t GetRunningAppInfo(team_id team, app_info *info) const
        @case 3                 info is not NULL, and an app with the team ID is running
        @results                Should fill the app info and return B_OK.
*/
void GetAppInfoTester::GetRunningAppInfoTest3()
{
        const char *signature = "application/x-vnd.obos-app-run-testapp1";
        uint32 flags = B_MULTIPLE_LAUNCH | B_ARGV_ONLY;
        // run the app
        AppRunner runner(true);
        CHK(runner.Run("AppRunTestApp1") == B_OK);
        // get and check the info
        BRoster roster;
        app_info info;
        CHK(roster.GetRunningAppInfo(runner.Team(), &info) == B_OK);
        CHK(check_app_info(info, runner, signature, flags));
        // quit the app
        runner.WaitFor(true);
}


Test* GetAppInfoTester::Suite()
{
        TestSuite* SuiteOfTests = new TestSuite;

        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester, GetAppInfoTestA1);
        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester, GetAppInfoTestA2);
        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester, GetAppInfoTestA3);

        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester, GetAppInfoTestB1);
        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester, GetAppInfoTestB2);
        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester, GetAppInfoTestB3);

        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester,
                          GetRunningAppInfoTest1);
        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester,
                          GetRunningAppInfoTest2);
        ADD_TEST4(BRoster, SuiteOfTests, GetAppInfoTester,
                          GetRunningAppInfoTest3);

        return SuiteOfTests;
}