Get code coverage to 100%

Running:

```bash
$ npm run coverage
```

Shows current coverage at:

=============================== Coverage summary ===============================
Statements   : 80.39% ( 287/357 )
Branches     : 75.77% ( 247/326 )
Functions    : 95% ( 38/40 )
Lines        : 80.39% ( 287/357 )

Would like to get this to 100% with a stronger focus on unit tests than the existing e2e tests.