def test_apaga_telefone():
c = Contato("Aluno", "1199999-9999", "*****@*****.**")
c.adiciona_telefone("113322-4567", "fixo")
c.adiciona_telefone("1198888-8888", "celular")
c.apaga_telefone('fixo')
c.apaga_telefone('celular')
assert not any(['fixo' in c.get_telefones()])
assert not any(['celular' in c.get_telefones()])
assert c.get_telefones()['principal'] == TelefoneAtualizado("1199999-9999")
def test_11_apaga_telefone_principal(self):
c1 = Contato('Rafael', '11999777888', '*****@*****.**')
try:
c1.apaga_telefone('principal')
except DeleteError:
pass
except Exception:
raise AssertionError('Levantou o tipo de erro incorreto')
else:
raise AssertionError(
'Não levantou erro ao tentar apagar o telefone principal')
def test_10_apaga_telefone(self):
c1 = Contato('Rafael', '11999777888', '*****@*****.**')
c1.adiciona_telefone('11987654321', 'trabalho')
telefones = c1.get_telefones()
assert telefones['principal'] == TelefoneAtualizado('11999777888'), (
'o telefone não foi salvo com o valor correto no dicionário')
assert telefones['trabalho'] == TelefoneAtualizado('11987654321'), (
'o telefone não foi salvo com o valor ou a chave correta no dicionário'
)
c1.apaga_telefone('trabalho')
telefones = c1.get_telefones()
assert 'trabalho' not in telefones, 'o telefone de trabalho não foi apagado do dicionário'
assert 'principal' in telefones, (
'o telefone principal não deveria ter sido apagado do dicionário')
def test_apaga_telefone_tipo_nao_existe():
c = Contato("Aluno", "1199999-9999", "*****@*****.**")
c.adiciona_telefone("113322-4567", "fixo")
c.adiciona_telefone("1198888-8888", "celular")
tipos = ['particular', '1198888-8888', 11988888888]
for tipo in tipos:
try:
c.apaga_telefone(tipo)
except KeyError:
pass
else:
raise AssertionError("Não levantou erro para chave inexistente")
assert c.get_telefones()['principal'] == TelefoneAtualizado("1199999-9999")
assert c.get_telefones()['fixo'] == TelefoneAtualizado("113322-4567")
assert c.get_telefones()['celular'] == TelefoneAtualizado("1198888-8888")
