Commit b98d9476b548bb1131102cbd5b3d4ad0ebfb6a5e
1 parent
dec22821
Implemented methods to obtain data from dataTable and schedule.
Showing
1 changed file
with
24 additions
and
4 deletions
@@ -6,7 +6,10 @@ import com.gargoylesoftware.htmlunit.html.DomNodeList; | @@ -6,7 +6,10 @@ import com.gargoylesoftware.htmlunit.html.DomNodeList; | ||
6 | import com.gargoylesoftware.htmlunit.html.HtmlElement; | 6 | import com.gargoylesoftware.htmlunit.html.HtmlElement; |
7 | import com.gargoylesoftware.htmlunit.html.HtmlInput; | 7 | import com.gargoylesoftware.htmlunit.html.HtmlInput; |
8 | import com.gargoylesoftware.htmlunit.html.HtmlPage; | 8 | import com.gargoylesoftware.htmlunit.html.HtmlPage; |
9 | +import com.gargoylesoftware.htmlunit.html.HtmlSpan; | ||
10 | +import com.gargoylesoftware.htmlunit.html.HtmlTable; | ||
9 | import com.gargoylesoftware.htmlunit.html.HtmlTableCell; | 11 | import com.gargoylesoftware.htmlunit.html.HtmlTableCell; |
12 | +import com.gargoylesoftware.htmlunit.html.HtmlTableRow; | ||
10 | 13 | ||
11 | import java.io.IOException; | 14 | import java.io.IOException; |
12 | import java.util.ArrayList; | 15 | import java.util.ArrayList; |
@@ -81,10 +84,12 @@ public class WebClientUtils { | @@ -81,10 +84,12 @@ public class WebClientUtils { | ||
81 | @SuppressWarnings("unchecked") | 84 | @SuppressWarnings("unchecked") |
82 | final DomNodeList<HtmlElement> suggestionRows = suggestElement.getElementsByTagName("tr"); | 85 | final DomNodeList<HtmlElement> suggestionRows = suggestElement.getElementsByTagName("tr"); |
83 | for (HtmlElement row : suggestionRows) { | 86 | for (HtmlElement row : suggestionRows) { |
84 | - @SuppressWarnings("unchecked") | ||
85 | - final DomNodeList<HtmlElement> cells = row.getElementsByTagName("td"); | ||
86 | - final HtmlTableCell cell = (HtmlTableCell) cells.get(column + 1); | ||
87 | - suggestions.put(cell.asText(), cell); | 87 | + if (!row.getId().endsWith("NothingLabel")) { |
88 | + @SuppressWarnings("unchecked") | ||
89 | + final DomNodeList<HtmlElement> cells = row.getElementsByTagName("td"); | ||
90 | + final HtmlTableCell cell = (HtmlTableCell) cells.get(column + 1); | ||
91 | + suggestions.put(cell.asText(), cell); | ||
92 | + } | ||
88 | } | 93 | } |
89 | return suggestions; | 94 | return suggestions; |
90 | } | 95 | } |
@@ -146,4 +151,19 @@ public class WebClientUtils { | @@ -146,4 +151,19 @@ public class WebClientUtils { | ||
146 | ((HtmlInput) page.getElementById(colorPicker.getId() + "-colorPicker-rgb-b")).setValueAttribute(Integer.toString(blue)); | 151 | ((HtmlInput) page.getElementById(colorPicker.getId() + "-colorPicker-rgb-b")).setValueAttribute(Integer.toString(blue)); |
147 | page.getElementById(colorPicker.getId() + "-colorPicker-popup").getHtmlElementsByTagName("button").get(0).click(); | 152 | page.getElementById(colorPicker.getId() + "-colorPicker-popup").getHtmlElementsByTagName("button").get(0).click(); |
148 | } | 153 | } |
154 | + | ||
155 | + @SuppressWarnings("unchecked") | ||
156 | + public static List<HtmlTableRow> getTableRows(HtmlTable table) { | ||
157 | + return (List<HtmlTableRow>) table.getByXPath(".//*[contains(@class,'rich-table-row')]"); | ||
158 | + } | ||
159 | + | ||
160 | + @SuppressWarnings("unchecked") | ||
161 | + public static List<String> getScheduleEventTitles(HtmlElement schedule) { | ||
162 | + final List<HtmlSpan> titleElements = (List<HtmlSpan>) schedule.getByXPath(".//*[@class='fc-event-title']"); | ||
163 | + List<String> titles = new ArrayList<String>(titleElements.size()); | ||
164 | + for (HtmlSpan o : titleElements) { | ||
165 | + titles.add(o.asText()); | ||
166 | + } | ||
167 | + return titles; | ||
168 | + } | ||
149 | } | 169 | } |
Please
register
or
login
to post a comment