PE-3136 Fix it-test.

hmrc · Jul 14, 2017 · 92fac1c · 92fac1c
1 parent d39e5e0
commit 92fac1c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/test/it/UserInfoServiceSpec.scala b/test/it/UserInfoServiceSpec.scala
@@ -17,15 +17,20 @@
 package it
 
 import org.joda.time.LocalDate
+import org.scalatest.{BeforeAndAfterAll, BeforeAndAfterEach}
 import play.api.libs.json.Json
 import uk.gov.hmrc.domain.Nino
+import uk.gov.hmrc.openidconnect.userinfo.config.{FeatureSwitch, UserInfoFeatureSwitches}
 import uk.gov.hmrc.openidconnect.userinfo.domain._
 import uk.gov.hmrc.play.auth.microservice.connectors.ConfidenceLevel
 import uk.gov.hmrc.play.http.Token
 
 import scalaj.http.Http
 
-class UserInfoServiceSpec extends BaseFeatureSpec {
+class UserInfoServiceSpec extends BaseFeatureSpec with BeforeAndAfterAll {
+
+  override protected def beforeAll() = {FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)}
+  override protected def afterAll() = {FeatureSwitch.disable(UserInfoFeatureSwitches.countryCode)}
 
   val authBearerToken = "AUTH_BEARER_TOKEN"
   val nino = "AB123456A"

diff --git a/test/unit/uk/gov/hmrc/openidconnect/userinfo/data/UserInfoGeneratorSpec.scala b/test/unit/uk/gov/hmrc/openidconnect/userinfo/data/UserInfoGeneratorSpec.scala
@@ -35,7 +35,7 @@ class UserInfoGeneratorSpec extends UnitSpec with PropertyChecks with BeforeAndA
   }
 
   override protected def afterEach(): Unit = {
-      FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)
+      FeatureSwitch.disable(UserInfoFeatureSwitches.countryCode)
   }
 
   "userInfo" should {

diff --git a/test/unit/uk/gov/hmrc/openidconnect/userinfo/services/UserInfoTransformerSpec.scala b/test/unit/uk/gov/hmrc/openidconnect/userinfo/services/UserInfoTransformerSpec.scala
@@ -67,7 +67,7 @@ class UserInfoTransformerSpec extends UnitSpec with MockitoSugar with BeforeAndA
   }
 
   override protected def afterEach() = {
-    FeatureSwitch.enable(UserInfoFeatureSwitches.countryCode)
+    FeatureSwitch.disable(UserInfoFeatureSwitches.countryCode)
   }
 
   trait Setup {