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.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * This file is available under and governed by the GNU General Public
  26  * License version 2 only, as published by the Free Software Foundation.
  27  * However, the following notice accompanied the original version of this
  28  * file:
  29  *
  30  * Copyright (c) 2009-2012, Stephen Colebourne & Michael Nascimento Santos
  31  *
  32  * All rights reserved.
  33  *
  34  * Redistribution and use in source and binary forms, with or without
  35  * modification, are permitted provided that the following conditions are met:
  36  *
  37  *  * Redistributions of source code must retain the above copyright notice,
  38  *    this list of conditions and the following disclaimer.
  39  *
  40  *  * Redistributions in binary form must reproduce the above copyright notice,
  41  *    this list of conditions and the following disclaimer in the documentation
  42  *    and/or other materials provided with the distribution.
  43  *
  44  *  * Neither the name of JSR-310 nor the names of its contributors
  45  *    may be used to endorse or promote products derived from this software
  46  *    without specific prior written permission.
  47  *
  48  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  49  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  50  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  51  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  52  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  53  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  54  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  55  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  56  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  57  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  58  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  59  */
  60 package tck.java.time.zone;
  61 
  62 import java.time.ZoneId;
  63 
  64 import static org.testng.Assert.assertEquals;
  65 import static org.testng.Assert.assertNotNull;
  66 import static org.testng.Assert.assertTrue;
  67 
  68 import java.util.Collections;
  69 import java.util.HashSet;
  70 import java.util.NavigableMap;
  71 import java.util.Set;
  72 import java.util.TreeMap;
  73 
  74 import java.time.ZoneOffset;
  75 import java.time.zone.ZoneRules;
  76 import java.time.zone.ZoneRulesException;
  77 import java.time.zone.ZoneRulesProvider;
  78 
  79 import org.testng.annotations.Test;
  80 
  81 /**
  82  * Test ZoneRulesProvider.
  83  */
  84 @Test
  85 public class TCKZoneRulesProvider {
  86 
  87     private static String TZDB_VERSION = "2012i";
  88 
  89     //-----------------------------------------------------------------------
  90     // getAvailableZoneIds()
  91     //-----------------------------------------------------------------------
  92     @Test
  93     public void test_getAvailableGroupIds() {
  94         Set<String> zoneIds = ZoneRulesProvider.getAvailableZoneIds();
  95         assertEquals(zoneIds.contains("Europe/London"), true);
  96         zoneIds.clear();
  97         assertEquals(zoneIds.size(), 0);
  98         Set<String> zoneIds2 = ZoneRulesProvider.getAvailableZoneIds();
  99         assertEquals(zoneIds2.contains("Europe/London"), true);
 100     }
 101 
 102     //-----------------------------------------------------------------------
 103     // getRules(String, boolean)
 104     //-----------------------------------------------------------------------
 105     @Test
 106     public void test_getRules_StringBoolean() {
 107         ZoneRules rules = ZoneRulesProvider.getRules("Europe/London", false);
 108         assertNotNull(rules);
 109         ZoneRules rules2 = ZoneRulesProvider.getRules("Europe/London", false);
 110         assertEquals(rules2, rules);
 111     }
 112 
 113     @Test(expectedExceptions=ZoneRulesException.class)
 114     public void test_getRules_StringBoolean_unknownId() {
 115         ZoneRulesProvider.getRules("Europe/Lon", false);
 116     }
 117 
 118     @Test(expectedExceptions=NullPointerException.class)
 119     public void test_getRules_StringBoolean_null() {
 120         ZoneRulesProvider.getRules(null, false);
 121     }
 122 
 123     @Test
 124     public void test_getRules_StringBoolean_dynamic() {
 125         MockDynamicProvider dynamicProvider = new MockDynamicProvider();
 126         ZoneRulesProvider.registerProvider(dynamicProvider);
 127 
 128         assertEquals(dynamicProvider.count, 0);
 129         ZoneRules rules1 = ZoneId.of("DynamicLocation").getRules();
 130         assertEquals(dynamicProvider.count, 2);
 131         assertEquals(rules1, dynamicProvider.BASE);
 132         ZoneRules rules2 = ZoneId.of("DynamicLocation").getRules();
 133         assertEquals(dynamicProvider.count, 4);
 134         assertEquals(rules2, dynamicProvider.ALTERNATE);
 135     }
 136 
 137     //-----------------------------------------------------------------------
 138     // getVersions(String)
 139     //-----------------------------------------------------------------------
 140     @Test
 141     public void test_getVersions_String() {
 142         NavigableMap<String, ZoneRules> versions = ZoneRulesProvider.getVersions("Europe/London");
 143         assertTrue(versions.size() >= 1);
 144         ZoneRules rules = ZoneRulesProvider.getRules("Europe/London", false);
 145         assertEquals(versions.lastEntry().getValue(), rules);
 146 
 147         NavigableMap<String, ZoneRules> copy = new TreeMap<>(versions);
 148         versions.clear();
 149         assertEquals(versions.size(), 0);
 150         NavigableMap<String, ZoneRules> versions2 = ZoneRulesProvider.getVersions("Europe/London");
 151         assertEquals(versions2, copy);
 152     }
 153 
 154     @Test(expectedExceptions=ZoneRulesException.class)
 155     public void test_getVersions_String_unknownId() {
 156         ZoneRulesProvider.getVersions("Europe/Lon");
 157     }
 158 
 159     @Test(expectedExceptions=NullPointerException.class)
 160     public void test_getVersions_String_null() {
 161         ZoneRulesProvider.getVersions(null);
 162     }
 163 
 164     //-----------------------------------------------------------------------
 165     // refresh()
 166     //-----------------------------------------------------------------------
 167     @Test
 168     public void test_refresh() {
 169         assertEquals(ZoneRulesProvider.refresh(), false);
 170     }
 171 
 172     //-----------------------------------------------------------------------
 173     // registerProvider()
 174     //-----------------------------------------------------------------------
 175     @Test(groups={"tck"})
 176     public void test_registerProvider() {
 177         Set<String> pre = ZoneRulesProvider.getAvailableZoneIds();
 178         assertEquals(pre.contains("FooLocation"), false);
 179         ZoneRulesProvider.registerProvider(new MockTempProvider());
 180         assertEquals(pre.contains("FooLocation"), false);
 181         Set<String> post = ZoneRulesProvider.getAvailableZoneIds();
 182         assertEquals(post.contains("FooLocation"), true);
 183         assertEquals(ZoneRulesProvider.getRules("FooLocation", false), ZoneOffset.of("+01:45").getRules());
 184     }
 185 
 186     static class MockTempProvider extends ZoneRulesProvider {
 187         final ZoneRules rules = ZoneOffset.of("+01:45").getRules();
 188         @Override
 189         public Set<String> provideZoneIds() {
 190             return new HashSet<String>(Collections.singleton("FooLocation"));
 191         }
 192         @Override
 193         protected NavigableMap<String, ZoneRules> provideVersions(String zoneId) {
 194             NavigableMap<String, ZoneRules> result = new TreeMap<>();
 195             result.put("BarVersion", rules);
 196             return result;
 197         }
 198         @Override
 199         protected ZoneRules provideRules(String zoneId, boolean forCaching) {
 200             if (zoneId.equals("FooLocation")) {
 201                 return rules;
 202             }
 203             throw new ZoneRulesException("Invalid");
 204         }
 205     }
 206 
 207     static class MockDynamicProvider extends ZoneRulesProvider {
 208         final ZoneRules BASE = ZoneOffset.of("+01:15").getRules();
 209         final ZoneRules ALTERNATE = ZoneOffset.of("+01:30").getRules();
 210         int count = 0;
 211         @Override
 212         public Set<String> provideZoneIds() {
 213             return new HashSet<>(Collections.singleton("DynamicLocation"));
 214         }
 215         @Override
 216         protected NavigableMap<String, ZoneRules> provideVersions(String zoneId) {
 217             NavigableMap<String, ZoneRules> result = new TreeMap<>();
 218             result.put("DynamicVersion1", BASE);
 219             if (count > 2) {
 220                 result.put("DynamicVersion2", ALTERNATE);
 221             }
 222             return result;
 223         }
 224         @Override
 225         protected ZoneRules provideRules(String zoneId, boolean forCaching) {
 226             count++;
 227             if (zoneId.equals("DynamicLocation")) {
 228                 return (forCaching ? null : (count > 2 ? ALTERNATE : BASE));
 229             }
 230             throw new ZoneRulesException("Invalid");
 231         }
 232     }
 233 
 234 }