import buffer2png
import unittest
import gst

class TestBuffer:
    def __init__(self):
        self.src = gst.element_factory_make('filesrc')
        self.gdpdepay = gst.element_factory_make('gdpdepay')
        self.sink = gst.element_factory_make('fakesink')
        self.pipeline = gst.Pipeline()
        self.pipeline.add(self.src,self.gdpdepay,self.sink)
        self.src.link(self.gdpdepay)
        self.gdpdepay.link(self.sink)
        self.src.set_property('location','testbuffer')
        self.sink.set_property("signal-handoffs", True)
    
        self.pipeline.get_bus().add_signal_watch()
        self.pipeline.get_bus().connect('message::handoff', self.__handoff_handler__)
    
        self.pipeline.set_state (gst.STATE_PLAYING)
        
    def __handoff_handler__(self, buffer, pad, user_data):
        print "I am in __handoff_handler__"
        self.buffer = buffer;
        
    def get_buffer(self):
        return self.buffer

class Testbuffer2pngfunction(unittest.TestCase):
    def setUp(self):
        mytestbuffer = TestBuffer()
        self.buffer = mytestbuffer.get_buffer()
    def tearDown(self):
        pass
    def testconversion(self):
        # Test with real-world buffer
        buffer2png.buffer2png(self.buffer, 'result.png')
        
if __name__ == '__main__':
    unittest.main()