黑马程序员技术交流社区

标题: 一个交通灯枚举,包含红灯、绿灯、黄灯 [打印本页]

作者: tfy    时间: 2012-12-2 12:41
标题: 一个交通灯枚举,包含红灯、绿灯、黄灯


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));  
            }  
        }  







欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/) 黑马程序员IT技术论坛 X3.2