ExtendAgreement() public method

public ExtendAgreement ( int customerId ) : Agreement
customerId int
return Agreement
        public void Should_Extend_Agreement_Successfully()
        {
            using (IUnitOfWork unitOfWork = Singleton.UnitOfWorkFactory.New())
            {
                unitOfWork.BeginTransaction();

                var newCustomer = Singleton.TestDataProvider.CreateNewRandomCustomer();
                unitOfWork.Session.SaveOrUpdate(newCustomer);

                var newAgreement = Singleton.TestDataProvider.CreateNewRandomAgreementForCustomer(newCustomer);
                unitOfWork.Session.SaveOrUpdate(newAgreement);

                Mock<IUnitOfWork> unitOfWorkMock = new Mock<IUnitOfWork>();
                unitOfWorkMock.Setup(f => f.BeginTransaction()).Verifiable();
                unitOfWorkMock.Setup(f => f.Commit()).Callback(() => unitOfWork.Session.Flush()).Verifiable();
                unitOfWorkMock.Setup(f => f.Dispose()).Verifiable();
                unitOfWorkMock.Setup(f => f.Session).Returns(unitOfWork.Session);

                Mock<IUnitOfWorkFactory> unitOfWorkFactoryMock = new Mock<IUnitOfWorkFactory>();
                unitOfWorkFactoryMock.Setup(f => f.New()).Returns(unitOfWorkMock.Object);

                IAgreementManagementService agreementManagementService =
                    new AgreementManagementService(unitOfWork,
                                                   Singleton.ConfigurationLoaderService);

                var customer = unitOfWork.Session.Query<Agreement>().Fetch(f => f.Customer).First().Customer;
                Assert.IsNotNull(customer);

                var agreement = agreementManagementService.ExtendAgreement(customer.Id);
                Assert.IsNotNull(agreement);
                Assert.AreEqual(agreement.Customer, customer);

                unitOfWorkMock.Verify();
            }
        }
        public void Should_Not_Extend_Agreement_For_Not_Existing_Customer()
        {
            using (IUnitOfWork unitOfWork = Singleton.UnitOfWorkFactory.New())
            {
                IAgreementManagementService agreementManagementService =
                    new AgreementManagementService(unitOfWork, Singleton.ConfigurationLoaderService);

                Exception ex = Assert.Catch<AgreementManagementException>(delegate
                                                                              {
                                                                                  agreementManagementService.
                                                                                      ExtendAgreement(-1);
                                                                              });
                Assert.IsTrue(ex.InnerException is EntityNotFoundException);
            }
        }