BACKLOG-5328: Add unit tests