68e589a7c8c4

--- a/review/tests/test_diffs.py	Sat Oct 10 10:51:02 2009 -0400
+++ b/review/tests/test_diffs.py	Sat Oct 10 10:51:51 2009 -0400
@@ -63,3 +63,49 @@
     
     assert '17:' not in output
     assert messages.REVIEW_LOG_SKIPPED % 5 in output
+
+
+@with_setup(setup_reviewed_sandbox, teardown_sandbox)
+def test_review_diff_with_comment():
+    sandbox = get_sandbox_repo()
+    
+    review(comment=True, rev='1', message='Test comment one.',
+        files=['long_file'], lines='6,7')
+    
+    output = review(rev='1', files=['long_file'], unified=0)
+    
+    # Make sure the comment is present at all.
+    assert a1 in output
+    assert a2 in output
+    assert messages.REVIEW_LOG_COMMENT_LINE % 'Test comment one.' in output
+    
+    # Make sure it's in the correct place
+    output = output.splitlines()
+    for n, line in enumerate(output):
+        if line.startswith('#'):
+            assert output[n-1].strip().startswith('7')
+            break
+
+
+@with_setup(setup_reviewed_sandbox, teardown_sandbox)
+def test_review_diff_with_skipped_comment():
+    sandbox = get_sandbox_repo()
+    
+    review(comment=True, rev='1', message='Test comment one.',
+        files=['long_file'], lines='3')
+    
+    output = review(rev='1', files=['long_file'], unified=0)
+    
+    # Make sure the comment is present at all.
+    assert a1 in output
+    assert a2 in output
+    assert messages.REVIEW_LOG_COMMENT_LINE % 'Test comment one.' in output
+    
+    # Make sure it's in the correct place
+    output = output.splitlines()
+    for n, line in enumerate(output):
+        if line.startswith('#'):
+            assert output[n-1].startswith(s1)
+            assert output[n-1].endswith(s2.strip())
+            break
+