def test_fail_add_contract_twice(contract, valid_federal_tax_number, valid_state_tax_number):
user = User(name="ECT",
federal_tax_number=valid_federal_tax_number,
state_tax_number=valid_state_tax_number,
status_number=1)
user.add_contract(contract)
with pytest.raises(InvalidUserContractError):
user.add_contract(contract)
def test_sanitize_user_data(valid_federal_tax_number, valid_state_tax_number):
user = User(name=" NAME WITH TRAILLING WHITESPACES ",
federal_tax_number=valid_federal_tax_number,
state_tax_number=valid_state_tax_number,
status_number="1 ")
assert user.name == "NAME WITH TRAILLING WHITESPACES"
assert user.status_number == 1
def test_basic_user(valid_federal_tax_number, valid_state_tax_number):
user = User(name="ECT",
federal_tax_number=valid_federal_tax_number,
state_tax_number=valid_state_tax_number,
status_number=1)
assert user.name == "ECT"
assert user.federal_tax_number == "73119555000120"
assert user.status_number == 1
assert user.contracts == []
def getCodRastreamento (self):
""" Adquire o código de rastreamento do objeto """
# Conexão com o webservice
env = SigepEnvironment.objects.get(ativo=True)
cliente = correios.Correios(username=env.usuario, password=env.senha, environment=env.ambiente)
# Solicita ao SIGEP o código de rastreamento.
user = User (env.nomeEmpresa, env.cnpj)
service = Service.get(self.servico.codigo)
self.codRastreamento = cliente.request_tracking_codes(user, service, 1)[0].code
def test_fail_add_contract_twice(contract, valid_federal_tax_number, valid_state_tax_number):
user = User(name="ECT",
federal_tax_number=valid_federal_tax_number,
state_tax_number=valid_state_tax_number,
status_number=1)
user.add_contract(contract)
with pytest.raises(InvalidUserContractError):
user.add_contract(contract)
def fecharPLP (self):
""" Fecha a PLP e a envia para o servidor, tornando-a uma lista de postagem """
# Conexão com o webservice
env = SigepEnvironment.objects.get(ativo=True)
cliente = correios.Correios(username=env.usuario, password=env.senha, environment=env.ambiente)
# Cria os objetos necessários para o fechamento da PLP
cartao = CartaoPostagem.objects.get(ativo=True)
user = User(env.nomeEmpresa, env.cnpj)
contract = Contract(user, cartao.nroContrato, cartao.codDR)
postingCard = PostingCard(contract, cartao.nroCartao, cartao.codAdmin)
senderAddr = self.converteEnderecoSelecionado()
# Inicialização da PLP
plp = PostingList(self.id)
# Inclusão dos objetos postais
for op in self.objetosPostais.all():
sl = ShippingLabel(postingCard, senderAddr, op.converteEnderecoSelecionado(), Servico.objects.first().codigo, op.codRastreamento, 1)
sl.posting_card = postingCard
plp.add_shipping_label(sl)
# Fechamento da PLP.
plp = cliente.close_posting_list(plp,cartao.nroCartao)