providerstatus.cpp

Pegasus is an open-source implementationof the DMTF CIM and WBEM standards. It is designed to be por

//%2006////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2000, 2001, 2002 BMC Software; Hewlett-Packard Development
// Company, L.P.; IBM Corp.; The Open Group; Tivoli Systems.
// Copyright (c) 2003 BMC Software; Hewlett-Packard Development Company, L.P.;
// IBM Corp.; EMC Corporation, The Open Group.
// Copyright (c) 2004 BMC Software; Hewlett-Packard Development Company, L.P.;
// IBM Corp.; EMC Corporation; VERITAS Software Corporation; The Open Group.
// Copyright (c) 2005 Hewlett-Packard Development Company, L.P.; IBM Corp.;
// EMC Corporation; VERITAS Software Corporation; The Open Group.
// Copyright (c) 2006 Hewlett-Packard Development Company, L.P.; IBM Corp.;
// EMC Corporation; Symantec Corporation; The Open Group.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// THE ABOVE COPYRIGHT NOTICE AND THIS PERMISSION NOTICE SHALL BE INCLUDED IN
// ALL COPIES OR SUBSTANTIAL PORTIONS OF THE SOFTWARE. THE SOFTWARE IS PROVIDED
// "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
// LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
// ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
//==============================================================================
//
//%/////////////////////////////////////////////////////////////////////////////

#include <Pegasus/Common/Config.h>
#include <Pegasus/ProviderManager2/Default/ProviderStatus.h>
#include <Pegasus/Common/Time.h>
#include <Pegasus/Common/PegasusAssert.h>

#include <iostream>

PEGASUS_USING_PEGASUS;
PEGASUS_USING_STD;

const char* verbose = 0;

void testInitialization()
{
    ProviderStatus status;

    PEGASUS_TEST_ASSERT(!status.isInitialized());

    AutoMutex lock1(status.getStatusMutex());

    // Test that the status mutex allows recursive locks
    AutoMutex lock2(status.getStatusMutex());

    status.setInitialized(true);
    PEGASUS_TEST_ASSERT(status.isInitialized());

    status.setModule((ProviderModule*) 5);
    PEGASUS_TEST_ASSERT(status.getModule() == (ProviderModule*) 5);

    CIMInstance providerInstance("Test_Class");
    status.setProviderInstance(providerInstance);
    PEGASUS_TEST_ASSERT(status.getProviderInstance().getClassName() ==
        "Test_Class");

    status.setInitialized(false);
    PEGASUS_TEST_ASSERT(!status.isInitialized());
    PEGASUS_TEST_ASSERT(status.getModule() == 0);
}

void testIndications()
{
    ProviderStatus status;
    status.setInitialized(true);

    PEGASUS_TEST_ASSERT(!status.getIndicationsEnabled());
    PEGASUS_TEST_ASSERT(status.isIdle());

    status.setIndicationsEnabled(true);
    PEGASUS_TEST_ASSERT(status.getIndicationsEnabled());
    PEGASUS_TEST_ASSERT(!status.isIdle());

    PEGASUS_TEST_ASSERT(!status.testSubscriptions());
    PEGASUS_TEST_ASSERT(status.testIfZeroAndIncrementSubscriptions());
    PEGASUS_TEST_ASSERT(!status.testIfZeroAndIncrementSubscriptions());
    PEGASUS_TEST_ASSERT(status.testSubscriptions());
    PEGASUS_TEST_ASSERT(!status.decrementSubscriptionsAndTestIfZero());
    PEGASUS_TEST_ASSERT(status.decrementSubscriptionsAndTestIfZero());
    PEGASUS_TEST_ASSERT(status.testIfZeroAndIncrementSubscriptions());
    status.resetSubscriptions();
    PEGASUS_TEST_ASSERT(status.getIndicationsEnabled());
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(!status.testSubscriptions());
    PEGASUS_TEST_ASSERT(status.testIfZeroAndIncrementSubscriptions());

    status.setInitialized(false);
    PEGASUS_TEST_ASSERT(!status.getIndicationsEnabled());
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(!status.testSubscriptions());
}

void testOperations()
{
    ProviderStatus status;

    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 0);
    status.setInitialized(true);

    PEGASUS_TEST_ASSERT(status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 0);
    status.operationBegin();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 1);
    status.operationBegin();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 2);
    status.operationEnd();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 1);
    status.operationEnd();
    PEGASUS_TEST_ASSERT(status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 0);

    struct timeval lastOperationTime;
    status.getLastOperationEndTime(&lastOperationTime);

    // Test that the operation time is within one second of now
    struct timeval now;
    Time::gettimeofday(&now);
    PEGASUS_TEST_ASSERT(now.tv_sec >= lastOperationTime.tv_sec);
    PEGASUS_TEST_ASSERT(now.tv_sec <= lastOperationTime.tv_sec + 1);

    status.operationBegin();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 1);

    status.setInitialized(false);
    PEGASUS_TEST_ASSERT(!status.isIdle());
    PEGASUS_TEST_ASSERT(status.numCurrentOperations() == 0);
}

void testCIMOMHandle()
{
    ProviderStatus status;
    status.setInitialized(true);

    status.setCIMOMHandle((CIMOMHandle*) 10);
    PEGASUS_TEST_ASSERT(status.getCIMOMHandle() == (CIMOMHandle*) 10);

    status.setInitialized(false);
    PEGASUS_TEST_ASSERT(status.getCIMOMHandle() == 0);

    status.setInitialized(true);
    CIMOMHandle cimomHandle;
    status.setCIMOMHandle(&cimomHandle);
    PEGASUS_TEST_ASSERT(status.isIdle());

    cimomHandle.disallowProviderUnload();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    cimomHandle.disallowProviderUnload();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    cimomHandle.allowProviderUnload();
    PEGASUS_TEST_ASSERT(!status.isIdle());
    cimomHandle.allowProviderUnload();
    PEGASUS_TEST_ASSERT(status.isIdle());

    // An extra allowProviderUnload() call should be ignored
    cimomHandle.allowProviderUnload();
    PEGASUS_TEST_ASSERT(status.isIdle());
}

int main(int argc, char** argv)
{
    verbose = getenv ("PEGASUS_TEST_VERBOSE");

    try
    {
        testInitialization();
        testIndications();
        testOperations();
        testCIMOMHandle();
    }
    catch (Exception& e)
    {
        cout << "Caught unexpected exception: " << e.getMessage() << endl;
        return 1;
    }
    catch (...)
    {
        cout << "Caught unexpected exception" << endl;
        return 1;
    }

    cout << argv[0] << " +++++ passed all tests" << endl;

    return 0;
}