Add tests

Problem/Motivation

The module does not have any test.

Steps to reproduce

Proposed resolution

Remaining tasks

User interface changes

API changes

Data model changes