Tests for updated RecordContainer

git-svn-id: https://svn.apache.org/repos/asf/jakarta/poi/trunk@353791 13f79535-47bb-0310-9956-ffa450edef68

src/scratchpad/testcases/org/apache/poi/hslf/record/TestRecordContainer.java

@@ -0,0 +1,111 @@
+
+/* ====================================================================
+   Copyright 2002-2004   Apache Software Foundation
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================== */
+        
+
+
+package org.apache.poi.hslf.record;
+
+
+import junit.framework.TestCase;
+
+import org.apache.poi.hslf.HSLFSlideShow;
+
+/**
+ * Tests that the helper methods on RecordContainer work properly
+ *
+ * @author Nick Burch (nick at torchbox dot com)
+ */
+public class TestRecordContainer extends TestCase {
+	private RecordContainer recordContainer;
+
+	public void testIsAnAtom() {
+		assertFalse( recordContainer.isAnAtom() );
+	}
+	
+	public void testAppendChildRecord() {
+		// Start with an empty set
+		Record[] rs = new Record[0];
+		Record r = recordContainer.getChildRecords()[0];
+		Record[] nrs = recordContainer.appendChildRecord(r, rs);
+		
+		assertEquals(1, nrs.length);
+		assertEquals(r, nrs[0]);
+		
+		// Now start with one with 3 entries
+		rs = new Record[3];
+		Record rb = recordContainer.getChildRecords()[1];
+		Record rc = recordContainer.getChildRecords()[2];
+		Record rd = recordContainer.getChildRecords()[3];
+		rs[0] = rb;
+		rs[1] = rc;
+		rs[2] = rd;
+		
+		nrs = recordContainer.appendChildRecord(r, rs);
+		
+		assertEquals(4, nrs.length);
+		assertEquals(rb, nrs[0]);
+		assertEquals(rc, nrs[1]);
+		assertEquals(rd, nrs[2]);
+		assertEquals(r, nrs[3]);
+	}
+	
+	public void testAddChildAfter() {
+		// Working with new StyleTextPropAtom
+		Record newRecord = new StyleTextPropAtom(0);
+		
+		// Try to add after a mid-record
+		Record[] cr = recordContainer.getChildRecords();
+		Record after = cr[2];
+		Record before = cr[3];
+		
+		recordContainer.addChildAfter(newRecord, after);
+		Record[] ncr = recordContainer.getChildRecords();
+		
+		assertEquals(cr.length+1, ncr.length);
+		assertEquals(after, ncr[2]);
+		assertEquals(newRecord, ncr[3]);
+		assertEquals(before, ncr[4]);
+		
+		// Try again at the end
+		recordContainer._children = cr;
+		after = cr[cr.length-1];
+		
+		recordContainer.addChildAfter(newRecord, after);
+		ncr = recordContainer.getChildRecords();
+		
+		assertEquals(cr.length+1, ncr.length);
+		assertEquals(after, ncr[cr.length-1]);
+		assertEquals(newRecord, ncr[cr.length]);
+	}
+	
+	protected void setUp() throws Exception {
+		super.setUp();
+		
+		// Find a real RecordContainer record
+		String dirname = System.getProperty("HSLF.testdata.path");
+		String filename = dirname + "/basic_test_ppt_file.ppt";
+		HSLFSlideShow hss = new HSLFSlideShow(filename);
+		
+		Record[] r = hss.getRecords();
+		for(int i=0; i<r.length; i++) {
+			if(r[i] instanceof RecordContainer) {
+				recordContainer = (RecordContainer)r[i];
+				return;
+			}
+		}
+	}
+}