def test_query_request_collection(self): query = DotDict() query.collection = 'test' self.discovery.query_collection = Mock() self.discovery.query_collection.return_value = 'test' retval = self.discovery.query_request(query) self.assertTrue(retval == 'test')