1 /*
   2  * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 /*
  27  * Copyright (c) 2012, Stephen Colebourne & Michael Nascimento Santos
  28  *
  29  * All rights reserved.
  30  *
  31  * Redistribution and use in source and binary forms, with or without
  32  * modification, are permitted provided that the following conditions are met:
  33  *
  34  *  * Redistributions of source code must retain the above copyright notice,
  35  *    this list of conditions and the following disclaimer.
  36  *
  37  *  * Redistributions in binary form must reproduce the above copyright notice,
  38  *    this list of conditions and the following disclaimer in the documentation
  39  *    and/or other materials provided with the distribution.
  40  *
  41  *  * Neither the name of JSR-310 nor the names of its contributors
  42  *    may be used to endorse or promote products derived from this software
  43  *    without specific prior written permission.
  44  *
  45  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  46  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  47  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  48  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  49  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  50  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  51  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  52  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  53  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  54  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  55  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  56  */
  57 package test.java.time.temporal;
  58 
  59 import static java.time.Month.AUGUST;
  60 import static java.time.Month.FEBRUARY;
  61 import static java.time.Month.JULY;
  62 import static java.time.Month.JUNE;
  63 import static java.time.Month.MARCH;
  64 import static java.time.Month.OCTOBER;
  65 import static java.time.Month.SEPTEMBER;
  66 import static java.time.temporal.ChronoUnit.DAYS;
  67 import static java.time.temporal.ChronoUnit.MONTHS;
  68 import static java.time.temporal.ChronoUnit.WEEKS;
  69 import static java.time.temporal.ChronoUnit.YEARS;
  70 import static org.testng.Assert.assertEquals;
  71 
  72 import java.time.LocalDate;
  73 import java.time.Month;
  74 import java.time.ZoneOffset;
  75 
  76 import org.testng.annotations.DataProvider;
  77 import org.testng.annotations.Test;
  78 
  79 /**
  80  * Test.
  81  */
  82 @Test
  83 public class TestChronoUnit {
  84 
  85     //-----------------------------------------------------------------------
  86     @DataProvider(name = "yearsBetween")
  87     Object[][] data_yearsBetween() {
  88         return new Object[][] {
  89             {date(1939, SEPTEMBER, 2), date(1939, SEPTEMBER, 1), 0},
  90             {date(1939, SEPTEMBER, 2), date(1939, SEPTEMBER, 2), 0},
  91             {date(1939, SEPTEMBER, 2), date(1939, SEPTEMBER, 3), 0},
  92 
  93             {date(1939, SEPTEMBER, 2), date(1940, SEPTEMBER, 1), 0},
  94             {date(1939, SEPTEMBER, 2), date(1940, SEPTEMBER, 2), 1},
  95             {date(1939, SEPTEMBER, 2), date(1940, SEPTEMBER, 3), 1},
  96 
  97             {date(1939, SEPTEMBER, 2), date(1938, SEPTEMBER, 1), -1},
  98             {date(1939, SEPTEMBER, 2), date(1938, SEPTEMBER, 2), -1},
  99             {date(1939, SEPTEMBER, 2), date(1938, SEPTEMBER, 3), 0},
 100 
 101             {date(1939, SEPTEMBER, 2), date(1945, SEPTEMBER, 3), 6},
 102             {date(1939, SEPTEMBER, 2), date(1945, OCTOBER, 3), 6},
 103             {date(1939, SEPTEMBER, 2), date(1945, AUGUST, 3), 5},
 104         };
 105     }
 106 
 107     @Test(dataProvider = "yearsBetween")
 108     public void test_yearsBetween(LocalDate start, LocalDate end, long expected) {
 109         assertEquals(YEARS.between(start, end), expected);
 110     }
 111 
 112     @Test(dataProvider = "yearsBetween")
 113     public void test_yearsBetweenReversed(LocalDate start, LocalDate end, long expected) {
 114         assertEquals(YEARS.between(end, start), -expected);
 115     }
 116 
 117     @Test(dataProvider = "yearsBetween")
 118     public void test_yearsBetween_LocalDateTimeSameTime(LocalDate start, LocalDate end, long expected) {
 119         assertEquals(YEARS.between(start.atTime(12, 30), end.atTime(12, 30)), expected);
 120     }
 121 
 122     @Test(dataProvider = "yearsBetween")
 123     public void test_yearsBetween_LocalDateTimeLaterTime(LocalDate start, LocalDate end, long expected) {
 124         assertEquals(YEARS.between(start.atTime(12, 30), end.atTime(12, 31)), expected);
 125     }
 126 
 127     @Test(dataProvider = "yearsBetween")
 128     public void test_yearsBetween_ZonedDateSameOffset(LocalDate start, LocalDate end, long expected) {
 129         assertEquals(YEARS.between(start.atStartOfDay(ZoneOffset.ofHours(2)), end.atStartOfDay(ZoneOffset.ofHours(2))), expected);
 130     }
 131 
 132     @Test(dataProvider = "yearsBetween")
 133     public void test_yearsBetween_ZonedDateLaterOffset(LocalDate start, LocalDate end, long expected) {
 134         // +01:00 is later than +02:00
 135         assertEquals(YEARS.between(start.atStartOfDay(ZoneOffset.ofHours(2)), end.atStartOfDay(ZoneOffset.ofHours(1))), expected);
 136     }
 137 
 138     //-----------------------------------------------------------------------
 139     @DataProvider(name = "monthsBetween")
 140     Object[][] data_monthsBetween() {
 141         return new Object[][] {
 142             {date(2012, JULY, 2), date(2012, JULY, 1), 0},
 143             {date(2012, JULY, 2), date(2012, JULY, 2), 0},
 144             {date(2012, JULY, 2), date(2012, JULY, 3), 0},
 145 
 146             {date(2012, JULY, 2), date(2012, AUGUST, 1), 0},
 147             {date(2012, JULY, 2), date(2012, AUGUST, 2), 1},
 148             {date(2012, JULY, 2), date(2012, AUGUST, 3), 1},
 149 
 150             {date(2012, JULY, 2), date(2012, SEPTEMBER, 1), 1},
 151             {date(2012, JULY, 2), date(2012, SEPTEMBER, 2), 2},
 152             {date(2012, JULY, 2), date(2012, SEPTEMBER, 3), 2},
 153 
 154             {date(2012, JULY, 2), date(2012, JUNE, 1), -1},
 155             {date(2012, JULY, 2), date(2012, JUNE, 2), -1},
 156             {date(2012, JULY, 2), date(2012, JUNE, 3), 0},
 157 
 158             {date(2012, FEBRUARY, 27), date(2012, MARCH, 26), 0},
 159             {date(2012, FEBRUARY, 27), date(2012, MARCH, 27), 1},
 160             {date(2012, FEBRUARY, 27), date(2012, MARCH, 28), 1},
 161 
 162             {date(2012, FEBRUARY, 28), date(2012, MARCH, 27), 0},
 163             {date(2012, FEBRUARY, 28), date(2012, MARCH, 28), 1},
 164             {date(2012, FEBRUARY, 28), date(2012, MARCH, 29), 1},
 165 
 166             {date(2012, FEBRUARY, 29), date(2012, MARCH, 28), 0},
 167             {date(2012, FEBRUARY, 29), date(2012, MARCH, 29), 1},
 168             {date(2012, FEBRUARY, 29), date(2012, MARCH, 30), 1},
 169         };
 170     }
 171 
 172     @Test(dataProvider = "monthsBetween")
 173     public void test_monthsBetween(LocalDate start, LocalDate end, long expected) {
 174         assertEquals(MONTHS.between(start, end), expected);
 175     }
 176 
 177     @Test(dataProvider = "monthsBetween")
 178     public void test_monthsBetweenReversed(LocalDate start, LocalDate end, long expected) {
 179         assertEquals(MONTHS.between(end, start), -expected);
 180     }
 181 
 182     @Test(dataProvider = "monthsBetween")
 183     public void test_monthsBetween_LocalDateTimeSameTime(LocalDate start, LocalDate end, long expected) {
 184         assertEquals(MONTHS.between(start.atTime(12, 30), end.atTime(12, 30)), expected);
 185     }
 186 
 187     @Test(dataProvider = "monthsBetween")
 188     public void test_monthsBetween_LocalDateTimeLaterTime(LocalDate start, LocalDate end, long expected) {
 189         assertEquals(MONTHS.between(start.atTime(12, 30), end.atTime(12, 31)), expected);
 190     }
 191 
 192     @Test(dataProvider = "monthsBetween")
 193     public void test_monthsBetween_ZonedDateSameOffset(LocalDate start, LocalDate end, long expected) {
 194         assertEquals(MONTHS.between(start.atStartOfDay(ZoneOffset.ofHours(2)), end.atStartOfDay(ZoneOffset.ofHours(2))), expected);
 195     }
 196 
 197     @Test(dataProvider = "monthsBetween")
 198     public void test_monthsBetween_ZonedDateLaterOffset(LocalDate start, LocalDate end, long expected) {
 199         // +01:00 is later than +02:00
 200         assertEquals(MONTHS.between(start.atStartOfDay(ZoneOffset.ofHours(2)), end.atStartOfDay(ZoneOffset.ofHours(1))), expected);
 201     }
 202 
 203     //-----------------------------------------------------------------------
 204     @DataProvider(name = "weeksBetween")
 205     Object[][] data_weeksBetween() {
 206         return new Object[][] {
 207             {date(2012, JULY, 2), date(2012, JUNE, 25), -1},
 208             {date(2012, JULY, 2), date(2012, JUNE, 26), 0},
 209             {date(2012, JULY, 2), date(2012, JULY, 2), 0},
 210             {date(2012, JULY, 2), date(2012, JULY, 8), 0},
 211             {date(2012, JULY, 2), date(2012, JULY, 9), 1},
 212 
 213             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 21), -1},
 214             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 22), 0},
 215             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 28), 0},
 216             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 29), 0},
 217             {date(2012, FEBRUARY, 28), date(2012, MARCH, 1), 0},
 218             {date(2012, FEBRUARY, 28), date(2012, MARCH, 5), 0},
 219             {date(2012, FEBRUARY, 28), date(2012, MARCH, 6), 1},
 220 
 221             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 22), -1},
 222             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 23), 0},
 223             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 28), 0},
 224             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 29), 0},
 225             {date(2012, FEBRUARY, 29), date(2012, MARCH, 1), 0},
 226             {date(2012, FEBRUARY, 29), date(2012, MARCH, 6), 0},
 227             {date(2012, FEBRUARY, 29), date(2012, MARCH, 7), 1},
 228         };
 229     }
 230 
 231     @Test(dataProvider = "weeksBetween")
 232     public void test_weeksBetween(LocalDate start, LocalDate end, long expected) {
 233         assertEquals(WEEKS.between(start, end), expected);
 234     }
 235 
 236     @Test(dataProvider = "weeksBetween")
 237     public void test_weeksBetweenReversed(LocalDate start, LocalDate end, long expected) {
 238         assertEquals(WEEKS.between(end, start), -expected);
 239     }
 240 
 241     //-----------------------------------------------------------------------
 242     @DataProvider(name = "daysBetween")
 243     Object[][] data_daysBetween() {
 244         return new Object[][] {
 245             {date(2012, JULY, 2), date(2012, JULY, 1), -1},
 246             {date(2012, JULY, 2), date(2012, JULY, 2), 0},
 247             {date(2012, JULY, 2), date(2012, JULY, 3), 1},
 248 
 249             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 27), -1},
 250             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 28), 0},
 251             {date(2012, FEBRUARY, 28), date(2012, FEBRUARY, 29), 1},
 252             {date(2012, FEBRUARY, 28), date(2012, MARCH, 1), 2},
 253 
 254             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 27), -2},
 255             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 28), -1},
 256             {date(2012, FEBRUARY, 29), date(2012, FEBRUARY, 29), 0},
 257             {date(2012, FEBRUARY, 29), date(2012, MARCH, 1), 1},
 258 
 259             {date(2012, MARCH, 1), date(2012, FEBRUARY, 27), -3},
 260             {date(2012, MARCH, 1), date(2012, FEBRUARY, 28), -2},
 261             {date(2012, MARCH, 1), date(2012, FEBRUARY, 29), -1},
 262             {date(2012, MARCH, 1), date(2012, MARCH, 1), 0},
 263             {date(2012, MARCH, 1), date(2012, MARCH, 2), 1},
 264 
 265             {date(2012, MARCH, 1), date(2013, FEBRUARY, 28), 364},
 266             {date(2012, MARCH, 1), date(2013, MARCH, 1), 365},
 267 
 268             {date(2011, MARCH, 1), date(2012, FEBRUARY, 28), 364},
 269             {date(2011, MARCH, 1), date(2012, FEBRUARY, 29), 365},
 270             {date(2011, MARCH, 1), date(2012, MARCH, 1), 366},
 271         };
 272     }
 273 
 274     @Test(dataProvider = "daysBetween")
 275     public void test_daysBetween(LocalDate start, LocalDate end, long expected) {
 276         assertEquals(DAYS.between(start, end), expected);
 277     }
 278 
 279     @Test(dataProvider = "daysBetween")
 280     public void test_daysBetweenReversed(LocalDate start, LocalDate end, long expected) {
 281         assertEquals(DAYS.between(end, start), -expected);
 282     }
 283 
 284     @Test(dataProvider = "daysBetween")
 285     public void test_daysBetween_LocalDateTimeSameTime(LocalDate start, LocalDate end, long expected) {
 286         assertEquals(DAYS.between(start.atTime(12, 30), end.atTime(12, 30)), expected);
 287     }
 288 
 289     @Test(dataProvider = "daysBetween")
 290     public void test_daysBetween_LocalDateTimeLaterTime(LocalDate start, LocalDate end, long expected) {
 291         assertEquals(DAYS.between(start.atTime(12, 30), end.atTime(12, 31)), expected);
 292     }
 293 
 294     @Test(dataProvider = "daysBetween")
 295     public void test_daysBetween_ZonedDateSameOffset(LocalDate start, LocalDate end, long expected) {
 296         assertEquals(DAYS.between(start.atStartOfDay(ZoneOffset.ofHours(2)), end.atStartOfDay(ZoneOffset.ofHours(2))), expected);
 297     }
 298 
 299     @Test(dataProvider = "daysBetween")
 300     public void test_daysBetween_ZonedDateLaterOffset(LocalDate start, LocalDate end, long expected) {
 301         // +01:00 is later than +02:00
 302         assertEquals(DAYS.between(start.atStartOfDay(ZoneOffset.ofHours(2)), end.atStartOfDay(ZoneOffset.ofHours(1))), expected);
 303     }
 304 
 305     //-----------------------------------------------------------------------
 306     private static LocalDate date(int year, Month month, int dom) {
 307         return LocalDate.of(year, month, dom);
 308     }
 309 
 310 }