Added all the tests. AH-294

Complete. Might need to look after if anything else needs testing like body in air.