WSOTWO-135: Added unit tests for diagram element widget