public void Setup()
{
// Prepare TestBall
using (var db = new BirdieDBContext())
{
var newBall = new Ball()
{
Name = "Test Ball",
ColourId = 1
};
db.Ball.Add(newBall);
db.SaveChanges();
TestBall = db.Ball
.Where(b => b.Name == "Test Ball")
.Single();
}
// Prepare NewTestBall
using (var db = new BirdieDBContext())
{
var dbColour = db.Colour.Where(c => c.Name == "Red").Single();
NewTestBall = new Ball()
{
Name = "New Test Ball",
Colour = dbColour
};
}
}
public void SelectColourCorrectRowCountTest()
{
var colours = crud.RetrieveColours();
int numRows;
using (var db = new BirdieDBContext())
{
numRows = db.Colour.Count();
}
Assert.AreEqual(colours.Count(), numRows);
}
public void SelectBallCorrectRowCountTest()
{
var balls = crud.RetrieveBalls();
int numRows;
using (var db = new BirdieDBContext())
{
numRows = db.Ball.Count();
}
Assert.AreEqual(balls.Count(), numRows);
}
public IEnumerable <Colour> RetrieveColours()
{
var output = new List <Colour>();
using (var db = new BirdieDBContext())
{
output = db.Colour
.ToList();
}
return(output);
}
public void DeleteBall(Ball inputBall)
{
using (var db = new BirdieDBContext())
{
var originalBall = db.Ball
.Where(b => b.Id == inputBall.Id)
.Single();
db.Ball.Remove(originalBall);
db.SaveChanges();
}
}
public IEnumerable <Ball> RetrieveBalls()
{
var output = new List <Ball>();
using (var db = new BirdieDBContext())
{
output = db.Ball
.Include(b => b.Colour)
.ToList();
}
return(output);
}
public void DeleteBallTest()
{
crud.DeleteBall(TestBall);
int numResults;
using (var db = new BirdieDBContext())
{
numResults = db.Ball
.Where(b => b.Id == TestBall.Id).Count();
}
Assert.AreEqual(numResults, 0);
}
public void AddBallSQLInjectionTest()
{
NewTestBall.Name = "Test; DROP TABLE Balls;";
crud.AddBall(NewTestBall);
using (var db = new BirdieDBContext())
{
var Balls = db.Ball.ToList();
Assert.AreNotEqual(0, Balls.Count());
}
}
public void UpdateBall(Ball inputBall)
{
using (var db = new BirdieDBContext())
{
var originalBall = db.Ball
.Where(b => b.Id == inputBall.Id)
.Single();
originalBall.Name = inputBall.Name;
originalBall.ColourId = inputBall.Colour.Id;
db.SaveChanges();
}
}
public void AddBall(Ball inputBall)
{
var newBall = new Ball()
{
Name = inputBall.Name,
ColourId = inputBall.Colour.Id
};
using (var db = new BirdieDBContext())
{
db.Ball.Add(newBall);
db.SaveChanges();
}
}
public void AddBallDetailsCorrectTest()
{
crud.AddBall(NewTestBall);
using (var db = new BirdieDBContext())
{
var dbBall = db.Ball
.Where(b => b.Name == "New Test Ball")
.Single();
Assert.AreEqual("New Test Ball", dbBall.Name);
Assert.AreEqual(3, dbBall.ColourId);
}
}
public void TearDown()
{
using (var db = new BirdieDBContext())
{
var TestBalls = db.Ball
.Where(b => b.Name.Contains("Test"))
.ToList();
foreach (var item in TestBalls)
{
db.Ball.Remove(item);
}
db.SaveChanges();
}
}
public void AddBallRowCountIncreases()
{
int numRows;
int newNumRows;
using (var db = new BirdieDBContext())
{
numRows = db.Ball.Count();
crud.AddBall(NewTestBall);
newNumRows = db.Ball.Count();
}
Assert.AreEqual(numRows + 1, newNumRows);
}
public void UpdateBallTest()
{
using (var db = new BirdieDBContext())
{
var newColour = db.Colour.Where(c => c.Name == "Blue").Single();
TestBall.Name = "Test NameChange";
TestBall.Colour = newColour;
crud.UpdateBall(TestBall);
var newBall = db.Ball
.Where(b => b.Id == TestBall.Id)
.Single();
Assert.AreEqual("Test NameChange", newBall.Name);
Assert.AreEqual(newColour.Id, newBall.ColourId);
}
}
