# Path: pytrustnfe/Servidores.py
# def localizar_url(servico, estado, mod="55", ambiente=2):
#     sigla = SIGLA_ESTADO[estado]
#     ws = ESTADO_WS[sigla]
# 
#     if servico in (WS_DFE_DISTRIBUICAO, WS_DOWNLOAD_NFE):
#         ws = AN
# 
#     if mod in ws:
#         dominio = ws[mod][ambiente]["servidor"]
#         complemento = ws[mod][ambiente][servico]
#     else:
#         dominio = ws[ambiente]["servidor"]
#         complemento = ws[ambiente][servico]
# 
#     if sigla == "RS" and servico == WS_NFE_CADASTRO:
#         dominio = "cad.sefazrs.rs.gov.br"
#     if sigla in ("AC", "RN", "PB", "SC", "RJ") and servico == WS_NFE_CADASTRO:
#         dominio = "cad.svrs.rs.gov.br"
#     if sigla == "AN" and servico == WS_NFE_RECEPCAO_EVENTO:
#         dominio = "www.nfe.fazenda.gov.br"
# 
#     return "https://%s/%s" % (dominio, complemento)
# 
# def localizar_qrcode(estado, ambiente=2):
#     sigla = SIGLA_ESTADO[estado]
#     ws_qrcode = ESTADO_WS[sigla][NFCE_MODELO][ambiente][WS_NFCE_QR_CODE]
#     return ws_qrcode


import unittest
from pytrustnfe.Servidores import localizar_url, localizar_qrcode
url_qrcode_homologacao_sp = "https://homologacao.nfce.fazenda.sp.gov.br/NFCEConsultaPublica/Paginas/ConstultaQRCode.aspx"

url_sc = "https://nfe.svrs.rs.gov.br/ws/NfeAutorizacao/NFeAutorizacao4.asmx?wsdl"

url_rs = "https://nfe.sefazrs.rs.gov.br/ws/NfeAutorizacao/NFeAutorizacao4.asmx?wsdl"

url_cad_rs = "https://cad.sefazrs.rs.gov.br/ws/cadconsultacadastro/cadconsultacadastro4.asmx?wsdl"

url_cad_sc = "https://cad.svrs.rs.gov.br/ws/cadconsultacadastro/cadconsulta\
cadastro4.asmx?wsdl"


class test_servidores(unittest.TestCase):
    def test_localizar_url(self):
        url = localizar_url("NfeAutorizacao", "29", ambiente=1)
        self.assertEqual(url, url_ba)
        url = localizar_url("NfeAutorizacao", "35", ambiente=1)
        self.assertEqual(url, url_sp)
        url = localizar_url("NfeAutorizacao", "42", ambiente=1)
        self.assertEqual(url, url_sc)
        url = localizar_url("NfeAutorizacao", "43", ambiente=1)
        self.assertEqual(url, url_rs)

        url = localizar_url("NfeConsultaCadastro", "43", ambiente=2)
        self.assertEqual(url, url_cad_rs)

        url = localizar_url("NfeConsultaCadastro", "42", ambiente=2)
        self.assertEqual(url, url_cad_sc)

    def test_localizar_qrcode(self):

# Path: pytrustnfe/Servidores.py
# def localizar_url(servico, estado, mod="55", ambiente=2):
#     sigla = SIGLA_ESTADO[estado]
#     ws = ESTADO_WS[sigla]
# 
#     if servico in (WS_DFE_DISTRIBUICAO, WS_DOWNLOAD_NFE):
#         ws = AN
# 
#     if mod in ws:
#         dominio = ws[mod][ambiente]["servidor"]
#         complemento = ws[mod][ambiente][servico]
#     else:
#         dominio = ws[ambiente]["servidor"]
#         complemento = ws[ambiente][servico]
# 
#     if sigla == "RS" and servico == WS_NFE_CADASTRO:
#         dominio = "cad.sefazrs.rs.gov.br"
#     if sigla in ("AC", "RN", "PB", "SC", "RJ") and servico == WS_NFE_CADASTRO:
#         dominio = "cad.svrs.rs.gov.br"
#     if sigla == "AN" and servico == WS_NFE_RECEPCAO_EVENTO:
#         dominio = "www.nfe.fazenda.gov.br"
# 
#     return "https://%s/%s" % (dominio, complemento)
# 
# def localizar_qrcode(estado, ambiente=2):
#     sigla = SIGLA_ESTADO[estado]
#     ws_qrcode = ESTADO_WS[sigla][NFCE_MODELO][ambiente][WS_NFCE_QR_CODE]
#     return ws_qrcode


# Path: tests/test_servidores.py
import unittest
from pytrustnfe.Servidores import localizar_url, localizar_qrcode

url_qrcode_homologacao_sp = "https://homologacao.nfce.fazenda.sp.gov.br/NFCEConsultaPublica/Paginas/ConstultaQRCode.aspx"

url_sc = "https://nfe.svrs.rs.gov.br/ws/NfeAutorizacao/NFeAutorizacao4.asmx?wsdl"

url_rs = "https://nfe.sefazrs.rs.gov.br/ws/NfeAutorizacao/NFeAutorizacao4.asmx?wsdl"

url_cad_rs = "https://cad.sefazrs.rs.gov.br/ws/cadconsultacadastro/cadconsultacadastro4.asmx?wsdl"

url_cad_sc = "https://cad.svrs.rs.gov.br/ws/cadconsultacadastro/cadconsulta\
cadastro4.asmx?wsdl"


class test_servidores(unittest.TestCase):
    def test_localizar_url(self):
        url = localizar_url("NfeAutorizacao", "29", ambiente=1)
        self.assertEqual(url, url_ba)
        url = localizar_url("NfeAutorizacao", "35", ambiente=1)
        self.assertEqual(url, url_sp)
        url = localizar_url("NfeAutorizacao", "42", ambiente=1)
        self.assertEqual(url, url_sc)
        url = localizar_url("NfeAutorizacao", "43", ambiente=1)
        self.assertEqual(url, url_rs)

        url = localizar_url("NfeConsultaCadastro", "43", ambiente=2)
        self.assertEqual(url, url_cad_rs)

        url = localizar_url("NfeConsultaCadastro", "42", ambiente=2)
        self.assertEqual(url, url_cad_sc)

    def test_localizar_qrcode(self):

        url = localizar_qrcode("35")