Add basic tests

Signed-off-by: Ben Agricola <bagricola@squiz.co.uk>