C语言枚举返回函数

#include "stdio.h"

typedef enum 
{
	A=1,
	B=2,
	C=3,
}TEST_ENUM;

TEST_ENUM fun(int i)
{
	if(i==1)
	{
		return A;
	}
		if(i==2)
	{
		return B;
	}
		if(i==3)
	{
		return C;
	}
	
	
}

void main()
{
	
	printf("test:%d",fun(2));
	while(1);
}

该函数只能返回枚举定义的内容，否则会出现错误。本次输出值为 ：2