Add unit tests for lint stage

Background

Acceptance Criteria

Acceptance Criteria:

• Code coverage increase from 31% to 70% for lint stage code
• Code coverage increase from 43% to 70% for module code

Definition of Done

• Code coverage is increased for both lint and module code
• All functions/methods have an existing test
  • Test does not need to completely test functionality yet
  • Test can be empty if functionality is too difficult to mock