gobookmarks/bookmarkCategoryEdit_test.go at main · arran4/gobookmarks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
package gobookmarks

import "testing"

const testBookmarkText = `Category: A
http://a.com a
Column
Category: B
http://b.com b
`

const anonBookmarkText = `Category:
http://one.com one
Category:
http://two.com two
Category: Named
http://three.com three
`

func TestExtractCategoryByIndex(t *testing.T) {
	got, err := ExtractCategoryByIndex(testBookmarkText, 1)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := "Category: B\nhttp://b.com b\n"
	if got != expected {
		t.Fatalf("expected %q got %q", expected, got)
	}
}

func TestExtractCategoryByIndexFirst(t *testing.T) {
	got, err := ExtractCategoryByIndex(testBookmarkText, 0)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := "Category: A\nhttp://a.com a"
	if got != expected {
		t.Fatalf("expected %q got %q", expected, got)
	}
}

func TestExtractCategoryByIndexError(t *testing.T) {
	if _, err := ExtractCategoryByIndex(testBookmarkText, 5); err == nil {
		t.Fatalf("expected error")
	}
}

func TestReplaceCategoryByIndex(t *testing.T) {
	newSection := "Category: B\nhttp://new.com n"
	updated, err := ReplaceCategoryByIndex(testBookmarkText, 1, newSection)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := "Category: A\nhttp://a.com a\nColumn\n" + newSection
	if updated != expected {
		t.Fatalf("expected %q got %q", expected, updated)
	}
}

func TestReplaceCategoryByIndexFirst(t *testing.T) {
	newSection := "Category: A\nhttp://changed.com x"
	updated, err := ReplaceCategoryByIndex(testBookmarkText, 0, newSection)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := newSection + "\nColumn\nCategory: B\nhttp://b.com b\n"
	if updated != expected {
		t.Fatalf("expected %q got %q", expected, updated)
	}
}

func TestReplaceCategoryByIndexError(t *testing.T) {
	if _, err := ReplaceCategoryByIndex(testBookmarkText, 3, "foo"); err == nil {
		t.Fatalf("expected error")
	}
}

func TestExtractCategoryByIndexAnonymous(t *testing.T) {
	got, err := ExtractCategoryByIndex(anonBookmarkText, 1)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := "Category:\nhttp://two.com two"
	if got != expected {
		t.Fatalf("expected %q got %q", expected, got)
	}
}

func TestReplaceCategoryByIndexAnonymous(t *testing.T) {
	newSection := "Category:\nhttp://changed.com x"
	updated, err := ReplaceCategoryByIndex(anonBookmarkText, 0, newSection)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := newSection + "\nCategory:\nhttp://two.com two\nCategory: Named\nhttp://three.com three\n"
	if updated != expected {
		t.Fatalf("expected %q got %q", expected, updated)
	}
}

const multiBookmarkText = `Category: A
http://a.com a
Page
Category: B
http://b.com b
Column
Category: C
http://c.com c
`

func TestReplaceCategoryWithPageAndColumn(t *testing.T) {
	newSec := "Category: B\nhttp://new.com"
	updated, err := ReplaceCategoryByIndex(multiBookmarkText, 1, newSec)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := "Category: A\nhttp://a.com a\nPage\n" + newSec + "\nColumn\nCategory: C\nhttp://c.com c\n"
	if updated != expected {
		t.Fatalf("expected %q got %q", expected, updated)
	}
}

func TestReplaceCategoryLastWithPageAndColumn(t *testing.T) {
	newSec := "Category: C\nhttp://changed.com"
	updated, err := ReplaceCategoryByIndex(multiBookmarkText, 2, newSec)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	expected := "Category: A\nhttp://a.com a\nPage\nCategory: B\nhttp://b.com b\nColumn\n" + newSec
	if updated != expected {
		t.Fatalf("expected %q got %q", expected, updated)
	}
}