gecko-dev/xpcom/tests/TestObserverService.cpp

163 строки
5.9 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* The contents of this file are subject to the Netscape Public
* License Version 1.1 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.mozilla.org/NPL/
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* The Original Code is mozilla.org code.
*
* The Initial Developer of the Original Code is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
* Pierre Phaneuf <pp@ludusdesign.com>
*/
#define NS_IMPL_IDS
#include "nsISupports.h"
#include "nsRepository.h"
#include "nsIObserverService.h"
#include "nsIObserver.h"
#include "nsString.h"
#include "prprf.h"
#include <iostream.h>
#include "nsWeakReference.h"
static nsIObserverService *anObserverService = NULL;
static void testResult( nsresult rv ) {
if ( NS_SUCCEEDED( rv ) ) {
cout << "...ok" << endl;
} else {
cout << "...failed, rv=0x" << hex << (int)rv << endl;
}
return;
}
extern ostream &operator<<( ostream &s, nsString &str ) {
const char *cstr = str.ToNewCString();
s << cstr;
delete [] (char*)cstr;
return s;
}
class TestObserver : public nsIObserver, public nsSupportsWeakReference {
public:
TestObserver( const nsString &name = NS_ConvertASCIItoUCS2("unknown") )
: mName( name ) {
NS_INIT_REFCNT();
}
virtual ~TestObserver() {}
NS_DECL_ISUPPORTS
NS_DECL_NSIOBSERVER
nsString mName;
};
NS_IMPL_ISUPPORTS2( TestObserver, nsIObserver, nsISupportsWeakReference );
NS_IMETHODIMP
TestObserver::Observe( nsISupports *aSubject,
const PRUnichar *aTopic,
const PRUnichar *someData ) {
nsString topic( aTopic );
nsString data( someData );
/*
The annoying double-cast below is to work around an annoying bug in
the compiler currently used on wensleydale. This is a test.
*/
cout << mName << " has observed something: subject@" << (void*)aSubject
<< " name=" << NS_REINTERPRET_CAST(TestObserver*, NS_REINTERPRET_CAST(void*, aSubject))->mName
<< " aTopic=" << topic
<< " someData=" << data << endl;
return NS_OK;
}
int main(int argc, char *argv[])
{
nsString topicA; topicA.AssignWithConversion( "topic-A" );
nsString topicB; topicB.AssignWithConversion( "topic-B" );
nsresult rv;
nsresult res = nsComponentManager::CreateInstance(NS_OBSERVERSERVICE_PROGID,
NULL,
NS_GET_IID(nsIObserverService),
(void **) &anObserverService);
if (res == NS_OK) {
nsIObserver *anObserver;
nsIObserver *aObserver = new TestObserver(NS_ConvertASCIItoUCS2("Observer-A"));
aObserver->AddRef();
nsIObserver *bObserver = new TestObserver(NS_ConvertASCIItoUCS2("Observer-B"));
bObserver->AddRef();
cout << "Adding Observer-A as observer of topic-A..." << endl;
rv = anObserverService->AddObserver(aObserver, topicA.GetUnicode());
testResult(rv);
cout << "Adding Observer-B as observer of topic-A..." << endl;
rv = anObserverService->AddObserver(bObserver, topicA.GetUnicode());
testResult(rv);
cout << "Adding Observer-B as observer of topic-B..." << endl;
rv = anObserverService->AddObserver(bObserver, topicB.GetUnicode());
testResult(rv);
cout << "Testing Notify(observer-A, topic-A)..." << endl;
rv = anObserverService->Notify( aObserver,
topicA.GetUnicode(),
NS_ConvertASCIItoUCS2("Testing Notify(observer-A, topic-A)").GetUnicode() );
testResult(rv);
cout << "Testing Notify(observer-B, topic-B)..." << endl;
rv = anObserverService->Notify( bObserver,
topicB.GetUnicode(),
NS_ConvertASCIItoUCS2("Testing Notify(observer-B, topic-B)").GetUnicode() );
testResult(rv);
cout << "Testing EnumerateObserverList (for topic-A)..." << endl;
nsIEnumerator* e;
rv = anObserverService->EnumerateObserverList(topicA.GetUnicode(), &e);
testResult(rv);
cout << "Enumerating observers of topic-A..." << endl;
if ( NS_SUCCEEDED( rv ) ) {
nsISupports *inst;
for (e->First(); e->IsDone() != NS_OK; e->Next()) {
rv = e->CurrentItem(&inst);
if (NS_SUCCEEDED(rv)) {
rv = inst->QueryInterface(NS_GET_IID(nsIObserver),(void**)&anObserver);
cout << "Calling observe on enumerated observer "
<< NS_REINTERPRET_CAST(TestObserver*, NS_REINTERPRET_CAST(void*, inst))->mName << "..." << endl;
rv = anObserver->Observe( inst, topicA.GetUnicode(), NS_ConvertASCIItoUCS2("during enumeration").GetUnicode() );
testResult(rv);
}
}
}
cout << "...done enumerating observers of topic-A" << endl;
cout << "Removing Observer-A..." << endl;
rv = anObserverService->RemoveObserver(aObserver, topicA.GetUnicode());
testResult(rv);
cout << "Removing Observer-B (topic-A)..." << endl;
rv = anObserverService->RemoveObserver(bObserver, topicB.GetUnicode());
testResult(rv);
cout << "Removing Observer-B (topic-B)..." << endl;
rv = anObserverService->RemoveObserver(bObserver, topicA.GetUnicode());
testResult(rv);
}
return NS_OK;
}