单元测试写cookie

　我们在开发WEB项目的时候，一般应用逻辑跟ASPX页面是分离的项目。应用逻辑一般会是一个DLL组件项目。如果这个组件项目中A方法使用了Session、Cookie等信息的读写，则这个方法就很难写单元测试。
　　但并不是写不出来，要写出来大致思路如下：

　　目标：
　　构建一个测试的环境，把需要的Session、Cookie等信息初始化好。 这样才好做测试。而且这个构建的环境，不应该影响实际功能代码的编写。

　　具体实现来说：

　　我们要使用Mock技术，但就HttpContext来言，直接mock这个对象会有一个问题，它不具备Session的功能。这时候我们就需要用 Mock 技术来构造一个可以满足我们需要的环境的原理：这个Mock的机制如下：

　　用反射机制，构造一个 HttpSessionState 对象（HttpSessionState类的构造函数是internal 的），然后把这个对象跟SimpleWorkerRequest 对象捆绑。

这样我们就可以 构建了一个满足自己需要的环境了，即 TestHttpContext 类。

　　以下是两个类的实现：

 

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Web.SessionState;
using System.Web;
using System.Threading;
using System.Globalization;
using System.Collections.Specialized;
using System.Collections;
using System.IO;
using System.Web.Hosting;
using System.Reflection;

namespace TestNamespace
{
　 public class TestHttpContext
{
private const string ContextKeyAspSession = " AspSession " ;
private HttpContext context = null ;
private TestHttpContext() : base () { }
public TestHttpContext( bool isSecure)
: this ()
{
MySessionState myState = new MySessionState(Guid.NewGuid().ToString( " N " ),
new SessionStateItemCollection(), new HttpStaticObjectsCollection(),
5 , true , HttpCookieMode.UseUri, SessionStateMode.InProc, false );



TextWriter tw = new StringWriter();
HttpWorkerRequest wr = new SimpleWorkerRequest( " /webapp " , " c:\\inetpub\\wwwroot\\webapp\\ " , " default.aspx " , "" , tw);
this .context = new HttpContext(wr);
HttpSessionState state = Activator.CreateInstance(
typeof (HttpSessionState),
BindingFlags.Public | BindingFlags.NonPublic |
BindingFlags.Instance | BindingFlags.CreateInstance,
null ,
new object [] { myState },
CultureInfo.CurrentCulture) as HttpSessionState;
this .context.Items[ContextKeyAspSession] = state;
HttpContext.Current = this .context;
}

public HttpContext Context
{
get
{
return this .context;
}
}

private class WorkerRequest : SimpleWorkerRequest
{
private bool isSecure = false ;
public WorkerRequest( string page, string query, TextWriter output, bool isSecure)
: base (page, query, output)
{
this .isSecure = isSecure;
}

public override bool IsSecure()
{
return this .isSecure;
}
}
}
public sealed class MySessionState : IHttpSessionState
{
const int MAX_TIMEOUT = 24 * 60 ; // Timeout cannot exceed 24 hours.

string pId;
ISessionStateItemCollection pSessionItems;
HttpStaticObjectsCollection pStaticObjects;
int pTimeout;
bool pNewSession;
HttpCookieMode pCookieMode;
SessionStateMode pMode;
bool pAbandon;
bool pIsReadonly;

public MySessionState( string id,
ISessionStateItemCollection sessionItems,
HttpStaticObjectsCollection staticObjects,
int timeout,
bool newSession,
HttpCookieMode cookieMode,
SessionStateMode mode,
bool isReadonly)
{
pId = id;
pSessionItems = sessionItems;
pStaticObjects = staticObjects;
pTimeout = timeout;
pNewSession = newSession;
pCookieMode = cookieMode;
pMode = mode;
pIsReadonly = isReadonly;
}


public int Timeout
{
get { return pTimeout; }
set
{
if (value <= 0 )
throw new ArgumentException( " Timeout value must be greater than zero. " );

if (value > MAX_TIMEOUT)
throw new ArgumentException( " Timout cannot be greater than " + MAX_TIMEOUT.ToString());

pTimeout = value;
}
}


public string SessionID
{
get { return pId; }
}


public bool IsNewSession
{
get { return pNewSession; }
}


public SessionStateMode Mode
{
get { return pMode; }
}


public bool IsCookieless
{
get { return CookieMode == HttpCookieMode.UseUri; }
}


public HttpCookieMode CookieMode
{
get { return pCookieMode; }
}


//
// Abandon marks the session as abandoned. The IsAbandoned property is used by the
// session state module to perform the abandon work during the ReleaseRequestState event.
//
public void Abandon()
{
pAbandon = true ;
}

public bool IsAbandoned
{
get { return pAbandon; }
}

//
// Session.LCID exists only to support legacy ASP compatibility. ASP.NET developers should use
// Page.LCID instead.
//
public int LCID
{
get { return Thread.CurrentThread.CurrentCulture.LCID; }
set { Thread.CurrentThread.CurrentCulture = CultureInfo.ReadOnly( new CultureInfo(value)); }
}


//
// Session.CodePage exists only to support legacy ASP compatibility. ASP.NET developers should use
// Response.ContentEncoding instead.
//
public int CodePage
{
get
{
if (HttpContext.Current != null )
return HttpContext.Current.Response.ContentEncoding.CodePage;
else
return Encoding.Default.CodePage;
}
set
{
if (HttpContext.Current != null )
HttpContext.Current.Response.ContentEncoding = Encoding.GetEncoding(value);
}
}


public HttpStaticObjectsCollection StaticObjects
{
get { return pStaticObjects; }
}


public object this [ string name]
{
get { return pSessionItems[name]; }
set { pSessionItems[name] = value; }
}


public object this [ int index]
{
get { return pSessionItems[index]; }
set { pSessionItems[index] = value; }
}


public void Add( string name, object value)
{
pSessionItems[name] = value;
}


public void Remove( string name)
{
pSessionItems.Remove(name);
}


public void RemoveAt( int index)
{
pSessionItems.RemoveAt(index);
}


public void Clear()
{
pSessionItems.Clear();
}

public void RemoveAll()
{
Clear();
}



public int Count
{
get { return pSessionItems.Count; }
}



public NameObjectCollectionBase.KeysCollection Keys
{
get { return pSessionItems.Keys; }
}


public IEnumerator GetEnumerator()
{
return pSessionItems.GetEnumerator();
}


public void CopyTo(Array items, int index)
{
foreach ( object o in items)
items.SetValue(o, index ++ );
}


public object SyncRoot
{
get { return this ; }
}


public bool IsReadOnly
{
get { return pIsReadonly; }
}


public bool IsSynchronized
{
get { return false ; }
}

}

}

 

这样我们在进行单元测试时就可以Mock掉Web下的Session,Cookie等对象。

例如：

 

[TestMethod()]
public void TestGetUserId()
{
TestHttpContext mock = new TestHttpContext( false );
System.Web.HttpContext context = mock.Context;
context.Session[ " UserId " ] = 1245 ;

Assert.AreEqual( long .Parse(context.Session[ " UserId " ].ToString()), 1245 , " 读取用户ID错误！ " );
}