def test_save_hbase_flags_unremove(self, _get_table):
table_mock = mock.MagicMock()
table_mock.delete.return_value = None
_get_table.return_value = table_mock
expected_output = {"status": "Success!"}
data = PIPPrincipaisInvestigadosDAO.save_hbase_flags(
"1", "2", "1234", "unremove")
hbspace = settings.PROMOTRON_HBASE_NAMESPACE
_get_table.assert_called_once_with(hbspace + "pip_investigados_flags")
table_mock.delete.assert_called_once_with(b"121234",
columns=["flags:is_removed"])
assert expected_output == data
def post(self, request, *args, **kwargs):
orgao_id = kwargs.get(self.orgao_url_kwarg)
cpf = kwargs.get("cpf")
# TODO: Verificar que o post foi feito pelo mesmo orgao
action = request.POST.get("action")
representante_dk = request.POST.get("representante_dk")
# Nome de personagem é necessário para a chave do HBase
if not representante_dk:
raise ValueError("Campo 'representante_dk' não foi dado!")
if not action:
raise ValueError("Campo 'action' não foi dado!")
data = PIPPrincipaisInvestigadosDAO.save_hbase_flags(
orgao_id, cpf, representante_dk, action)
return Response(data)
def test_save_hbase_flags_remove(self, _get_table):
table_mock = mock.MagicMock()
table_mock.put.return_value = None
_get_table.return_value = table_mock
expected_output = {"status": "Success!"}
expected_call_arguments = {
b"identificacao:orgao_id": b"1",
b"identificacao:cpf": b"2",
b"identificacao:representante_dk": b"1234",
b"flags:is_removed": b"True"
}
data = PIPPrincipaisInvestigadosDAO.save_hbase_flags(
"1", "2", "1234", "remove")
hbspace = settings.PROMOTRON_HBASE_NAMESPACE
_get_table.assert_called_once_with(hbspace + "pip_investigados_flags")
table_mock.put.assert_called_once_with(b"121234",
expected_call_arguments)
assert expected_output == data