Allow overriding of ZoneId. Fixes #65 (#67)

appmattus · Sep 21, 2020 · aadc581 · aadc581
1 parent 0c89d59
commit aadc581
Show file tree

Hide file tree

Showing 6 changed files with 52 additions and 9 deletions.
diff --git a/fixture/src/main/kotlin/com/appmattus/kotlinfixture/resolver/JodaTimeResolver.kt b/fixture/src/main/kotlin/com/appmattus/kotlinfixture/resolver/JodaTimeResolver.kt
@@ -37,10 +37,10 @@ internal class JodaTimeResolver : Resolver {
         return if (hasJodaTime) {
             when (obj) {
                 Instant::class -> context.generateInstant()
-                LocalDate::class -> LocalDate(context.generateInstant(), context.randomDateTimeZone())
-                LocalTime::class -> LocalTime(context.generateInstant(), context.randomDateTimeZone())
-                LocalDateTime::class -> LocalDateTime(context.generateInstant(), context.randomDateTimeZone())
-                DateTime::class -> DateTime(context.generateInstant(), context.randomDateTimeZone())
+                LocalDate::class -> LocalDate(context.generateInstant(), context.generateDateTimeZone())
+                LocalTime::class -> LocalTime(context.generateInstant(), context.generateDateTimeZone())
+                LocalDateTime::class -> LocalDateTime(context.generateInstant(), context.generateDateTimeZone())
+                DateTime::class -> DateTime(context.generateInstant(), context.generateDateTimeZone())
                 Period::class -> context.generatePeriod()
                 Duration::class -> context.generateDuration()
                 DateTimeZone::class -> context.randomDateTimeZone()
@@ -54,6 +54,8 @@ internal class JodaTimeResolver : Resolver {
 
     private fun Context.generateInstant(): Instant = Instant(resolve(typeOf<Date>()) as Date)
 
+    private fun Context.generateDateTimeZone(): DateTimeZone = resolve(typeOf<DateTimeZone>()) as DateTimeZone
+
     private fun Context.randomDateTimeZone(): DateTimeZone =
         DateTimeZone.forID(DateTimeZone.getAvailableIDs().random(random))
 

diff --git a/fixture/src/main/kotlin/com/appmattus/kotlinfixture/resolver/ThreeTenResolver.kt b/fixture/src/main/kotlin/com/appmattus/kotlinfixture/resolver/ThreeTenResolver.kt
@@ -78,7 +78,8 @@ internal class ThreeTenResolver : Resolver {
 
     private fun Context.generateInstant(): Instant = DateTimeUtils.toInstant(resolve(typeOf<Date>()) as Date)
 
-    private fun Context.generateZonedDateTime(): ZonedDateTime = generateInstant().atZone(randomZoneId())
+    private fun Context.generateZonedDateTime(): ZonedDateTime =
+        generateInstant().atZone(resolve(typeOf<ZoneId>()) as ZoneId)
 
     private fun Context.randomZoneId(): ZoneId = ZoneId.of(ZoneId.getAvailableZoneIds().random(random))
 

diff --git a/fixture/src/main/kotlin/com/appmattus/kotlinfixture/resolver/TimeResolver.kt b/fixture/src/main/kotlin/com/appmattus/kotlinfixture/resolver/TimeResolver.kt
@@ -79,7 +79,8 @@ internal class TimeResolver : Resolver {
 
     private fun Context.generateInstant(): Instant = (resolve(typeOf<Date>()) as Date).toInstant()
 
-    private fun Context.generateZonedDateTime(): ZonedDateTime = generateInstant().atZone(randomZoneId())
+    private fun Context.generateZonedDateTime(): ZonedDateTime =
+        generateInstant().atZone(resolve(typeOf<ZoneId>()) as ZoneId)
 
     private fun Context.randomZoneId(): ZoneId = ZoneId.of(ZoneId.getAvailableZoneIds().random(random))
 

diff --git a/fixture/src/test/kotlin/com/appmattus/kotlinfixture/resolver/JodaTimeResolverTest.kt b/fixture/src/test/kotlin/com/appmattus/kotlinfixture/resolver/JodaTimeResolverTest.kt
@@ -22,6 +22,7 @@ import com.appmattus.kotlinfixture.assertIsRandom
 import com.appmattus.kotlinfixture.config.Configuration
 import com.appmattus.kotlinfixture.config.ConfigurationBuilder
 import com.appmattus.kotlinfixture.config.before
+import com.appmattus.kotlinfixture.kotlinFixture
 import org.joda.time.DateTime
 import org.joda.time.DateTimeZone
 import org.joda.time.Duration
@@ -37,6 +38,7 @@ import org.junit.runners.Parameterized
 import java.util.Date
 import kotlin.reflect.KClass
 import kotlin.test.Test
+import kotlin.test.assertEquals
 import kotlin.test.assertNotNull
 import kotlin.test.assertTrue
 
@@ -48,7 +50,7 @@ class JodaTimeResolverTest {
 
         private val context = TestContext(
             ConfigurationBuilder().apply { factory<Date> { before(now) } }.build(),
-            CompositeResolver(JodaTimeResolver(), FactoryResolver())
+            CompositeResolver(JodaTimeResolver(), FactoryResolver(), KTypeResolver())
         )
 
         @Test
@@ -68,6 +70,17 @@ class JodaTimeResolverTest {
                 }
             }
         }
+
+        @Test
+        fun `Can override DateTimeZone when creating DateTime`() {
+            repeat(100) {
+                val dateTime = kotlinFixture {
+                    factory<DateTimeZone> { DateTimeZone.forID("Europe/London") }
+                } <DateTime>()
+
+                assertEquals("Europe/London", dateTime.zone.id)
+            }
+        }
     }
 
     @RunWith(Parameterized::class)

diff --git a/fixture/src/test/kotlin/com/appmattus/kotlinfixture/resolver/ThreeTenResolverTest.kt b/fixture/src/test/kotlin/com/appmattus/kotlinfixture/resolver/ThreeTenResolverTest.kt
@@ -22,6 +22,7 @@ import com.appmattus.kotlinfixture.assertIsRandom
 import com.appmattus.kotlinfixture.config.Configuration
 import com.appmattus.kotlinfixture.config.ConfigurationBuilder
 import com.appmattus.kotlinfixture.config.before
+import com.appmattus.kotlinfixture.kotlinFixture
 import org.junit.experimental.runners.Enclosed
 import org.junit.runner.RunWith
 import org.junit.runners.Parameterized
@@ -44,6 +45,7 @@ import org.threeten.bp.ZonedDateTime
 import java.util.Date
 import kotlin.reflect.KClass
 import kotlin.test.Test
+import kotlin.test.assertEquals
 import kotlin.test.assertNotNull
 import kotlin.test.assertTrue
 
@@ -55,7 +57,7 @@ class ThreeTenResolverTest {
 
         private val context = TestContext(
             ConfigurationBuilder().apply { factory<Date> { before(now) } }.build(),
-            CompositeResolver(ThreeTenResolver(), FactoryResolver())
+            CompositeResolver(ThreeTenResolver(), FactoryResolver(), KTypeResolver())
         )
 
         @Test
@@ -75,6 +77,17 @@ class ThreeTenResolverTest {
                 }
             }
         }
+
+        @Test
+        fun `Can override ZoneId when creating ZonedDateTime`() {
+            repeat(100) {
+                val zonedDateTime = kotlinFixture {
+                    factory<ZoneId> { ZoneId.of("Europe/London") }
+                } <ZonedDateTime>()
+
+                assertEquals("Europe/London", zonedDateTime.zone.id)
+            }
+        }
     }
 
     @RunWith(Parameterized::class)

diff --git a/fixture/src/test/kotlin/com/appmattus/kotlinfixture/resolver/TimeResolverTest.kt b/fixture/src/test/kotlin/com/appmattus/kotlinfixture/resolver/TimeResolverTest.kt
@@ -22,6 +22,7 @@ import com.appmattus.kotlinfixture.assertIsRandom
 import com.appmattus.kotlinfixture.config.Configuration
 import com.appmattus.kotlinfixture.config.ConfigurationBuilder
 import com.appmattus.kotlinfixture.config.before
+import com.appmattus.kotlinfixture.kotlinFixture
 import org.junit.experimental.runners.Enclosed
 import org.junit.runner.RunWith
 import org.junit.runners.Parameterized
@@ -44,6 +45,7 @@ import java.util.Date
 import java.util.TimeZone
 import kotlin.reflect.KClass
 import kotlin.test.Test
+import kotlin.test.assertEquals
 import kotlin.test.assertNotNull
 import kotlin.test.assertTrue
 
@@ -55,7 +57,7 @@ class TimeResolverTest {
 
         private val context = TestContext(
             ConfigurationBuilder().apply { factory<Date> { before(now) } }.build(),
-            CompositeResolver(TimeResolver(), FactoryResolver())
+            CompositeResolver(TimeResolver(), FactoryResolver(), KTypeResolver())
         )
 
         @Test
@@ -75,6 +77,17 @@ class TimeResolverTest {
                 }
             }
         }
+
+        @Test
+        fun `Can override ZoneId when creating ZonedDateTime`() {
+            repeat(100) {
+                val zonedDateTime = kotlinFixture {
+                    factory<ZoneId> { ZoneId.of("Europe/London") }
+                } <ZonedDateTime>()
+
+                assertEquals("Europe/London", zonedDateTime.zone.id)
+            }
+        }
     }
 
     @RunWith(Parameterized::class)