PE-3136 Fix test leakage.

hmrc · Jul 14, 2017 · d39e5e0 · d39e5e0
1 parent 90c075a
commit d39e5e0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/test/unit/uk/gov/hmrc/openidconnect/userinfo/data/UserInfoGeneratorSpec.scala b/test/unit/uk/gov/hmrc/openidconnect/userinfo/data/UserInfoGeneratorSpec.scala
@@ -34,6 +34,10 @@ class UserInfoGeneratorSpec extends UnitSpec with PropertyChecks with BeforeAndA
       FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)
   }
 
+  override protected def afterEach(): Unit = {
+      FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)
+  }
+
   "userInfo" should {
     "generate an OpenID Connect compliant UserInfo response" in forAll(UserInfoGenerator.userInfo) { userInfo: UserInfo =>
       UserInfoGenerator.firstNames should contain (userInfo.given_name)

diff --git a/test/unit/uk/gov/hmrc/openidconnect/userinfo/services/UserInfoTransformerSpec.scala b/test/unit/uk/gov/hmrc/openidconnect/userinfo/services/UserInfoTransformerSpec.scala
@@ -66,6 +66,10 @@ class UserInfoTransformerSpec extends UnitSpec with MockitoSugar with BeforeAndA
     FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)
   }
 
+  override protected def afterEach() = {
+    FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)
+  }
+
   trait Setup {
     implicit val hc: HeaderCarrier = HeaderCarrier()