def test_ChangingPackageStatuses(self): # DeliveryStep = 0, first status package = PackageObjectMother.CreatePackageNoPickupsNoReturn() package2 = PackageObjectMother.CreatePackageNoPickupsNoReturn() statuses = Package.FindAllPackageStatuses() for status in statuses: package.GetStatus().Name = status.Name self.assertEqual(package.GetStatus().Name, package2.GetStatus().Name) self.assertEqual(package.GetStatus().DeliveryStep, package2.GetStatus().DeliveryStep) package2.GetStatus().NextDeliveryStep()
def test_ChangePackageStatusForwardAndBackward(self): package = PackageObjectMother.CreatePackageNoPickupsNoReturn() package2 = PackageObjectMother.CreatePackageNoPickupsNoReturn() statuses = Package.FindAllPackageStatuses() statusNames = [status.Name for status in statuses] for i in range(10): for j in range(i): package.GetStatus().NextDeliveryStep() assert package.GetStatus().Name in statusNames for j in range(i): package.GetStatus().PrevDeliveryStep() assert package.GetStatus().Name == package2.GetStatus().Name
def GetAllPackageStatuses(self): return Package.FindAllPackageStatuses()