public void AddModel(Model.Model model)
{
if (_context.Models.Any(m => m.Name == model.Name && m.Make == model.Make))
throw new ArgumentException("Model already exists");
_context.Models.Add(model);
_context.SaveChanges();
}
public VehicleControllerTests()
{
_context = new MockContext();
_context.Users.Add(new User
{
Id = 1,
Username = "******"
});
var make = new Make
{
Id = 1,
Name = "make"
};
var model = new Model
{
Id = 1,
Name = "model",
Make = make
};
var user = new User
{
Id = 1,
Email = "Test"
};
make.Models = new List<Model> {model};
_context.Models.Add(model);
_context.Makes.Add(make);
_context.Vehicles.Add(new Vehicle
{
Id = 1,
Year = "2015",
Kilometers = 100,
Model = model,
User = user
});
_context.Vehicles.Add(new Vehicle
{
Id = 2,
Year = "2014",
Kilometers = 1000,
Model = model,
User = user
});
user.Vehicles = new List<Vehicle>();
user.Vehicles.Add(_context.Vehicles.First());
user.Vehicles.Add(_context.Vehicles.Last());
}
public void AddWorkItem(int vehicleId, Model.WorkItem item)
{
var vehicle = _context.Vehicles.FirstOrDefault(v => v.Id == vehicleId);
if (vehicle == null)
throw new ArgumentException("Invalid Vehicle");
item.Vehicle = vehicle;
if (item.GetType() == typeof(Schedule))
_context.Schedules.Add((Schedule)item);
else
_context.Tasks.Add((MaintenanceTask)item);
_context.SaveChanges();
}
public void Add_Duplicate_Model()
{
var context = new MockContext();
var store = new VehicleStore(context);
var model = new Model()
{
Id = 2,
Make = _make,
Name = "Klaker"
};
context.Models.Add(model);
model.Id = 3;
store.AddModel(model);
}
public void Duplicate_Model_Different_Make()
{
var context = new MockContext();
var store = new VehicleStore(context);
var model = new Model()
{
Id = 2,
Make = _make,
Name = "Klaker"
};
context.Models.Add(model);
model = new Model()
{
Id = 3,
Name = "Klaker",
Make = new Make
{
Id = 3,
Name = "Suzoooki"
}
};
store.AddModel(model);
model = context.Models.Last();
Assert.AreEqual(3, model.Id);
Assert.AreEqual("Klaker", model.Name);
var make = model.Make;
Assert.AreEqual(3, make.Id);
Assert.AreEqual("Suzoooki", make.Name);
Assert.AreEqual(1, context.SaveChangesCalls);
}
