Add test for Diagnostic