Client（客戶端）
public static void main(String[] args)
{
Component leaf1 = new Leaf();
Component leaf2 = new Leaf();
Composite comp1 = new Composite();
comp1.add(leaf1);
comp1.add(leaf2);
Component leaf3 = new Leaf();
Component leaf4 = new Leaf();
Composite comp2 = new Composite();
comp2.add(comp1);
comp2.add(leaf3);
comp2.add(leaf4);
comp2.doSomething();
}
這樣對組合模式基本分析完成，繼續接著看下在Junit組合模式是樣實現的呢?
在Junit中有連個重要的概念，一個是TestCase一個是TestSuite；TestSuite是一個測試集合，一個TestSuite中可以包含一個或者多個TestSuite，當然也可以一個或者多個TestCase，而這兩個對象都繼承與Test接口。這樣一分析，Test接口相當于組合模式中的抽象構件角色（Component），TestSuite相當于樹枝構件角色（Composite），TestCase相當于樹葉構件角色（Leaf）。接著我們對具體代碼分析看看這塊是否滿足組合模式。
Test接口類：
public interface Test {
public abstract int countTestCases();
public abstract void run(TestResult result);
}
TestSuite實現：
public int countTestCases() {
int count= 0;
for (Enumeration e= tests(); e.hasMoreElements(); ) {
Test test= (Test)e.nextElement();
count= count + test.countTestCases();
}
return count;
}
public void run(TestResult result) {
for (Enumeration e= tests(); e.hasMoreElements(); ) {
if (result.shouldStop() )
break;
Test test= (Test)e.nextElement();
runTest(test， result);
}
}
TestCase實現
public int countTestCases() {
return 1;
}
public void run(TestResult result) {
result.run(this);
}
根據代碼分析，Junit使用Composite模式后簡化了JUnit的代碼，JUnit可以統一處理組合結構TestSuite和單個對象TestCase。使JUnit開發變得簡單容易，因為不需要區分部分和整體的區別，不需要寫一些充斥著if else的選擇語句。
另外，通過對TestCase對象和TestSuite的類層次結構基本對象的定義，TestCase可以被組合成更復雜的組合對象TestSuite，而這些組合對象又可以被組合，這樣不斷地遞歸下去。在程序的代碼中，任何使用基本對象的地方都可方便的使用組合對象，大大簡化系統維護和開發。
