Add keyword module tests

Last of the frontend tests! :sweat_smile: