From 04210abd7092322022aa504af23464f056e2f3e1 Mon Sep 17 00:00:00 2001
From: Leandro Freitas <llf@certi.org.br>
Date: Wed, 6 May 2015 10:00:35 -0300
Subject: [PATCH] Fixes #37. Replacing existing priority

---
 autoload/todo/txt.vim     |  2 +-
 test/tc_priority.todo.txt |  7 +++++++
 test/tc_priority.vim      | 28 ++++++++++++++++++++++++++++
 3 files changed, 36 insertions(+), 1 deletion(-)
 create mode 100644 test/tc_priority.todo.txt
 create mode 100644 test/tc_priority.vim

diff --git a/autoload/todo/txt.vim b/autoload/todo/txt.vim
index e41dbc4..6c59ee4 100644
--- a/autoload/todo/txt.vim
+++ b/autoload/todo/txt.vim
@@ -104,7 +104,7 @@ function! todo#txt#prioritize_add(priority)
 endfunction
 
 function! todo#txt#prioritize_add_action(priority)
-    execute "normal! mq0i(".a:priority.") \<esc>`q"
+    execute 's/^\(([a-zA-Z]) \)\?/(' . a:priority . ') /'
 endfunction
 
 " Modeline {{{1
diff --git a/test/tc_priority.todo.txt b/test/tc_priority.todo.txt
new file mode 100644
index 0000000..7cc2126
--- /dev/null
+++ b/test/tc_priority.todo.txt
@@ -0,0 +1,7 @@
+# insert_priority
+example task
+# end_insert_priority
+
+# replace_priority
+(A) example task
+# end_replace_priority
diff --git a/test/tc_priority.vim b/test/tc_priority.vim
new file mode 100644
index 0000000..7aa361d
--- /dev/null
+++ b/test/tc_priority.vim
@@ -0,0 +1,28 @@
+let s:here = expand('<sfile>:p:h')
+let s:context = todo#txt#__context__()
+let s:context['data'] = s:here . '/tc_priority.todo.txt'
+let s:tc = unittest#testcase#new('Priority', s:context)
+
+let s:TODAY = strftime("%Y-%m-%d")
+
+let s:PRIORITY_INSERTED = [
+      \ '(A) example task',
+      \ ]
+
+let s:PRIORITY_REPLACED = [
+      \ '(C) example task',
+      \ ]
+
+function! s:tc.test_insert_priority()
+  call self.data.goto('insert_priority')
+  call todo#txt#prioritize_add('A')
+  call self.assert_equal(s:PRIORITY_INSERTED, self.data.get('insert_priority'))
+endfunction
+
+function! s:tc.test_replace_priority()
+  call self.data.goto('replace_priority')
+  call todo#txt#prioritize_add('C')
+  call self.assert_equal(s:PRIORITY_REPLACED, self.data.get('replace_priority'))
+endfunction
+
+unlet s:tc