redid mock test in TestClass.cs