1 files changed, 15 insertions, 3 deletions

diff --git a/src/Report.js b/src/Report.js
index ac7dfac..955c97d 100644
--- a/src/Report.js
+++ b/src/Report.js
@@ -18,7 +18,7 @@
  */
 
 import React, { Component } from 'react';
-import { Counters, SessionInfo, PackagesCoverage, PackagesSourceCoverage } from './listComponents.js';
+import { Counters, SessionInfo, GroupsCoverage, PackagesCoverage, GroupsSourceCoverage, PackagesSourceCoverage } from './listComponents.js';
 
 export class Report extends Component {  
   _renderAppInfo() {
@@ -37,6 +37,18 @@ export class Report extends Component {
     );
   }
 
+  _renderCoverageTree() {
+    return this.props.report.group
+      ? (<GroupsCoverage groups={this.props.report.group} />)
+      : (<PackagesCoverage packages={this.props.report.package} />);
+  }
+
+  _renderSourceCoverage() {
+    return this.props.report.group
+      ? (<GroupsSourceCoverage groups={this.props.report.group} sourceSet={this.props.sourceSet} />)
+      : (<PackagesSourceCoverage packages={this.props.report.package} sourceSet={this.props.sourceSet} />);
+  }
+
   _renderReport() {
     return (
       <div>
@@ -54,12 +66,12 @@ export class Report extends Component {
 
         <section>
           <h3>Coverage tree</h3>
-          <PackagesCoverage packages={this.props.report.package} />
+          {this._renderCoverageTree()}
         </section>
 
         <section>
           <h3>Source coverage</h3>
-          <PackagesSourceCoverage packages={this.props.report.package} sourceSet={this.props.sourceSet} />
+          {this._renderSourceCoverage()}
         </section>
       </div>
     );