package com.grayben.riskExtractor.htmlScorer.scorers.elementScorers;

import com.grayben.riskExtractor.htmlScorer.scorers.Scorer;
import com.grayben.riskExtractor.htmlScorer.scorers.ScorerTest;
import org.jsoup.nodes.Element;
import org.jsoup.parser.Tag;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.rules.ExpectedException;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

/**
 * Created by beng on 28/11/2015.
 */
@RunWith(MockitoJUnitRunner.class)
public class SegmentationElementScorerTest
        implements ScorerTest {

    SegmentationElementScorer elementScorerSUT;
    @Mock
    Scorer<Tag> tagScorerMock;
    @Mock
    Element elementMock;

    @Rule
    ExpectedException thrown = ExpectedException.none();

    @Before
    public void setUp() throws Exception {
        elementScorerSUT = new SegmentationElementScorer(tagScorerMock);
    }

    @After
    public void tearDown() throws Exception {
        elementScorerSUT = null;
    }

}
@@ -301,73 +301,8 @@
 re;%0A
-import org.junit.Rule;%0Aimport org.junit.rules.ExpectedException;%0A
 impo
@@ -535,17 +535,14 @@
     
-implement
+extend
 s Sc
@@ -549,16 +549,25 @@
 orerTest
+%3CElement%3E
  %7B%0A%0A    
@@ -691,76 +691,8 @@
 k;%0A%0A
-    @Rule%0A    ExpectedException thrown = ExpectedException.none();%0A%0A
     
@@ -815,16 +815,55 @@
 rMock);%0A
+        super.setUp(elementScorerSUT);%0A
     %7D%0A%0A 

Align test with new base class