01TAWs6A4

If I am purchasing a parts car that is a LS1/T56 can I just swap over the engine and trans into my A4? I feel like its a stupid question, but I can't find it online. Anyone know how this will work with the wiring harness? That is my main concern the wiring and PCM

Thanks