def add_movies_view(request):
mc = MovieController()
if request.method == 'GET':
return mc.add(request)
if request.method == 'POST':
mc.add(request)
return redirect(list_movies_view)
def edit_movies_view(request, asset_id):
mc = MovieController()
response = mc.edit(request, asset_id)
if request.method == 'GET':
if mc.code_return == RETURN_ERROR:
return redirect(list_movies_view)
if mc.code_return == RETURN_OK:
return response
if request.method == 'POST':
return redirect(list_movies_view)
def testMovieController():
"""
Test MovieController
"""
repo = MoviesRepository()
repo.add(Movie(1, "a", "aa", "aaa"))
movie = Movie(2, "b", "bb", "bbb")
repo.add(movie)
controller = MovieController(repo, RentRepository(), UnreController())
controller.removeMovie(1)
assert controller.getAll() == [movie]
controller.removeMovie(2)
assert controller.getAll() == []
try:
controller.searchMovie(23)
assert False
except ExceptionMess:
pass
def list_movies_view(request):
gc = MovieController()
return gc.list(request)
def publish_movies_view(request, id):
controller = MovieController()
controller.publish(request, id)
if controller.code_return == RETURN_OK:
return redirect(list_movies_view)
def api_edit_movies_view(request):
mc = MovieController()
resp = mc.api_edit(request)
return HttpResponse(resp)
def testUnReController():
"""
Test UnReController
"""
client = ClientsRepository()
movie = MoviesRepository()
rent = RentRepository()
unre = UnreController()
clientC = ClientController(client, rent, unre)
movieC = MovieController(movie, rent, unre)
rentC = RentController(client, movie, rent, unre)
#Test Client Undo
clientC.addClient(Client(1, "A", 12121))
unre.Undo()
try:
clientC.GetClient(1)
assert False
except ExceptionMess:
pass
unre.Redo()
try:
clientC.GetClient(1)
except ExceptionMess:
assert False
try:
unre.Redo()
assert False
except ExceptionMess:
pass
unre.Undo()
try:
clientC.GetClient(1)
assert False
except ExceptionMess:
unre.Redo()
clientC.removeClient(1)
unre.Undo()
try:
clientC.GetClient(1)
except ExceptionMess:
assert False
unre.Redo()
try:
clientC.GetClient(1)
assert False
except ExceptionMess:
pass
try:
unre.Undo()
except ExceptionMess:
assert False
clientC.updateClient(Client(1, "v", 123))
unre.Undo()
assert clientC.GetClient(1) == Client(1, "A", 12121)
unre.Redo()
assert clientC.GetClient(1) == Client(1, "v", 123)
#Test Movie Undo
movieC.addMovie(Movie(1, "A", "AA", "AAA"))
unre.Undo()
try:
movieC.GetMovie(1)
assert False
except ExceptionMess:
pass
unre.Redo()
try:
movieC.GetMovie(1)
except ExceptionMess:
assert False
try:
unre.Redo()
assert False
except ExceptionMess:
pass
unre.Undo()
try:
movieC.GetMovie(1)
assert False
except ExceptionMess:
unre.Redo()
movieC.removeMovie(1)
unre.Undo()
try:
movieC.GetMovie(1)
except ExceptionMess:
assert False
unre.Redo()
try:
movieC.GetMovie(1)
assert False
except ExceptionMess:
pass
try:
unre.Undo()
except ExceptionMess:
assert False
movieC.updateMovie(Movie(1, "B", "BB", "BBB"))
unre.Undo()
assert movieC.GetMovie(1) == Movie(1, "A", "AA", "AAA")
unre.Redo()
assert movieC.GetMovie(1) == Movie(1, "B", "BB", "BBB")
#Test Rent Undo
rentC.rentMovie(Rent(1, 1))
unre.Undo()
assert not rentC.isClient(1)
unre.Redo()
assert rentC.isClient(1)
try:
unre.Redo()
assert False
except ExceptionMess:
pass
unre.Undo()
assert not rentC.isClient(1)
unre.Redo()
rentC.returnMovie(Rent(1,1))
unre.Undo()
assert rentC.isClient(1)
unre.Redo()
assert not rentC.isClient(1)
try:
unre.Undo()
except ExceptionMess:
assert False
