Using mock to test django orm