def test_alterar_filme(self):
elenco = {"Ator": "Beltrano", "Diretor": "Fulano"}
filme.criar_filme(1, "Filme", "terror", elenco, 120, "18 Anos", "UCI")
filme.alterar_filme(1, "NovoFilme", "aventura", {
"Ator": "Ciclano",
"Diretor": "Fulano"
}, 100, "10 Anos", "CineMark")
f = filme.obter_filme(1)
self.assertEqual("NovoFilme", f.nome)
def test_obter_filme(self):
filme.criar_filme(1, "Filme", "terror", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 120, "18 Anos", "UCI")
filme.criar_filme(2, "Filme2", "terror2", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 130, "16 Anos", "UCA")
filmeObtido = filme.obter_filme(1)
self.assertEqual(1, filmeObtido.cod)
def test_excluir_filme(self):
filme.criar_filme(1, "Filme", "terror", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 120, "18 Anos", "UCI")
filme.criar_filme(2, "Filme2", "terror2", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 130, "16 Anos", "UCA")
filme.excluir_filme(2)
f = filme.obter_filme(2)
self.assertIsNone(f)
def test_criar_filme(self):
filme.criar_filme(1, "Filme", "terror", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 120, "18 Anos", "UCI")
filmes = filme.listar_filmes()
self.assertEqual(1, len(filmes))
s = filmes[0]
self.assertEqual(1, s.cod)
self.assertEqual("Filme", s.nome)
self.assertEqual("terror", s.genero)
self.assertEqual({"Ator": "Beltrano", "Diretor": "Fulano"}, s.elenco)
self.assertEqual(120, s.duracao)
self.assertEqual("18 Anos", s.classificacao)
self.assertEqual("UCI", s.distribuidora)
def menu_criar():
cod_filme = int(input('Digite o código do filme: '))
nome = str(input('Digite o nome do filme :'))
genero = str(input('Digite o genero do filme :'))
menuElenco = 0
listaElenco = []
while menuElenco == 0:
print("\n----------------\n" + "(1) Criar Ator \n" +
"(2) Listar Atores \n" + "(3) Adicionar Ator ao Elenco \n" +
"(4) Ver Elenco \n" + "(0) Terminar Elenco \n" +
"----------------")
opt = int(input("Escolha a opção: "))
if opt == 1:
menu_ator.adicionar_ator()
if opt == 2:
menu_ator.listar_ator()
if opt == 3:
cod_ator = int(input("Insira o codigo do ator: "))
atorElenco = ator.obter_ator(cod_ator)
if atorElenco == None:
print("Ator não encontrado")
else:
print("Deseja adicionar: ", atorElenco[1],
" ao elenco? (S/N)")
rodando = 0
while rodando == 0:
resp = str(input(": "))
if resp == "S" or resp == "s":
nomeAtor = atorElenco[1]
listaElenco.append(nomeAtor)
print("Ator ADICIONADO")
rodando = 1
elif resp == "N" or resp == "n":
print("Ator NAO adicionado")
rodando = 1
else:
print("Insira uma resposta valida")
if opt == 4:
print(listaElenco)
if opt == 0:
menuElenco = 1
dura = int(input('Digite a duracao do filme :'))
clasi = str(input('Digite a classificacao do filme :'))
distr = str(input('Digite a distribuidora do filme :'))
filme.criar_filme(cod_filme, nome, genero, listaElenco, dura, clasi, distr)