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 package info.magnolia.module.googlesitemap.app.subapp.sitemapdetail.contentviews.formatter;
35
36 import static org.hamcrest.CoreMatchers.is;
37 import static org.junit.Assert.assertThat;
38 import static org.mockito.Mockito.mock;
39
40 import info.magnolia.jcr.util.NodeTypes;
41 import info.magnolia.test.mock.jcr.MockNode;
42 import info.magnolia.ui.vaadin.integration.jcr.DefaultProperty;
43 import info.magnolia.ui.vaadin.integration.jcr.JcrNodeAdapter;
44 import info.magnolia.ui.workbench.column.definition.PropertyColumnDefinition;
45 import info.magnolia.util.EscapeUtil;
46
47 import javax.jcr.Node;
48 import javax.jcr.RepositoryException;
49
50 import org.junit.Before;
51 import org.junit.Test;
52 import org.mockito.Mockito;
53
54 import com.vaadin.data.Item;
55 import com.vaadin.ui.Table;
56
57
58
59
60 public class FolderNameColumnFormatterTest {
61
62 private final String columnId = "displayName";
63
64 private Table table;
65 private Item item;
66 private Object itemId;
67
68 private PropertyColumnDefinition propertyColumnDefinition;
69 private FolderNameColumnFormatter folderNameColumnFormatter;
70
71 @Before
72 public void setIp() {
73 table = mock(Table.class);
74 item = mock(Item.class);
75 itemId = item;
76
77 propertyColumnDefinition = new PropertyColumnDefinition();
78 folderNameColumnFormatter = new FolderNameColumnFormatter(propertyColumnDefinition);
79 }
80
81 @Test
82 public void testGenerateCellWhenFolder() throws RepositoryException {
83
84 final String nodeName = "nodeName";
85
86 JcrNodeAdapter nodeAdapter = mock(JcrNodeAdapter.class);
87 Node node = new MockNode(nodeName, NodeTypes.Folder.NAME);
88
89 Mockito.when(nodeAdapter.getJcrItem()).thenReturn(node);
90 Mockito.when(table.getItem(itemId)).thenReturn(nodeAdapter);
91
92
93 Object cell = folderNameColumnFormatter.generateCell(table, itemId, columnId);
94
95
96 assertThat((String) cell, is(nodeName));
97 }
98
99 @Test
100 public void testGenerateCellWhenItemPropertyIsNotNull() {
101
102 final String value = "testValue";
103 Mockito.when(table.getItem(itemId)).thenReturn(item);
104 Mockito.when(item.getItemProperty(columnId)).thenReturn(new DefaultProperty(value));
105
106
107 Object cell = folderNameColumnFormatter.generateCell(table, itemId, columnId);
108
109
110 assertThat((String) cell, is(value));
111 }
112
113 @Test
114 public void testGenerateCellWhenItemPropertyIsNull() {
115
116 Mockito.when(table.getItem(itemId)).thenReturn(item);
117 Mockito.when(item.getItemProperty(columnId)).thenReturn(null);
118
119
120 Object cell = folderNameColumnFormatter.generateCell(table, itemId, columnId);
121
122
123 assertThat((String) cell, is(""));
124 }
125
126 @Test
127 public void testGenerateCellWhenItemIsNull() {
128
129 Mockito.when(table.getItem(itemId)).thenReturn(null);
130
131
132 Object cell = folderNameColumnFormatter.generateCell(table, itemId, columnId);
133
134
135 assertThat((String) cell, is(""));
136 }
137
138 @Test
139 public void testXSSOnItemName() throws Exception {
140
141 String unEscapedHTML = "<A onmouseover=alert('XSS_by_Vishal_V_Sonar_&_Akash_Chavan')> XSS </A>";
142
143 Mockito.when(table.getItem(itemId)).thenReturn(item);
144 Mockito.when(item.getItemProperty(columnId)).thenReturn(new DefaultProperty(unEscapedHTML));
145
146
147 Object cell = folderNameColumnFormatter.generateCell(table, itemId, columnId);
148
149
150 assertThat((String) cell, is(EscapeUtil.escapeXss(unEscapedHTML)));
151 }
152
153 }