* Add diagnostics * Add tests --------- Co-authored-by: Maciej Bieniek <478555+bieniu@users.noreply.github.com>