You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

TestXWPFRun.java 29KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859
  1. /* ====================================================================
  2. Licensed to the Apache Software Foundation (ASF) under one or more
  3. contributor license agreements. See the NOTICE file distributed with
  4. this work for additional information regarding copyright ownership.
  5. The ASF licenses this file to You under the Apache License, Version 2.0
  6. (the "License"); you may not use this file except in compliance with
  7. the License. You may obtain a copy of the License at
  8. http://www.apache.org/licenses/LICENSE-2.0
  9. Unless required by applicable law or agreed to in writing, software
  10. distributed under the License is distributed on an "AS IS" BASIS,
  11. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. See the License for the specific language governing permissions and
  13. limitations under the License.
  14. ==================================================================== */
  15. package org.apache.poi.xwpf.usermodel;
  16. import static org.junit.Assert.assertEquals;
  17. import static org.junit.Assert.assertFalse;
  18. import static org.junit.Assert.assertNotNull;
  19. import static org.junit.Assert.assertNull;
  20. import static org.junit.Assert.assertTrue;
  21. import java.io.ByteArrayInputStream;
  22. import java.io.ByteArrayOutputStream;
  23. import java.io.IOException;
  24. import java.math.BigInteger;
  25. import java.util.Iterator;
  26. import java.util.List;
  27. import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
  28. import org.apache.poi.util.LocaleUtil;
  29. import org.apache.poi.util.Units;
  30. import org.apache.poi.wp.usermodel.HeaderFooterType;
  31. import org.apache.poi.xwpf.XWPFTestDataSamples;
  32. import org.apache.poi.xwpf.model.XWPFHeaderFooterPolicy;
  33. import org.junit.After;
  34. import org.junit.Before;
  35. import org.junit.Test;
  36. import org.openxmlformats.schemas.drawingml.x2006.main.CTBlip;
  37. import org.openxmlformats.schemas.drawingml.x2006.main.CTBlipFillProperties;
  38. import org.openxmlformats.schemas.drawingml.x2006.picture.CTPicture;
  39. import org.openxmlformats.schemas.officeDocument.x2006.sharedTypes.STOnOff1;
  40. import org.openxmlformats.schemas.officeDocument.x2006.sharedTypes.STVerticalAlignRun;
  41. import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTBr;
  42. import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTOnOff;
  43. import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTP;
  44. import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTR;
  45. import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTRPr;
  46. import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTText;
  47. import org.openxmlformats.schemas.wordprocessingml.x2006.main.STBrClear;
  48. import org.openxmlformats.schemas.wordprocessingml.x2006.main.STEm;
  49. import org.openxmlformats.schemas.wordprocessingml.x2006.main.STHighlightColor;
  50. import org.openxmlformats.schemas.wordprocessingml.x2006.main.STThemeColor;
  51. import org.openxmlformats.schemas.wordprocessingml.x2006.main.STUnderline;
  52. /**
  53. * Tests for XWPF Run
  54. */
  55. public class TestXWPFRun {
  56. private CTR ctRun;
  57. private XWPFParagraph p;
  58. private IRunBody irb;
  59. private XWPFDocument doc;
  60. @Before
  61. public void setUp() {
  62. doc = new XWPFDocument();
  63. p = doc.createParagraph();
  64. irb = p;
  65. this.ctRun = CTR.Factory.newInstance();
  66. }
  67. @After
  68. public void tearDown() throws Exception {
  69. doc.close();
  70. }
  71. @Test
  72. public void testSetGetText() {
  73. ctRun.addNewT().setStringValue("TEST STRING");
  74. ctRun.addNewT().setStringValue("TEST2 STRING");
  75. ctRun.addNewT().setStringValue("TEST3 STRING");
  76. assertEquals(3, ctRun.sizeOfTArray());
  77. XWPFRun run = new XWPFRun(ctRun, irb);
  78. assertEquals("TEST2 STRING", run.getText(1));
  79. run.setText("NEW STRING", 0);
  80. assertEquals("NEW STRING", run.getText(0));
  81. //run.setText("xxx",14);
  82. //fail("Position wrong");
  83. }
  84. /*
  85. * bug 59208
  86. * Purpose: test all valid boolean-like values
  87. * exercise isCTOnOff(CTOnOff) through all valid permutations
  88. */
  89. @Test
  90. public void testCTOnOff() {
  91. CTRPr rpr = ctRun.addNewRPr();
  92. CTOnOff bold = rpr.addNewB();
  93. XWPFRun run = new XWPFRun(ctRun, irb);
  94. // True values: "true", "1", "on"
  95. bold.setVal(STOnOff1.ON);
  96. assertTrue(run.isBold());
  97. bold.setVal(STOnOff1.ON);
  98. assertTrue(run.isBold());
  99. bold.setVal(STOnOff1.ON);
  100. assertTrue(run.isBold());
  101. // False values: "false", "0", "off"
  102. bold.setVal(STOnOff1.OFF);
  103. assertFalse(run.isBold());
  104. bold.setVal(STOnOff1.OFF);
  105. assertFalse(run.isBold());
  106. bold.setVal(STOnOff1.OFF);
  107. assertFalse(run.isBold());
  108. }
  109. @Test
  110. public void testSetGetBold() {
  111. CTRPr rpr = ctRun.addNewRPr();
  112. rpr.addNewB().setVal(STOnOff1.ON);
  113. XWPFRun run = new XWPFRun(ctRun, irb);
  114. assertTrue(run.isBold());
  115. run.setBold(false);
  116. // Implementation detail: POI natively prefers <w:b w:val="false"/>,
  117. // but should correctly read val="0" and val="off"
  118. assertEquals("off", rpr.getB().getVal());
  119. }
  120. @Test
  121. public void testSetGetItalic() {
  122. CTRPr rpr = ctRun.addNewRPr();
  123. rpr.addNewI().setVal(STOnOff1.ON);
  124. XWPFRun run = new XWPFRun(ctRun, irb);
  125. assertTrue(run.isItalic());
  126. run.setItalic(false);
  127. assertEquals("off", rpr.getI().getVal());
  128. }
  129. @Test
  130. public void testSetGetStrike() {
  131. CTRPr rpr = ctRun.addNewRPr();
  132. rpr.addNewStrike().setVal(STOnOff1.ON);
  133. XWPFRun run = new XWPFRun(ctRun, irb);
  134. assertTrue(run.isStrikeThrough());
  135. run.setStrikeThrough(false);
  136. assertEquals("off", rpr.getStrike().getVal());
  137. }
  138. @Test
  139. public void testSetGetUnderline() {
  140. CTRPr rpr = ctRun.addNewRPr();
  141. XWPFRun run = new XWPFRun(ctRun, irb);
  142. rpr.addNewU().setVal(STUnderline.DASH);
  143. assertEquals(UnderlinePatterns.DASH.getValue(), run.getUnderline()
  144. .getValue());
  145. run.setUnderline(UnderlinePatterns.NONE);
  146. assertEquals(STUnderline.NONE.intValue(), rpr.getU().getVal()
  147. .intValue());
  148. }
  149. @Test
  150. public void testSetGetFontFamily() {
  151. CTRPr rpr = ctRun.addNewRPr();
  152. rpr.addNewRFonts().setAscii("Times New Roman");
  153. XWPFRun run = new XWPFRun(ctRun, irb);
  154. assertEquals("Times New Roman", run.getFontFamily());
  155. run.setFontFamily("Verdana");
  156. assertEquals("Verdana", rpr.getRFonts().getAscii());
  157. }
  158. @Test
  159. public void testSetGetFontSize() {
  160. CTRPr rpr = ctRun.addNewRPr();
  161. rpr.addNewSz().setVal(BigInteger.valueOf(14));
  162. XWPFRun run = new XWPFRun(ctRun, irb);
  163. assertEquals(7, run.getFontSize());
  164. assertEquals(7.0, run.getFontSizeAsDouble(), 0.01);
  165. run.setFontSize(24);
  166. assertEquals("48", rpr.getSz().getVal().toString());
  167. run.setFontSize(24.5f);
  168. assertEquals("49", rpr.getSz().getVal().toString());
  169. assertEquals(25, run.getFontSize());
  170. assertEquals(24.5, run.getFontSizeAsDouble(), 0.01);
  171. }
  172. @Test
  173. public void testSetGetTextForegroundBackground() {
  174. CTRPr rpr = ctRun.addNewRPr();
  175. rpr.addNewPosition().setVal(new BigInteger("4000"));
  176. XWPFRun run = new XWPFRun(ctRun, irb);
  177. assertEquals(4000, run.getTextPosition());
  178. run.setTextPosition(2400);
  179. assertEquals("2400", rpr.getPosition().getVal().toString());
  180. }
  181. @Test
  182. public void testSetGetColor() {
  183. XWPFRun run = new XWPFRun(ctRun, irb);
  184. run.setColor("0F0F0F");
  185. String clr = run.getColor();
  186. assertEquals("0F0F0F", clr);
  187. }
  188. @Test
  189. public void testAddCarriageReturn() {
  190. ctRun.addNewT().setStringValue("TEST STRING");
  191. ctRun.addNewCr();
  192. ctRun.addNewT().setStringValue("TEST2 STRING");
  193. ctRun.addNewCr();
  194. ctRun.addNewT().setStringValue("TEST3 STRING");
  195. assertEquals(2, ctRun.sizeOfCrArray());
  196. XWPFRun run = new XWPFRun(CTR.Factory.newInstance(), irb);
  197. run.setText("T1");
  198. run.addCarriageReturn();
  199. run.addCarriageReturn();
  200. run.setText("T2");
  201. run.addCarriageReturn();
  202. assertEquals(3, run.getCTR().sizeOfCrArray());
  203. assertEquals("T1\n\nT2\n", run.toString());
  204. }
  205. @Test
  206. public void testAddTabsAndLineBreaks() {
  207. ctRun.addNewT().setStringValue("TEST STRING");
  208. ctRun.addNewCr();
  209. ctRun.addNewT().setStringValue("TEST2 STRING");
  210. ctRun.addNewTab();
  211. ctRun.addNewT().setStringValue("TEST3 STRING");
  212. assertEquals(1, ctRun.sizeOfCrArray());
  213. assertEquals(1, ctRun.sizeOfTabArray());
  214. XWPFRun run = new XWPFRun(CTR.Factory.newInstance(), irb);
  215. run.setText("T1");
  216. run.addCarriageReturn();
  217. run.setText("T2");
  218. run.addTab();
  219. run.setText("T3");
  220. assertEquals(1, run.getCTR().sizeOfCrArray());
  221. assertEquals(1, run.getCTR().sizeOfTabArray());
  222. assertEquals("T1\nT2\tT3", run.toString());
  223. }
  224. @Test
  225. public void testAddPageBreak() {
  226. ctRun.addNewT().setStringValue("TEST STRING");
  227. ctRun.addNewBr();
  228. ctRun.addNewT().setStringValue("TEST2 STRING");
  229. CTBr breac = ctRun.addNewBr();
  230. breac.setClear(STBrClear.LEFT);
  231. ctRun.addNewT().setStringValue("TEST3 STRING");
  232. assertEquals(2, ctRun.sizeOfBrArray());
  233. XWPFRun run = new XWPFRun(CTR.Factory.newInstance(), irb);
  234. run.setText("TEXT1");
  235. run.addBreak();
  236. run.setText("TEXT2");
  237. run.addBreak(BreakType.TEXT_WRAPPING);
  238. assertEquals(2, run.getCTR().sizeOfBrArray());
  239. }
  240. /**
  241. * Test that on an existing document, we do the
  242. * right thing with it
  243. */
  244. @Test
  245. public void testExisting() throws IOException {
  246. XWPFDocument doc = XWPFTestDataSamples.openSampleDocument("TestDocument.docx");
  247. XWPFParagraph p;
  248. XWPFRun run;
  249. // First paragraph is simple
  250. p = doc.getParagraphArray(0);
  251. assertEquals("This is a test document.", p.getText());
  252. assertEquals(2, p.getRuns().size());
  253. run = p.getRuns().get(0);
  254. assertEquals("This is a test document", run.toString());
  255. assertFalse(run.isBold());
  256. assertFalse(run.isItalic());
  257. assertFalse(run.isStrikeThrough());
  258. assertNull(run.getCTR().getRPr());
  259. run = p.getRuns().get(1);
  260. assertEquals(".", run.toString());
  261. assertFalse(run.isBold());
  262. assertFalse(run.isItalic());
  263. assertFalse(run.isStrikeThrough());
  264. assertNull(run.getCTR().getRPr());
  265. // Next paragraph is all in one style, but a different one
  266. p = doc.getParagraphArray(1);
  267. assertEquals("This bit is in bold and italic", p.getText());
  268. assertEquals(1, p.getRuns().size());
  269. run = p.getRuns().get(0);
  270. assertEquals("This bit is in bold and italic", run.toString());
  271. assertTrue(run.isBold());
  272. assertTrue(run.isItalic());
  273. assertFalse(run.isStrikeThrough());
  274. assertTrue(run.getCTR().getRPr().isSetB());
  275. assertFalse(run.getCTR().getRPr().getB().isSetVal());
  276. // Back to normal
  277. p = doc.getParagraphArray(2);
  278. assertEquals("Back to normal", p.getText());
  279. assertEquals(1, p.getRuns().size());
  280. run = p.getRuns().get(0);
  281. assertEquals("Back to normal", run.toString());
  282. assertFalse(run.isBold());
  283. assertFalse(run.isItalic());
  284. assertFalse(run.isStrikeThrough());
  285. assertNull(run.getCTR().getRPr());
  286. // Different styles in one paragraph
  287. p = doc.getParagraphArray(3);
  288. assertEquals("This contains BOLD, ITALIC and BOTH, as well as RED and YELLOW text.", p.getText());
  289. assertEquals(11, p.getRuns().size());
  290. run = p.getRuns().get(0);
  291. assertEquals("This contains ", run.toString());
  292. assertFalse(run.isBold());
  293. assertFalse(run.isItalic());
  294. assertFalse(run.isStrikeThrough());
  295. assertNull(run.getCTR().getRPr());
  296. run = p.getRuns().get(1);
  297. assertEquals("BOLD", run.toString());
  298. assertTrue(run.isBold());
  299. assertFalse(run.isItalic());
  300. assertFalse(run.isStrikeThrough());
  301. run = p.getRuns().get(2);
  302. assertEquals(", ", run.toString());
  303. assertFalse(run.isBold());
  304. assertFalse(run.isItalic());
  305. assertFalse(run.isStrikeThrough());
  306. assertNull(run.getCTR().getRPr());
  307. run = p.getRuns().get(3);
  308. assertEquals("ITALIC", run.toString());
  309. assertFalse(run.isBold());
  310. assertTrue(run.isItalic());
  311. assertFalse(run.isStrikeThrough());
  312. run = p.getRuns().get(4);
  313. assertEquals(" and ", run.toString());
  314. assertFalse(run.isBold());
  315. assertFalse(run.isItalic());
  316. assertFalse(run.isStrikeThrough());
  317. assertNull(run.getCTR().getRPr());
  318. run = p.getRuns().get(5);
  319. assertEquals("BOTH", run.toString());
  320. assertTrue(run.isBold());
  321. assertTrue(run.isItalic());
  322. assertFalse(run.isStrikeThrough());
  323. run = p.getRuns().get(6);
  324. assertEquals(", as well as ", run.toString());
  325. assertFalse(run.isBold());
  326. assertFalse(run.isItalic());
  327. assertFalse(run.isStrikeThrough());
  328. assertNull(run.getCTR().getRPr());
  329. run = p.getRuns().get(7);
  330. assertEquals("RED", run.toString());
  331. assertFalse(run.isBold());
  332. assertFalse(run.isItalic());
  333. assertFalse(run.isStrikeThrough());
  334. run = p.getRuns().get(8);
  335. assertEquals(" and ", run.toString());
  336. assertFalse(run.isBold());
  337. assertFalse(run.isItalic());
  338. assertFalse(run.isStrikeThrough());
  339. assertNull(run.getCTR().getRPr());
  340. run = p.getRuns().get(9);
  341. assertEquals("YELLOW", run.toString());
  342. assertFalse(run.isBold());
  343. assertFalse(run.isItalic());
  344. assertFalse(run.isStrikeThrough());
  345. run = p.getRuns().get(10);
  346. assertEquals(" text.", run.toString());
  347. assertFalse(run.isBold());
  348. assertFalse(run.isItalic());
  349. assertFalse(run.isStrikeThrough());
  350. assertNull(run.getCTR().getRPr());
  351. doc.close();
  352. }
  353. @Test
  354. public void testPictureInHeader() throws IOException {
  355. XWPFDocument sampleDoc = XWPFTestDataSamples.openSampleDocument("headerPic.docx");
  356. XWPFHeaderFooterPolicy policy = sampleDoc.getHeaderFooterPolicy();
  357. XWPFHeader header = policy.getDefaultHeader();
  358. int count = 0;
  359. for (XWPFParagraph p : header.getParagraphs()) {
  360. for (XWPFRun r : p.getRuns()) {
  361. List<XWPFPicture> pictures = r.getEmbeddedPictures();
  362. for (XWPFPicture pic : pictures) {
  363. assertNotNull(pic.getPictureData());
  364. assertEquals("DOZOR", pic.getDescription());
  365. }
  366. count += pictures.size();
  367. }
  368. }
  369. assertEquals(1, count);
  370. sampleDoc.close();
  371. }
  372. @Test
  373. public void testSetGetLang() {
  374. XWPFRun run = p.createRun();
  375. assertNull(run.getLang());
  376. run.setLang("en-CA");
  377. assertEquals("en-CA", run.getLang());
  378. run.setLang("fr-CA");
  379. assertEquals("fr-CA", run.getLang());
  380. run.setLang(null);
  381. assertNull(run.getLang());
  382. }
  383. @Test
  384. public void testSetGetLang2() {
  385. XWPFRun run = p.createRun();
  386. assertNull(run.getLang());
  387. run.getCTR().addNewRPr().addNewLang().setVal("en-CA");
  388. assertEquals("en-CA", run.getLang());
  389. run.getCTR().getRPr().getLang().setVal("fr-CA");
  390. assertEquals("fr-CA", run.getLang());
  391. run.getCTR().getRPr().getLang().setVal(null);
  392. assertNull(run.getLang());
  393. }
  394. @Test
  395. public void testAddPicture() throws Exception {
  396. XWPFDocument doc = XWPFTestDataSamples.openSampleDocument("TestDocument.docx");
  397. XWPFParagraph p = doc.getParagraphArray(2);
  398. XWPFRun r = p.getRuns().get(0);
  399. assertEquals(0, doc.getAllPictures().size());
  400. assertEquals(0, r.getEmbeddedPictures().size());
  401. r.addPicture(new ByteArrayInputStream(new byte[0]), Document.PICTURE_TYPE_JPEG, "test.jpg", 21, 32);
  402. assertEquals(1, doc.getAllPictures().size());
  403. assertEquals(1, r.getEmbeddedPictures().size());
  404. XWPFDocument docBack = XWPFTestDataSamples.writeOutAndReadBack(doc);
  405. XWPFParagraph pBack = docBack.getParagraphArray(2);
  406. XWPFRun rBack = pBack.getRuns().get(0);
  407. assertEquals(1, docBack.getAllPictures().size());
  408. assertEquals(1, rBack.getEmbeddedPictures().size());
  409. docBack.close();
  410. doc.close();
  411. }
  412. /**
  413. * Bugzilla #58237 - Unable to add image to word document header
  414. */
  415. @Test
  416. public void testAddPictureInHeader() throws IOException, InvalidFormatException {
  417. XWPFDocument doc = XWPFTestDataSamples.openSampleDocument("TestDocument.docx");
  418. XWPFHeader hdr = doc.createHeader(HeaderFooterType.DEFAULT);
  419. XWPFParagraph p = hdr.createParagraph();
  420. XWPFRun r = p.createRun();
  421. assertEquals(0, hdr.getAllPictures().size());
  422. assertEquals(0, r.getEmbeddedPictures().size());
  423. r.addPicture(new ByteArrayInputStream(new byte[0]), Document.PICTURE_TYPE_JPEG, "test.jpg", 21, 32);
  424. assertEquals(1, hdr.getAllPictures().size());
  425. assertEquals(1, r.getEmbeddedPictures().size());
  426. XWPFPicture pic = r.getEmbeddedPictures().get(0);
  427. CTPicture ctPic = pic.getCTPicture();
  428. CTBlipFillProperties ctBlipFill = ctPic.getBlipFill();
  429. assertNotNull(ctBlipFill);
  430. CTBlip ctBlip = ctBlipFill.getBlip();
  431. assertNotNull(ctBlip);
  432. assertEquals("rId1", ctBlip.getEmbed());
  433. XWPFDocument docBack = XWPFTestDataSamples.writeOutAndReadBack(doc);
  434. XWPFHeader hdrBack = docBack.getHeaderArray(0);
  435. XWPFParagraph pBack = hdrBack.getParagraphArray(0);
  436. XWPFRun rBack = pBack.getRuns().get(0);
  437. assertEquals(1, hdrBack.getAllPictures().size());
  438. assertEquals(1, rBack.getEmbeddedPictures().size());
  439. docBack.close();
  440. doc.close();
  441. }
  442. /**
  443. * Bugzilla #52288 - setting the font family on the
  444. * run mustn't NPE
  445. */
  446. @Test
  447. public void testSetFontFamily_52288() throws IOException {
  448. XWPFDocument doc = XWPFTestDataSamples.openSampleDocument("52288.docx");
  449. final Iterator<XWPFParagraph> paragraphs = doc.getParagraphsIterator();
  450. while (paragraphs.hasNext()) {
  451. final XWPFParagraph paragraph = paragraphs.next();
  452. for (final XWPFRun run : paragraph.getRuns()) {
  453. if (run != null) {
  454. final String text = run.getText(0);
  455. if (text != null) {
  456. run.setFontFamily("Times New Roman");
  457. }
  458. }
  459. }
  460. }
  461. doc.close();
  462. }
  463. @Test
  464. public void testBug55476() throws IOException, InvalidFormatException {
  465. byte[] image = XWPFTestDataSamples.getImage("abstract1.jpg");
  466. XWPFDocument document = new XWPFDocument();
  467. document.createParagraph().createRun().addPicture(
  468. new ByteArrayInputStream(image), Document.PICTURE_TYPE_JPEG, "test.jpg", Units.toEMU(300), Units.toEMU(100));
  469. XWPFDocument docBack = XWPFTestDataSamples.writeOutAndReadBack(document);
  470. List<XWPFPicture> pictures = docBack.getParagraphArray(0).getRuns().get(0).getEmbeddedPictures();
  471. assertEquals(1, pictures.size());
  472. docBack.close();
  473. /*OutputStream stream = new FileOutputStream("c:\\temp\\55476.docx");
  474. try {
  475. document.write(stream);
  476. } finally {
  477. stream.close();
  478. }*/
  479. document.close();
  480. }
  481. @Test
  482. public void testBug58922() throws IOException {
  483. XWPFDocument document = new XWPFDocument();
  484. final XWPFRun run = document.createParagraph().createRun();
  485. assertEquals(-1, run.getFontSize());
  486. run.setFontSize(10);
  487. assertEquals(10, run.getFontSize());
  488. run.setFontSize(Short.MAX_VALUE-1);
  489. assertEquals(Short.MAX_VALUE-1, run.getFontSize());
  490. run.setFontSize(Short.MAX_VALUE);
  491. assertEquals(Short.MAX_VALUE, run.getFontSize());
  492. run.setFontSize(Short.MAX_VALUE+1);
  493. assertEquals(Short.MAX_VALUE+1, run.getFontSize());
  494. run.setFontSize(Integer.MAX_VALUE-1);
  495. assertEquals(Integer.MAX_VALUE-1, run.getFontSize());
  496. run.setFontSize(Integer.MAX_VALUE);
  497. assertEquals(Integer.MAX_VALUE, run.getFontSize());
  498. run.setFontSize(-1);
  499. assertEquals(-1, run.getFontSize());
  500. assertEquals(-1, run.getTextPosition());
  501. run.setTextPosition(10);
  502. assertEquals(10, run.getTextPosition());
  503. run.setTextPosition(Short.MAX_VALUE-1);
  504. assertEquals(Short.MAX_VALUE-1, run.getTextPosition());
  505. run.setTextPosition(Short.MAX_VALUE);
  506. assertEquals(Short.MAX_VALUE, run.getTextPosition());
  507. run.setTextPosition(Short.MAX_VALUE+1);
  508. assertEquals(Short.MAX_VALUE+1, run.getTextPosition());
  509. run.setTextPosition(Short.MAX_VALUE+1);
  510. assertEquals(Short.MAX_VALUE+1, run.getTextPosition());
  511. run.setTextPosition(Integer.MAX_VALUE-1);
  512. assertEquals(Integer.MAX_VALUE-1, run.getTextPosition());
  513. run.setTextPosition(Integer.MAX_VALUE);
  514. assertEquals(Integer.MAX_VALUE, run.getTextPosition());
  515. run.setTextPosition(-1);
  516. assertEquals(-1, run.getTextPosition());
  517. document.close();
  518. }
  519. @Test
  520. public void testSetters() {
  521. XWPFDocument document = new XWPFDocument();
  522. final XWPFRun run = document.createParagraph().createRun();
  523. // at least trigger some of the setters to ensure classes are included in
  524. // the poi-ooxml-lite
  525. run.setBold(true);
  526. run.setCapitalized(true);
  527. run.setCharacterSpacing(2);
  528. assertEquals(2, run.getCharacterSpacing());
  529. run.setColor("000000");
  530. run.setDoubleStrikethrough(true);
  531. run.setEmbossed(true);
  532. run.setFontFamily("Calibri");
  533. assertEquals("Calibri", run.getFontFamily());
  534. run.setFontSize(10);
  535. assertEquals(10, run.getFontSize());
  536. run.setImprinted(true);
  537. run.setItalic(true);
  538. }
  539. @Test
  540. public void testSetGetTextScale() throws IOException {
  541. XWPFDocument document = new XWPFDocument();
  542. final XWPFRun run = document.createParagraph().createRun();
  543. assertEquals(100, run.getTextScale());
  544. run.setTextScale(200);
  545. assertEquals(200, run.getTextScale());
  546. document.close();
  547. }
  548. @Test
  549. public void testSetGetTextHighlightColor() throws IOException {
  550. XWPFDocument document = new XWPFDocument();
  551. final XWPFRun run = document.createParagraph().createRun();
  552. assertEquals(STHighlightColor.NONE, run.getTextHightlightColor());
  553. assertFalse(run.isHighlighted());
  554. run.setTextHighlightColor("darkGreen"); // See 17.18.40 ST_HighlightColor (Text Highlight Colors)
  555. assertEquals(STHighlightColor.DARK_GREEN, run.getTextHightlightColor());
  556. assertTrue(run.isHighlighted());
  557. run.setTextHighlightColor("none");
  558. assertFalse(run.isHighlighted());
  559. document.close();
  560. }
  561. @Test
  562. public void testSetGetVanish() throws IOException {
  563. XWPFDocument document = new XWPFDocument();
  564. final XWPFRun run = document.createParagraph().createRun();
  565. assertFalse(run.isVanish());
  566. run.setVanish(true);
  567. assertTrue(run.isVanish());
  568. run.setVanish(false);
  569. assertFalse(run.isVanish());
  570. document.close();
  571. }
  572. @Test
  573. public void testSetGetVerticalAlignment() throws IOException {
  574. XWPFDocument document = new XWPFDocument();
  575. XWPFRun run = document.createParagraph().createRun();
  576. assertEquals(STVerticalAlignRun.BASELINE, run.getVerticalAlignment());
  577. // Reset to a fresh run so we test case of run not having vertical alignment at all
  578. run = document.createParagraph().createRun();
  579. run.setVerticalAlignment("subscript");
  580. assertEquals(STVerticalAlignRun.SUBSCRIPT, run.getVerticalAlignment());
  581. run.setVerticalAlignment("superscript");
  582. assertEquals(STVerticalAlignRun.SUPERSCRIPT, run.getVerticalAlignment());
  583. document.close();
  584. }
  585. @Test
  586. public void testSetGetVAlign() {
  587. CTRPr rpr = ctRun.addNewRPr();
  588. rpr.addNewVertAlign().setVal(STVerticalAlignRun.SUBSCRIPT);
  589. XWPFRun run = new XWPFRun(ctRun, irb);
  590. run.setSubscript(VerticalAlign.BASELINE);
  591. assertEquals(STVerticalAlignRun.BASELINE, rpr.getVertAlign().getVal());
  592. }
  593. @Test
  594. public void testSetGetEmphasisMark() throws IOException {
  595. XWPFDocument document = new XWPFDocument();
  596. XWPFRun run = document.createParagraph().createRun();
  597. assertEquals(STEm.NONE, run.getEmphasisMark());
  598. // Reset to a fresh run so we test case of run not having property at all
  599. run = document.createParagraph().createRun();
  600. run.setEmphasisMark("dot");
  601. assertEquals(STEm.DOT, run.getEmphasisMark());
  602. document.close();
  603. }
  604. @Test
  605. public void testSetGetUnderlineColor() throws IOException {
  606. XWPFDocument document = new XWPFDocument();
  607. XWPFRun run = document.createParagraph().createRun();
  608. assertEquals("auto", run.getUnderlineColor());
  609. // Reset to a fresh run so we test case of run not having property at all
  610. run = document.createParagraph().createRun();
  611. String colorRgb = "C0F1a2";
  612. run.setUnderlineColor(colorRgb);
  613. assertEquals(colorRgb.toUpperCase(LocaleUtil.getUserLocale()), run.getUnderlineColor());
  614. run.setUnderlineColor("auto");
  615. assertEquals("auto", run.getUnderlineColor());
  616. document.close();
  617. }
  618. @Test
  619. public void testSetGetUnderlineThemeColor() throws IOException {
  620. XWPFDocument document = new XWPFDocument();
  621. XWPFRun run = document.createParagraph().createRun();
  622. assertEquals(STThemeColor.NONE, run.getUnderlineThemeColor());
  623. // Reset to a fresh run so we test case of run not having property at all
  624. run = document.createParagraph().createRun();
  625. String colorName = "accent4";
  626. run.setUnderlineThemeColor(colorName);
  627. assertEquals(STThemeColor.Enum.forString(colorName), run.getUnderlineThemeColor());
  628. run.setUnderlineThemeColor("none");
  629. assertEquals(STThemeColor.NONE, run.getUnderlineThemeColor());
  630. document.close();
  631. }
  632. @Test
  633. public void testSetStyleId() throws IOException {
  634. XWPFDocument document = XWPFTestDataSamples.openSampleDocument("SampleDoc.docx");
  635. final XWPFRun run = document.createParagraph().createRun();
  636. String styleId = "bolditalic";
  637. run.setStyle(styleId);
  638. String candStyleId = run.getCTR().getRPr().getRStyle().getVal();
  639. assertNotNull("Expected to find a run style ID", candStyleId);
  640. assertEquals(styleId, candStyleId);
  641. assertEquals(styleId, run.getStyle());
  642. document.close();
  643. }
  644. @Test
  645. public void testGetDepthWidth() throws IOException, InvalidFormatException {
  646. try (XWPFDocument doc = XWPFTestDataSamples.openSampleDocument("TestDocument.docx")) {
  647. XWPFHeader hdr = doc.createHeader(HeaderFooterType.DEFAULT);
  648. XWPFParagraph p = hdr.createParagraph();
  649. XWPFRun r = p.createRun();
  650. assertEquals(0, hdr.getAllPictures().size());
  651. assertEquals(0, r.getEmbeddedPictures().size());
  652. r.addPicture(new ByteArrayInputStream(new byte[0]), Document.PICTURE_TYPE_JPEG, "test.jpg", 21, 32);
  653. assertEquals(1, hdr.getAllPictures().size());
  654. assertEquals(1, r.getEmbeddedPictures().size());
  655. XWPFPicture pic = r.getEmbeddedPictures().get(0);
  656. assertEquals(pic.getWidth(), Units.toPoints(21), 0.0);
  657. assertEquals(pic.getDepth(), Units.toPoints(32), 0.0);
  658. }
  659. }
  660. @Test
  661. public void testWhitespace() throws IOException {
  662. String[] text = new String[] {
  663. " The quick brown fox",
  664. "\t\tjumped over the lazy dog"
  665. };
  666. ByteArrayOutputStream bos = new ByteArrayOutputStream();
  667. try (XWPFDocument doc = new XWPFDocument();) {
  668. for(String s : text) {
  669. XWPFParagraph p1 = doc.createParagraph();
  670. XWPFRun r1 = p1.createRun();
  671. r1.setText(s);
  672. }
  673. doc.write(bos);
  674. bos.flush();
  675. }
  676. try (
  677. ByteArrayInputStream bis = new ByteArrayInputStream(bos.toByteArray());
  678. XWPFDocument doc = new XWPFDocument(bis)
  679. ) {
  680. List<XWPFParagraph> paragraphs = doc.getParagraphs();
  681. assertEquals(2, paragraphs.size());
  682. for (int i = 0; i < text.length; i++) {
  683. XWPFParagraph p1 = paragraphs.get(i);
  684. String expected = text[i];
  685. assertEquals(expected, p1.getText());
  686. CTP ctp = p1.getCTP();
  687. CTR ctr = ctp.getRArray(0);
  688. CTText ctText = ctr.getTArray(0);
  689. // if text has leading whitespace then expect xml-fragment to have xml:space="preserve" set
  690. // <xml-fragment xml:space="preserve" xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
  691. boolean isWhitespace = Character.isWhitespace(expected.charAt(0));
  692. assertEquals(isWhitespace, ctText.isSetSpace());
  693. }
  694. }
  695. }
  696. }