More fixes for test

2 files changed, 8 insertions, 2 deletions

diff --git a/features/steps/shared/diff_note.rb b/features/steps/shared/diff_note.rb
index 431ef022d0b..2ae0f1241b8 100644
--- a/features/steps/shared/diff_note.rb
+++ b/features/steps/shared/diff_note.rb
@@ -8,6 +8,7 @@ module SharedDiffNote
   end
 
   Given 'I delete a diff comment' do
+    sleep 1
     within(".diff_file") do
       first(".js-note-delete").trigger("click")
     end
@@ -25,6 +26,7 @@ module SharedDiffNote
       fill_in "note[note]", with: "Typo, please fix"
       #click_button("Add Comment")
       find(".js-comment-button").trigger("click")
+      sleep 0.05
     end
   end
 
diff --git a/features/steps/shared/note.rb b/features/steps/shared/note.rb
index 934aba6a91a..5dcc75f9165 100644
--- a/features/steps/shared/note.rb
+++ b/features/steps/shared/note.rb
@@ -39,8 +39,6 @@ module SharedNote
     end
   end
 
-
-
   Then 'I should not see a comment saying "XML attached"' do
     page.should_not have_css(".note")
   end
@@ -113,4 +111,10 @@ module SharedNote
   Then 'I should see project wall note "my special test message"' do
     page.should have_content "my special test message"
   end
+
+  Then 'I should see comment "XML attached"' do
+    within(".note") do
+      page.should have_content("XML attached")
+    end
+  end
 end