public IActionResult AddEmployee(AddEmployeeToCompanyViewModel addEmployeeToCompanyViewModel)
{
if (ModelState.IsValid)
{
var employeeId = addEmployeeToCompanyViewModel.EmployeeId;
var companyId = addEmployeeToCompanyViewModel.CompanyId;
var existingItems = _companyEmployeeRepository.GetQueryable()
.Where(cm => cm.EmployeeId == employeeId)
.Where(cm => cm.CompanyId == companyId)
.ToList();
if (existingItems.Count == 0)
{
var newCompanyEmployee = new CompanyEmployee
{
EmployeeId = employeeId,
CompanyId = companyId
};
_companyEmployeeRepository.Add(newCompanyEmployee);
return(Redirect("/Companies/Details?id=" + companyId));
}
else
{
return(Redirect("/Characters/Details?id=" + companyId));
}
}
else
{
return(View(addEmployeeToCompanyViewModel));
}
}
public IActionResult AddEmployee(int id)
{
var company = _companyRepository.Get(id);
if (company != null)
{
var employees = _employeeRepository.GetAll().ToList();
if (employees != null && employees.Count > 0)
{
var addEmployeeToCompanyViewModel = new AddEmployeeToCompanyViewModel(company, employees);
return(View(addEmployeeToCompanyViewModel));
}
}
return(NotFound());
}
public void Ctor_ShouldSetupListOfEmployees_InitializedEmployees()
{
// Arrange
var company = new Company()
{
Name = "Alfavox"
};
var employees = new List <Employee>()
{
new Employee(), new Employee()
};
// Act
var viewModel = new AddEmployeeToCompanyViewModel(company, employees);
var result = viewModel.Employees;
// Assert
Assert.Equal(employees.Count(), result.Count());
}
public void Ctor_Should_SetupCompany()
{
// Arrange
var company = new Company()
{
Name = "Alfavox"
};
var employees = new List <Employee>()
{
new Employee(), new Employee()
};
// Act
var viewModel = new AddEmployeeToCompanyViewModel(company, employees);
var result = viewModel.Company;
// Assert
Assert.Equal(company, result);
}
