tests: Update tests affected by multi-source handling changes

Signed-off-by: Euan Harris <euan.harris@citrix.com>