#! /usr/bin/python3

import comando_alterar_trecho
import tabelas
import trecho
import usuario
import sessao
import base_sql
import utils_testes

import sys

# Conecta no banco e carrega alimenta com as informações para o teste


sys.stderr.write("Conectando com base de dados...\n")
res = base_sql.conecta("DB", None, None)
assert res is None

sys.stderr.write("Criando alguns objetos...\n")
tabelas.cria_todos_os_testes()

# Sessao de teste
ses = sessao.busca_por_identificador("S-00000001")


def testa(rotulo, *args):
    """Testa {funcao(*args)}, grava resultado
    em "testes/saida/{modulo}.{funcao}.{rotulo}.html"."""

    modulo = comando_alterar_trecho
    funcao = modulo.processa
    frag = False  # {True} se for apenas um fragmento HTML, {False} se for página completa.
    pretty = False  # Se {True}, formata HTML para legibilidate (mas introduz brancos nos textos).
    utils_testes.testa_gera_html(modulo, funcao, rotulo, frag, pretty, *args)

# !!! CONSERTAR !!!

def testa_atualiza_atributo_com_sucesso():
    novo_trecho = "RIO-SP"
    args = {
        'id_trecho': "U-00000001",
        'trecho': novo_trecho,
    }
    testa("Suc", ses, args)

    updated_trecho = trecho.busca_por_identificador("U-00000001")

    assert trecho.obtem_atributo(updated_trecho, "trecho") == novo_trecho, "Trecho não atualizado"


def testa_atualiza_atributo_invalido():
    trecho_invalido = "RIO-LUA"
    args = {
        'id_trecho': "U-00000001",
        'trecho': trecho_invalido,
    }
    testa("Inv", ses, args)

    updated_trecho = trecho.busca_por_identificador("U-00000001")

    assert trecho.obtem_atributo(updated_trecho, "trecho") != trecho_invalido, "Trecho atualizado, mas nao deveria"

# Executa os testes

testa_atualiza_atributo_com_sucesso()
testa_atualiza_atributo_invalido()