33import static org .assertj .core .api .Assertions .assertThat ;
44
55import com .hubspot .jinjava .features .DateTimeFeatureActivationStrategy ;
6- import com .hubspot .jinjava .features .DelegatingFeatureActivationStrategy ;
76import com .hubspot .jinjava .features .FeatureConfig ;
87import com .hubspot .jinjava .features .FeatureStrategies ;
98import com .hubspot .jinjava .features .Features ;
9+ import com .hubspot .jinjava .interpret .Context ;
1010import java .time .LocalDateTime ;
1111import org .junit .Before ;
1212import org .junit .Test ;
@@ -22,6 +22,8 @@ public class FeaturesTest {
2222
2323 private boolean delegateActive = false ;
2424
25+ private Context context = new Context ();
26+
2527 @ Before
2628 public void setUp () throws Exception {
2729 features =
@@ -32,41 +34,41 @@ public void setUp() throws Exception {
3234 .add (ALWAYS_ON , FeatureStrategies .ACTIVE )
3335 .add (DATE_PAST , DateTimeFeatureActivationStrategy .of (LocalDateTime .MIN ))
3436 .add (DATE_FUTURE , DateTimeFeatureActivationStrategy .of (LocalDateTime .MAX ))
35- .add (DELEGATING , DelegatingFeatureActivationStrategy . of (() -> delegateActive ) )
37+ .add (DELEGATING , d -> delegateActive )
3638 .build ()
3739 );
3840 }
3941
4042 @ Test
4143 public void itHasEnabledFeature () {
42- assertThat (features .isActive (ALWAYS_ON )).isTrue ();
44+ assertThat (features .isActive (ALWAYS_ON , context )).isTrue ();
4345 }
4446
4547 @ Test
4648 public void itDoesNotHaveDisabledFeature () {
47- assertThat (features .isActive (ALWAYS_OFF )).isFalse ();
49+ assertThat (features .isActive (ALWAYS_OFF , context )).isFalse ();
4850 }
4951
5052 @ Test
5153 public void itHasPastEnabledFeature () {
52- assertThat (features .isActive (DATE_PAST )).isTrue ();
54+ assertThat (features .isActive (DATE_PAST , context )).isTrue ();
5355 }
5456
5557 @ Test
5658 public void itDoesNotHaveFutureEnabledFeature () {
57- assertThat (features .isActive (DATE_FUTURE )).isFalse ();
59+ assertThat (features .isActive (DATE_FUTURE , context )).isFalse ();
5860 }
5961
6062 @ Test
6163 public void itUsesDelegate () {
6264 delegateActive = false ;
63- assertThat (features .isActive (DELEGATING )).isEqualTo (delegateActive );
65+ assertThat (features .isActive (DELEGATING , context )).isEqualTo (delegateActive );
6466 delegateActive = true ;
65- assertThat (features .isActive (DELEGATING )).isEqualTo (delegateActive );
67+ assertThat (features .isActive (DELEGATING , context )).isEqualTo (delegateActive );
6668 }
6769
6870 @ Test
6971 public void itDefaultsToFalse () {
70- assertThat (features .isActive ("unknown" )).isFalse ();
72+ assertThat (features .isActive ("unknown" , context )).isFalse ();
7173 }
7274}
0 commit comments