87 lines
4.0 KiB
C#
87 lines
4.0 KiB
C#
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
using System.Data;
|
|
using System.Data.Common;
|
|
using System.Linq;
|
|
|
|
namespace Bootstrap.DataAccess.Tests
|
|
{
|
|
[TestClass]
|
|
public class MenuHelperTests
|
|
{
|
|
private Menu Menu { get; set; }
|
|
private Role Role { get; set; }
|
|
|
|
[TestInitialize]
|
|
public void Initialized()
|
|
{
|
|
Menu = new Menu() { Name = "__测试菜单__", Order = 999, Category = "0" };
|
|
Role = new Role() { RoleName = "_测试角色_", Description = "这是一个测试角色", Checked = "0" };
|
|
}
|
|
[TestCleanup]
|
|
public void CleanUp()
|
|
{
|
|
using (DbCommand cmd = DBAccessManager.SqlDBAccess.CreateCommand(CommandType.Text, "Delete from Navigations where Name = '__测试菜单__'"))
|
|
{
|
|
DBAccessManager.SqlDBAccess.ExecuteNonQuery(cmd);
|
|
}
|
|
}
|
|
|
|
[TestMethod]
|
|
public void RetrieveMenusTest()
|
|
{
|
|
Assert.IsTrue(MenuHelper.RetrieveMenus().Count() > 1, "不带参数的MenuHelper.RetrieveMenus方法调用失败");
|
|
}
|
|
|
|
[TestMethod]
|
|
public void SaveMenuTest()
|
|
{
|
|
// 测试插入菜单方法 ID = 0
|
|
Assert.IsTrue(MenuHelper.SaveMenu(Menu), "插入菜单操作失败，请检查 MenuHelper.SaveMenu 方法");
|
|
var menus = MenuHelper.RetrieveMenus();
|
|
Assert.IsTrue(menus.Count() > 0, "插入菜单操作失败，请检查 MenuHelper.SaveMenu 方法");
|
|
|
|
// 测试更新菜单方法 ID != 0
|
|
var menu = menus.FirstOrDefault(m => m.Name == Menu.Name);
|
|
menu.Icon = "fa";
|
|
Assert.IsTrue(MenuHelper.SaveMenu(menu), string.Format("更新菜单ID = {0} 操作失败，请检查 MenuHelper.SaveMenu 方法", menu.ID));
|
|
var dest = MenuHelper.RetrieveMenus(menu.ID.ToString());
|
|
Assert.IsTrue(dest.Count() == 1, "带参数的MenuHelper.RetrieveMenus方法调用失败");
|
|
Assert.AreEqual(menu.Icon, dest.First().Icon, string.Format("更新菜单ID = {0} 操作失败，请检查 MenuHelper.SaveMenu 方法", menu.ID));
|
|
}
|
|
|
|
[TestMethod]
|
|
public void DeleteMenuTest()
|
|
{
|
|
// 先判断数据环境是否可以删除，没有数据先伪造数据
|
|
var menu = MenuHelper.RetrieveMenus().FirstOrDefault(m => m.Name == Menu.Name);
|
|
if (menu == null) MenuHelper.SaveMenu(Menu);
|
|
menu = MenuHelper.RetrieveMenus().FirstOrDefault(m => m.Name == Menu.Name);
|
|
Assert.IsTrue(MenuHelper.DeleteMenu(menu.ID.ToString()), "MenuHelper.DeleteMenu 方法调用失败");
|
|
}
|
|
|
|
[TestMethod]
|
|
public void SavaRolesByMenuIdTest()
|
|
{
|
|
var menu = MenuHelper.RetrieveMenus().FirstOrDefault(m => m.Name == Menu.Name);
|
|
if (menu == null) MenuHelper.SaveMenu(Menu);
|
|
menu = MenuHelper.RetrieveMenus().FirstOrDefault(m => m.Name == Menu.Name);
|
|
|
|
var role = RoleHelper.RetrieveRoles().FirstOrDefault(m => m.RoleName == Role.RoleName);
|
|
if (role == null) RoleHelper.SaveRole(Role);
|
|
role = RoleHelper.RetrieveRoles().FirstOrDefault(m => m.RoleName == Role.RoleName);
|
|
|
|
Assert.IsTrue(RoleHelper.SavaRolesByMenuId(menu.ID, role.ID.ToString()), "保存菜单角色关系失败");
|
|
Assert.IsTrue(RoleHelper.RetrieveRolesByMenuId(menu.ID).Count() > 0, string.Format("获取菜单ID={0}的角色关系失败", menu.ID));
|
|
|
|
//删除数据
|
|
string sql = "delete from Navigations where Name='__测试菜单__';";
|
|
sql += "delete from Roles where RoleName='_测试角色_';";
|
|
sql += string.Format("delete from NavigationRole where NavigationID={0}", menu.ID);
|
|
using (DbCommand cmd = DBAccessManager.SqlDBAccess.CreateCommand(CommandType.Text, sql))
|
|
{
|
|
DBAccessManager.SqlDBAccess.ExecuteNonQuery(cmd);
|
|
}
|
|
}
|
|
}
|
|
}
|