RetroZilla/extensions/python/xpcom/test/test_misc.py
2015-10-20 23:03:22 -04:00

233 lines
9.0 KiB
Python

# ***** BEGIN LICENSE BLOCK *****
# Version: MPL 1.1/GPL 2.0/LGPL 2.1
#
# The contents of this file are subject to the Mozilla 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/MPL/
#
# 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 the Python XPCOM language bindings.
#
# The Initial Developer of the Original Code is
# Activestate Tool Corp.
# Portions created by the Initial Developer are Copyright (C) 2000
# the Initial Developer. All Rights Reserved.
#
# Contributor(s):
# Mark Hammond <MarkH@ActiveState.com>
#
# Alternatively, the contents of this file may be used under the terms of
# either the GNU General Public License Version 2 or later (the "GPL"), or
# the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
# in which case the provisions of the GPL or the LGPL are applicable instead
# of those above. If you wish to allow use of your version of this file only
# under the terms of either the GPL or the LGPL, and not to allow others to
# use your version of this file under the terms of the MPL, indicate your
# decision by deleting the provisions above and replace them with the notice
# and other provisions required by the GPL or the LGPL. If you do not delete
# the provisions above, a recipient may use your version of this file under
# the terms of any one of the MPL, the GPL or the LGPL.
#
# ***** END LICENSE BLOCK *****
import xpcom
import xpcom.client
import xpcom.server
import xpcom._xpcom
import xpcom.components
import string
import unittest
import traceback, getopt, sys
verbose_level = 0
reportedSampleMissing = 0
def get_sample_component_cpp():
global reportedSampleMissing
contractid = "@mozilla.org/sample;1" # The C++ version.
try:
return xpcom.components.classes[contractid].createInstance()
except xpcom.COMException:
if not reportedSampleMissing:
print "***"
print "*** This test requires an XPCOM sample component,"
print "*** which does not exist. To build this test, you"
print "*** should change to the 'mozilla/xpcom/sample' directory,"
print "*** and run 'make', then run this test again."
print "***"
reportedSampleMissing = 1
else:
print "(skipping - no C++ sample...) ",
return None
def get_sample_component_js():
# This should *always* exist - no special make process.
contractid = "@mozilla.org/jssample;1" # the JS version
return xpcom.components.classes[contractid].createInstance()
class TestDumpInterfaces(unittest.TestCase):
def testAllInterfaces(self):
"Dump every interface under the sun!"
import xpcom, xpcom.xpt, xpcom._xpcom
iim = xpcom._xpcom.XPTI_GetInterfaceInfoManager()
if verbose_level:
print "Dumping every interface I can find"
enum = iim.EnumerateInterfaces()
rc = enum.First()
num = 0
while rc==0:
item = enum.CurrentItem(xpcom._xpcom.IID_nsIInterfaceInfo)
try:
iid = item.GetIID()
except xpcom.COMException:
if verbose_level:
print "Can't dump", item
continue # Dont bother dumping this.
interface = xpcom.xpt.Interface(iid)
num = num + 1
text = interface.Describe()
if verbose_level:
print text
rc = enum.Next()
if num < 200:
print "Only found", num, "interfaces - this seems unusually low!"
class TestEnumContractIDs(unittest.TestCase):
def testContractIDs(self):
"""Enumerate all the ContractIDs registered"""
enum = xpcom.components.registrar.enumerateContractIDs()
n = 0
while enum.hasMoreElements():
item = enum.getNext(xpcom.components.interfaces.nsISupportsCString)
n = n + 1
if verbose_level:
print "ContractID:", item.data
if n < 200:
print "Only found", n, "ContractIDs - this seems unusually low!"
class TestSampleComponent(unittest.TestCase):
def _doTestSampleComponent(self, test_flat = 0):
"""Test the standard Netscape 'sample' sample"""
c = get_sample_component_cpp()
if c is None:
return
if not test_flat:
c = c.queryInterface(xpcom.components.interfaces.nsISample)
self.failUnlessEqual(c.value, "initial value")
c.value = "new value"
self.failUnlessEqual(c.value, "new value")
c.poke("poked value")
self.failUnlessEqual(c.value, "poked value")
c.writeValue("Python just poked:")
def testSampleComponentFlat(self):
"""Test the standard Netscape 'sample' sample using interface flattening"""
self._doTestSampleComponent(1)
def testSampleComponentOld(self):
"""Test the standard Netscape 'sample' sample using explicit QI"""
self._doTestSampleComponent(0)
def _doTestHash(self, c):
"Test that hashing COM objects works"
d = {}
d[c] = None
if not d.has_key(c):
raise RuntimeError, "Can't get the exact same object back!"
if not d.has_key(c.queryInterface(xpcom.components.interfaces.nsISupports)):
raise RuntimeError, "Can't get back as nsISupports"
# And the same in reverse - stick an nsISupports in, and make sure an explicit interface comes back.
d = {}
# contractid = "@mozilla.org/sample;1" # The C++ version.
# c = xpcom.components.classes[contractid].createInstance() \
# .queryInterface(xpcom.components.interfaces.nsISupports)
d[c] = None
if not d.has_key(c):
raise RuntimeError, "Can't get the exact same object back!"
if not d.has_key(c.queryInterface(xpcom.components.interfaces.nsISample)):
raise RuntimeError, "Can't get back as nsISupports"
def testHashJS(self):
c = get_sample_component_js()
self._doTestHash(c)
def testHashCPP(self):
c = get_sample_component_cpp()
if c is not None:
self._doTestHash(c)
class TestIIDs(unittest.TestCase):
def TestIIDs(self):
"Do some basic IID semantic tests."
iid_str = "{7ee4bdc6-cb53-42c1-a9e4-616b8e012aba}"
IID = xpcom._xpcom.IID
self.failUnlessEqual(IID(iid_str), IID(iid_str))
self.failUnlessEqual(hash(IID(iid_str)), hash(IID(iid_str)))
self.failUnlessEqual(IID(iid_str), IID(iid_str.upper()))
self.failUnlessEqual(hash(IID(iid_str)), hash(IID(iid_str.upper())))
# If the above work, this shoud too, but WTF
dict = {}
dict[IID(iid_str)] = None
self.failUnless(dict.has_key(IID(iid_str)), "hashes failed in dictionary")
self.failUnless(dict.has_key(IID(iid_str.upper())), "uppercase hash failed in dictionary")
class TestRepr(unittest.TestCase):
def _doTestRepr(self, progid, interfaces):
if isinstance(progid, str):
ob = xpcom.components.classes[progid].createInstance()
else:
ob = progid
self.failUnless(repr(ob).find(str(progid)) >= 0, repr(ob))
for interface_name in interfaces.split():
self.failUnless(repr(ob).find(interface_name) >= 0, repr(ob))
def testReprPython(self):
"Test repr() of Python objects"
self._doTestRepr("Python.TestComponent", "nsIPythonTestInterfaceDOMStrings nsIPythonTestInterfaceExtra nsIPythonTestInterface")
# JS does not provide class-info :(
#def testReprJS(self):
# self._doTestRepr("@mozilla.org/jssample;1", "nsISample")
def testReprSample(self):
"Test repr() of non-Python objects"
ob = get_sample_component_cpp()
if ob is None:
return
self._doTestRepr(ob, "nsISample")
class TestUnwrap(unittest.TestCase):
"Test the unwrap facilities"
def testUnwrap(self):
# First test that a Python object can be unwrapped.
ob = xpcom.components.classes["Python.TestComponent"].createInstance()
pyob = xpcom.server.UnwrapObject(ob)
# This depends on our __repr__ implementation, but that's OK - it
# can be updated should our __repr__ change :)
self.failUnless(str(pyob).startswith("<component:py_test_component.PythonTestComponent"))
# Test that a non-Python implemented object can NOT be unwrapped.
ob = get_sample_component_cpp()
if ob is None:
return
self.failUnlessRaises(ValueError, xpcom.server.UnwrapObject, ob)
if __name__=='__main__':
unittest.main()
xpcom._xpcom.NS_ShutdownXPCOM()
ni = xpcom._xpcom._GetInterfaceCount()
ng = xpcom._xpcom._GetGatewayCount()
if ni or ng:
print "********* WARNING - Leaving with %d/%d objects alive" % (ni,ng)