//------------------------------------------------------------------------------
//      TeamForTester.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 "TeamForTester.h"

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

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

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

/*
        team_id TeamFor(const char *signature) const
        @case 1                 signature is NULL
        @results                Should return B_BAD_VALUE.
*/
void TeamForTester::TeamForTestA1()
{
// R5: crashes when passing a NULL signature
#ifndef TEST_R5
        BRoster roster;
        CHK(roster.TeamFor((const char*)NULL) == B_BAD_VALUE);
#endif
}

/*
        team_id TeamFor(const char *signature) const
        @case 2                 signature is not NULL, but no app with this signature is
                                        running
        @results                Should return B_ERROR.
*/
void TeamForTester::TeamForTestA2()
{
        BRoster roster;
        CHK(roster.TeamFor("application/x-vnd.obos-app-run-testapp1") == B_ERROR);
}

/*
        team_id TeamFor(const char *signature) const
        @case 3                 signature is not NULL and an (two) app(s) with this
                                        signature is (are) running; quit one; quit the second one
        @results                Should return the ID of one of the teams;
                                        the ID of the second team; B_ERROR.
*/
void TeamForTester::TeamForTestA3()
{
        // 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;
        team_id team = roster.TeamFor("application/x-vnd.obos-app-run-testapp1");
        CHK(team == runner1.Team() || team == runner2.Team());
        // quit app 1
        runner1.WaitFor(true);
        CHK(roster.TeamFor("application/x-vnd.obos-app-run-testapp1")
                == runner2.Team());
        // quit app 2
        runner2.WaitFor(true);
        CHK(roster.TeamFor("application/x-vnd.obos-app-run-testapp1") == B_ERROR);
}

/*
        team_id TeamFor(entry_ref *ref) const
        @case 1                 ref is NULL
        @results                Should return B_BAD_VALUE.
*/
void TeamForTester::TeamForTestB1()
{
// R5: crashes when passing a NULL ref
#ifndef TEST_R5
        BRoster roster;
        CHK(roster.TeamFor((entry_ref*)NULL) == B_BAD_VALUE);
#endif
}

/*
        team_id TeamFor(entry_ref *ref) const
        @case 2                 ref is not NULL, but no app with this ref is running
        @results                Should return B_ERROR.
*/
void TeamForTester::TeamForTestB2()
{
        BRoster roster;
        entry_ref ref;
        CHK(find_test_app("AppRunTestApp1", &ref) == B_OK);
        CHK(roster.TeamFor(&ref) == B_ERROR);
}

/*
        team_id TeamFor(entry_ref *ref) const
        @case 3                 ref is not NULL and an (two) app(s) with this ref is (are)
                                        running; quit one; quit the second one
        @results                Should return the ID of one of the teams;
                                        the ID of the second team; B_ERROR.
*/
void TeamForTester::TeamForTestB3()
{
        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;
        team_id team = roster.TeamFor(&ref);
        CHK(team == runner1.Team() || team == runner2.Team());
        // quit app 1
        runner1.WaitFor(true);
        CHK(roster.TeamFor(&ref) == runner2.Team());
        // quit app 2
        runner2.WaitFor(true);
        CHK(roster.TeamFor(&ref) == B_ERROR);
}


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

        ADD_TEST4(BRoster, SuiteOfTests, TeamForTester, TeamForTestA1);
        ADD_TEST4(BRoster, SuiteOfTests, TeamForTester, TeamForTestA2);
        ADD_TEST4(BRoster, SuiteOfTests, TeamForTester, TeamForTestA3);

        ADD_TEST4(BRoster, SuiteOfTests, TeamForTester, TeamForTestB1);
        ADD_TEST4(BRoster, SuiteOfTests, TeamForTester, TeamForTestB2);
        ADD_TEST4(BRoster, SuiteOfTests, TeamForTester, TeamForTestB3);

        return SuiteOfTests;
}