Add missing  methods to mock module