Isn't this just simply a matter of collecting measurements, Mike?
With your extended swingarm (sans gearcase) in place, and the extended driveshaft seated in the splines of the u-joint, there is now a set distance between the outboard end of the shaft and the mating flange of the swingarm.
And the distance between the mating flange of the gearcase and the total depth of the gear coupling can be measured, before installation.
Perhaps a quicker way to check - does your setup still use the compression spring? The total amount of compression could be measured on a bench, then compared to the stand-off measurement when the mating flanges are brought together, before tightening the gearcase nuts. Of course this measurement should be less. The difference in the measurements should be the amount of shaft axial movement allowed, with everything tight.
Does this make sense, Mr. Morley??