Python mock class in another class [on hold]

I'm trying to test one class which is using instance of another class and calls its method which use DB. I'd like to mock that class with DB but unfortunately my code is not working.

Here is prototype of what I'm trying: http://ift.tt/2nm9er9

Thank you