[wip] added unit tests