Add unit tests

[astrofrog]

In addition to having the 'live run' test, we could add some unit tests to make sure the environment variables are set correctly - this would allow tests to be run quickly and catch any issues early.