fix imports

Signed-off-by: Jaka Hudoklin <jakahudoklin@gmail.com>
add support for testing with users with roles

Signed-off-by: Jaka Hudoklin <jakahudoklin@gmail.com>