def test_query_request_association(self): query = DotDict() query.association = 'resource_id' self.discovery.query_association = Mock() self.discovery.query_association.return_value = 'test' retval = self.discovery.query_request(query) self.assertTrue(retval == 'test')