package com.itheima;

import static org.junit.Assert.assertEquals;  

import org.junit.Test;  

/**
* 第三题  3、 定义一个交通灯枚举，包含红灯、绿灯、黄灯，需要有获得下一个灯的方法，
* 例如：红灯获取下一个灯是绿灯，绿灯获取下一个灯是黄灯。
* @author Administrator
*
*/



public class Test3 {
        public enum TrafficLight{  
            RED,GREEN,YELLOW;  
             
            public static TrafficLight getNextLight(TrafficLight currentLight){  
                switch(currentLight){  
                case RED:  
                    return GREEN;  
                case GREEN:  
                    return YELLOW;  
                case YELLOW:  
                    return RED;  
                }  
                return null;  
            }  
        }  

            @Test  
            public void test1(){  
                assertEquals(TrafficLight.GREEN,TrafficLight.getNextLight(TrafficLight.RED));  
                assertEquals(TrafficLight.YELLOW,TrafficLight.getNextLight(TrafficLight.GREEN));  
                assertEquals(TrafficLight.RED,TrafficLight.getNextLight(TrafficLight.YELLOW));  
            }  
        }  

package com.itheima;

import static org.junit.Assert.assertEquals;  

import org.junit.Test;  

/**
* 第三题  3、 定义一个交通灯枚举，包含红灯、绿灯、黄灯，需要有获得下一个灯的方法，
* 例如：红灯获取下一个灯是绿灯，绿灯获取下一个灯是黄灯。
* @author Administrator
*
*/



public class Test3 {
        public enum TrafficLight{  
            RED,GREEN,YELLOW;  
             
            public static TrafficLight getNextLight(TrafficLight currentLight){  
                switch(currentLight){  
                case RED:  
                    return GREEN;  
                case GREEN:  
                    return YELLOW;  
                case YELLOW:  
                    return RED;  
                }  
                return null;  
            }  
        }  

            @Test  
            public void test1(){  
                assertEquals(TrafficLight.GREEN,TrafficLight.getNextLight(TrafficLight.RED));  
                assertEquals(TrafficLight.YELLOW,TrafficLight.getNextLight(TrafficLight.GREEN));  
                assertEquals(TrafficLight.RED,TrafficLight.getNextLight(TrafficLight.YELLOW));  
            }  
        }