mirror of
https://github.com/wxWidgets/Phoenix.git
synced 2025-07-21 12:41:10 +02:00
344 lines
11 KiB
Python
344 lines
11 KiB
Python
import unittest
|
|
from unittests import wtc
|
|
import wx
|
|
import os
|
|
import sys
|
|
|
|
pngFile = os.path.join(os.path.dirname(__file__), 'toucan.png')
|
|
|
|
#print os.getpid(); raw_input('Press enter...')
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
class DataObjTests(wtc.WidgetTestCase):
|
|
|
|
def test_DataFormat(self):
|
|
fmt1 = wx.DataFormat('my custom format')
|
|
fmt2 = wx.DataFormat(wx.DF_TEXT)
|
|
self.assertTrue(fmt1 != fmt2)
|
|
fmt3 = wx.DataFormat(wx.DF_TEXT)
|
|
self.assertTrue(fmt2 == fmt3)
|
|
|
|
|
|
|
|
def test_DataFormatIDsExist(self):
|
|
wx.DF_INVALID
|
|
wx.DF_TEXT
|
|
wx.DF_BITMAP
|
|
wx.DF_METAFILE
|
|
wx.DF_SYLK
|
|
wx.DF_DIF
|
|
wx.DF_TIFF
|
|
wx.DF_OEMTEXT
|
|
wx.DF_DIB
|
|
wx.DF_PALETTE
|
|
wx.DF_PENDATA
|
|
wx.DF_RIFF
|
|
wx.DF_WAVE
|
|
wx.DF_UNICODETEXT
|
|
wx.DF_ENHMETAFILE
|
|
wx.DF_FILENAME
|
|
wx.DF_LOCALE
|
|
wx.DF_PRIVATE
|
|
wx.DF_HTML
|
|
wx.DF_MAX
|
|
|
|
|
|
|
|
def test_DataObjectGetAllFormats(self):
|
|
class MyDataObject(wx.DataObject):
|
|
def __init__(self):
|
|
wx.DataObject.__init__(self)
|
|
self.myFormats = [wx.DataFormat(wx.DF_TEXT),
|
|
wx.DataFormat(wx.DF_UNICODETEXT)]
|
|
|
|
def GetAllFormats(self, d):
|
|
return self.myFormats
|
|
|
|
def GetFormatCount(self, d):
|
|
return len(self.myFormats)
|
|
|
|
data = MyDataObject()
|
|
if hasattr(data, '_testGetAllFormats'):
|
|
data._testGetAllFormats()
|
|
|
|
# TODO: Get this fixed! See https://groups.google.com/d/topic/wx-dev/wFxevpvbhvQ/discussion
|
|
@unittest.skipIf(sys.platform == 'darwin', 'Using wx.DF_TEXT currently fails on Mac')
|
|
def test_DataObject(self):
|
|
class MyDataObject(wx.DataObject):
|
|
def __init__(self, value=''):
|
|
wx.DataObject.__init__(self)
|
|
self.myFormats = [wx.DataFormat(wx.DF_TEXT)]
|
|
self.myData = wtc.mybytes(value)
|
|
|
|
def GetAllFormats(self, d):
|
|
return self.myFormats
|
|
def GetFormatCount(self, d):
|
|
return len(self.myFormats)
|
|
def GetPreferredFormat(self, d):
|
|
return self.myFormats[0]
|
|
|
|
def GetDataSize(self, format):
|
|
# On windows strlen is used internally for stock string types
|
|
# to calculate the size, so we need to make room for and add
|
|
# a null character at the end for our custom format using
|
|
# wx.DF_TEXT.
|
|
return len(self.myData)+1
|
|
|
|
def GetDataHere(self, format, buf):
|
|
# copy our local data value to buf
|
|
assert isinstance(buf, memoryview)
|
|
assert len(buf) == len(self.myData)+1
|
|
buf[:] = (self.myData + b'\0') # add the extra byte
|
|
return True
|
|
|
|
def SetData(self, format, buf):
|
|
# copy from buf to our local data value
|
|
assert isinstance(buf, memoryview)
|
|
self.myData = buf.tobytes()
|
|
if self.myData[-1] in [b'\0', 0]:
|
|
# strip off the extra byte at the end
|
|
self.myData = self.myData[:-1]
|
|
return True
|
|
|
|
# copy
|
|
data1 = MyDataObject('This is some data.')
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.SetData(data1)
|
|
wx.TheClipboard.Close()
|
|
|
|
# paste
|
|
data2 = MyDataObject()
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.GetData(data2)
|
|
wx.TheClipboard.Close()
|
|
|
|
self.assertEqual(data1.myData, data2.myData)
|
|
|
|
|
|
def test_DataObject2(self):
|
|
# More-or-less a duplicate of the above, but with a custom data format
|
|
class MyDataObject(wx.DataObject):
|
|
def __init__(self, value=''):
|
|
wx.DataObject.__init__(self)
|
|
self.myFormats = [wx.DataFormat("application.MyCustomDataFormat")]
|
|
self.myData = wtc.mybytes(value)
|
|
|
|
def GetAllFormats(self, d):
|
|
return self.myFormats
|
|
def GetFormatCount(self, d):
|
|
return len(self.myFormats)
|
|
def GetPreferredFormat(self, d):
|
|
return self.myFormats[0]
|
|
def GetDataSize(self, format):
|
|
return len(self.myData)
|
|
|
|
def GetDataHere(self, format, buf):
|
|
# copy our local data value to buf
|
|
assert isinstance(buf, memoryview)
|
|
buf[:] = self.myData
|
|
return True
|
|
|
|
def SetData(self, format, buf):
|
|
# copy from buf to our local data value
|
|
assert isinstance(buf, memoryview)
|
|
self.myData = buf.tobytes()
|
|
return True
|
|
|
|
# copy
|
|
data1 = MyDataObject('This is some data.')
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.SetData(data1)
|
|
wx.TheClipboard.Close()
|
|
|
|
# paste
|
|
data2 = MyDataObject()
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.GetData(data2)
|
|
wx.TheClipboard.Close()
|
|
|
|
self.assertEqual(data1.myData, data2.myData)
|
|
|
|
|
|
def test_DataObjectSimple1(self):
|
|
df = wx.DataFormat(wx.DF_TEXT)
|
|
dobj = wx.DataObjectSimple(df)
|
|
self.assertTrue(dobj.GetFormatCount() == 1)
|
|
self.assertTrue(dobj.GetFormat() == df)
|
|
self.assertTrue(dobj.GetAllFormats()[0] == df)
|
|
|
|
|
|
# TODO: Get this fixed! See https://groups.google.com/d/topic/wx-dev/wFxevpvbhvQ/discussion
|
|
@unittest.skipIf(sys.platform == 'darwin', 'Using wx.DF_TEXT currently fails on Mac')
|
|
def test_DataObjectSimple2(self):
|
|
class MyDataObject(wx.DataObjectSimple):
|
|
def __init__(self, value=''):
|
|
wx.DataObjectSimple.__init__(self)
|
|
self.SetFormat(wx.DataFormat(wx.DF_TEXT))
|
|
self.myData = wtc.mybytes(value)
|
|
|
|
def GetDataSize(self):
|
|
# See notes above about the need for a null terminator when
|
|
# using wx.DF_TEXT on Windows.
|
|
return len(self.myData)+1
|
|
|
|
def GetDataHere(self, buf):
|
|
# copy our local data value to buf
|
|
assert isinstance(buf, memoryview)
|
|
buf[:] = (self.myData + b'\0')
|
|
return True
|
|
|
|
def SetData(self, buf):
|
|
# copy from buf to our local data value
|
|
assert isinstance(buf, memoryview)
|
|
self.myData = buf.tobytes()
|
|
if self.myData[-1] in [b'\0', 0]:
|
|
self.myData = self.myData[:-1]
|
|
return True
|
|
|
|
# copy
|
|
data1 = MyDataObject('This is some data.')
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.SetData(data1)
|
|
wx.TheClipboard.Close()
|
|
|
|
# paste
|
|
data2 = MyDataObject()
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.GetData(data2)
|
|
wx.TheClipboard.Close()
|
|
|
|
self.assertEqual(data1.myData, data2.myData)
|
|
|
|
|
|
def test_DataObjectSimple3(self):
|
|
# More-or-less a duplicate of the above, but with a custom data format
|
|
class MyDataObject(wx.DataObjectSimple):
|
|
def __init__(self, value=''):
|
|
wx.DataObjectSimple.__init__(self)
|
|
self.SetFormat(wx.DataFormat("application.MyCustomDataFormat"))
|
|
self.myData = wtc.mybytes(value)
|
|
|
|
def GetDataSize(self):
|
|
return len(self.myData)
|
|
|
|
def GetDataHere(self, buf):
|
|
# copy our local data value to buf
|
|
assert isinstance(buf, memoryview)
|
|
buf[:] = self.myData
|
|
return True
|
|
|
|
def SetData(self, buf):
|
|
# copy from buf to our local data value
|
|
assert isinstance(buf, memoryview)
|
|
self.myData = buf.tobytes()
|
|
return True
|
|
|
|
# copy
|
|
data1 = MyDataObject('This is some data.')
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.SetData(data1)
|
|
wx.TheClipboard.Close()
|
|
|
|
# paste
|
|
data2 = MyDataObject()
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.GetData(data2)
|
|
wx.TheClipboard.Close()
|
|
|
|
self.assertEqual(data1.myData, data2.myData)
|
|
|
|
|
|
def test_CustomDataObject(self):
|
|
import pickle
|
|
data1 = list(range(10))
|
|
obj = wx.CustomDataObject('my custom format')
|
|
obj.SetData(pickle.dumps(data1))
|
|
assert obj.GetDataSize() > 0
|
|
data2 = pickle.loads(obj.GetData().tobytes())
|
|
self.assertEqual(data1, data2)
|
|
|
|
|
|
|
|
def test_DataObjectComposite(self):
|
|
do = wx.DataObjectComposite()
|
|
df1 = wx.DataFormat("data type 1")
|
|
df2 = wx.DataFormat("data type 2")
|
|
d1 = wx.CustomDataObject(df1)
|
|
d2 = wx.CustomDataObject(df2)
|
|
do.Add(d1, True)
|
|
do.Add(d2)
|
|
|
|
self.assertTrue(do.GetPreferredFormat() == df1)
|
|
d3 = do.GetObject(df2)
|
|
self.assertTrue(isinstance(d3, wx.CustomDataObject))
|
|
self.assertTrue(d3 is d2)
|
|
|
|
|
|
def test_BitmapDataObject(self):
|
|
do = wx.BitmapDataObject()
|
|
do.Bitmap = wx.Bitmap(pngFile)
|
|
self.assertTrue(do.GetBitmap().IsOk())
|
|
self.assertTrue(do.Bitmap.IsOk())
|
|
|
|
|
|
def test_TextDataObject(self):
|
|
data = "This is some data"
|
|
do = wx.TextDataObject(data)
|
|
self.assertEqual(do.GetText(), data)
|
|
self.assertEqual(do.Text, data)
|
|
self.assertAlmostEqual(do.GetTextLength(), len(data), delta=1)
|
|
self.assertAlmostEqual(do.TextLength, len(data), delta=1)
|
|
|
|
|
|
def test_TextDataObjectClipboard(self):
|
|
# copy
|
|
text = 'This is some MORE data.'
|
|
data1 = wx.TextDataObject(text)
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.SetData(data1)
|
|
wx.TheClipboard.Close()
|
|
|
|
# paste
|
|
data2 = wx.TextDataObject()
|
|
if wx.TheClipboard.Open():
|
|
wx.TheClipboard.GetData(data2)
|
|
wx.TheClipboard.Close()
|
|
|
|
self.assertEqual(text, data2.Text)
|
|
|
|
|
|
def test_URLDataObject(self):
|
|
url = 'http://wxPython.org/'
|
|
do = wx.URLDataObject()
|
|
do.URL = url
|
|
self.assertEqual(do.GetURL(), url)
|
|
self.assertEqual(do.URL, url)
|
|
|
|
|
|
def test_FileDataObject(self):
|
|
do = wx.FileDataObject()
|
|
do.AddFile('filename1')
|
|
do.AddFile('filename2')
|
|
do.AddFile('filename3')
|
|
|
|
names = do.GetFilenames()
|
|
self.assertTrue(len(names) == 3)
|
|
self.assertTrue(names[0] == 'filename1')
|
|
self.assertTrue(names == do.Filenames)
|
|
|
|
|
|
def test_HTMLDataObject(self):
|
|
data = "<html><body>This is some data</body></html>"
|
|
do = wx.HTMLDataObject(data)
|
|
self.assertEqual(do.GetHTML(), data)
|
|
self.assertEqual(do.HTML, data)
|
|
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|