@@ -20,6 +20,17 @@ func TestExtractCategoryByIndex(t *testing.T) {
2020 }
2121}
2222
23+ func TestExtractCategoryByIndexFirst (t * testing.T ) {
24+ got , err := ExtractCategoryByIndex (testBookmarkText , 0 )
25+ if err != nil {
26+ t .Fatalf ("unexpected error: %v" , err )
27+ }
28+ expected := "Category: A\n http://a.com a"
29+ if got != expected {
30+ t .Fatalf ("expected %q got %q" , expected , got )
31+ }
32+ }
33+
2334func TestExtractCategoryByIndexError (t * testing.T ) {
2435 if _ , err := ExtractCategoryByIndex (testBookmarkText , 5 ); err == nil {
2536 t .Fatalf ("expected error" )
@@ -38,6 +49,18 @@ func TestReplaceCategoryByIndex(t *testing.T) {
3849 }
3950}
4051
52+ func TestReplaceCategoryByIndexFirst (t * testing.T ) {
53+ newSection := "Category: A\n http://changed.com x"
54+ updated , err := ReplaceCategoryByIndex (testBookmarkText , 0 , newSection )
55+ if err != nil {
56+ t .Fatalf ("unexpected error: %v" , err )
57+ }
58+ expected := newSection + "\n Column\n Category: B\n http://b.com b\n "
59+ if updated != expected {
60+ t .Fatalf ("expected %q got %q" , expected , updated )
61+ }
62+ }
63+
4164func TestReplaceCategoryByIndexError (t * testing.T ) {
4265 if _ , err := ReplaceCategoryByIndex (testBookmarkText , 3 , "foo" ); err == nil {
4366 t .Fatalf ("expected error" )
0 commit comments