Add automated test coverage

Problem/Motivation

There are currently no automated tests included with the module to check the element works as expected.

Proposed resolution

Add automated test coverage.

Remaining tasks

1. Write a patch
2. Review
3. Commit

User interface changes

None.

API changes

None.

Data model changes

None.