* Add the rest of the wx.DataObject classes and unittests.

* Enable all the important virtual methods to be overridden in Pytyhon subclasses.
* Use Python buffer object APIs and objects for data transfer where appropriate.

git-svn-id: https://svn.wxwidgets.org/svn/wx/wxPython/Phoenix/trunk@71412 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775

unittests/test_dataobj.py

@@ -1,6 +1,11 @@
 import imp_unittest, unittest
 import wtc
 import wx
+import os
+
+pngFile = os.path.join(os.path.dirname(__file__), 'toucan.png')
+
+#print os.getpid(); raw_input('Press enter...')
 
 #---------------------------------------------------------------------------
 
@@ -10,6 +15,9 @@ class DataObjTests(wtc.WidgetTestCase):
         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):
@@ -34,6 +42,178 @@ class DataObjTests(wtc.WidgetTestCase):
         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()
+        
+        
+        
+    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 = bytes(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)
+
+        
+    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 = bytes(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 = range(10)
+        obj = wx.CustomDataObject('my custom format')
+        obj.SetData(pickle.dumps(data1))
+        data2 = pickle.loads(obj.GetData().tobytes())
+        self.assertTrue(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_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)
+        
+        
         
 #---------------------------------------------------------------------------