113 lines
4.7 KiB
Python
113 lines
4.7 KiB
Python
from unittest.mock import patch
|
|
|
|
def test_scrape_preview_endpoint(client, user_headers):
|
|
with patch('app.services.geocoding_service.GeocodingService.get_location_by_cep') as mock_geo, \
|
|
patch('app.services.scraper_service.ScraperService.scrape_google_maps') as mock_scrape:
|
|
|
|
mock_geo.return_value = {
|
|
'cep': '01310100', 'logradouro': 'Avenida Paulista', 'bairro': 'Bela Vista',
|
|
'cidade': 'São Paulo', 'uf': 'SP', 'formatted_cep': '01310-100'
|
|
}
|
|
mock_scrape.return_value = [
|
|
{
|
|
'nome_empresa': 'Padaria Paulista Real',
|
|
'telefone': '(11) 98888-7777',
|
|
'endereco': 'Av. Paulista, 1000',
|
|
'google_rating': 4.7,
|
|
'total_avaliacoes': 120,
|
|
'google_maps_url': 'https://maps.google.com'
|
|
}
|
|
]
|
|
|
|
response = client.post('/api/v1/leads/scrape-preview', headers=user_headers, json={
|
|
'cep': '01310-100',
|
|
'ramo': 'Padaria',
|
|
'max_results': 5
|
|
})
|
|
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert data['count'] == 1
|
|
assert data['results'][0]['nome_empresa'] == 'Padaria Paulista Real'
|
|
|
|
def test_import_selected_leads_endpoint(client, user_headers):
|
|
selected_items = [
|
|
{
|
|
'nome_empresa': 'Padaria Paulista Real',
|
|
'telefone': '(11) 98888-7777',
|
|
'endereco': 'Av. Paulista, 1000',
|
|
'google_rating': 4.7,
|
|
'total_avaliacoes': 120,
|
|
'google_maps_url': 'https://maps.google.com'
|
|
}
|
|
]
|
|
|
|
with patch('app.services.geocoding_service.GeocodingService.get_location_by_cep') as mock_geo:
|
|
mock_geo.return_value = {
|
|
'cep': '01310100', 'logradouro': 'Avenida Paulista', 'bairro': 'Bela Vista',
|
|
'cidade': 'São Paulo', 'uf': 'SP', 'formatted_cep': '01310-100'
|
|
}
|
|
|
|
response = client.post('/api/v1/leads/import-selected', headers=user_headers, json={
|
|
'cep': '01310-100',
|
|
'ramo': 'Padaria',
|
|
'selected_leads': selected_items
|
|
})
|
|
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert data['summary']['created_count'] == 1
|
|
|
|
def test_update_lead_status_and_interaction(client, user_headers):
|
|
# Criar lead via import-selected
|
|
with patch('app.services.geocoding_service.GeocodingService.get_location_by_cep') as mock_geo:
|
|
mock_geo.return_value = {
|
|
'cep': '01310100', 'logradouro': 'Avenida Paulista', 'bairro': 'Bela Vista',
|
|
'cidade': 'São Paulo', 'uf': 'SP', 'formatted_cep': '01310-100'
|
|
}
|
|
client.post('/api/v1/leads/import-selected', headers=user_headers, json={
|
|
'cep': '01310-100', 'ramo': 'Padaria',
|
|
'selected_leads': [{'nome_empresa': 'Padaria Teste', 'telefone': '(11) 98888-7777'}]
|
|
})
|
|
|
|
list_res = client.get('/api/v1/leads', headers=user_headers)
|
|
assert list_res.status_code == 200
|
|
leads = list_res.get_json()['leads']
|
|
assert len(leads) > 0
|
|
lead_id = leads[0]['id']
|
|
|
|
# Atualizar status para 'contatado'
|
|
update_res = client.put(f'/api/v1/leads/{lead_id}', headers=user_headers, json={
|
|
'status_funil': 'contatado',
|
|
'notas': 'Primeiro contato realizado via WhatsApp.'
|
|
})
|
|
assert update_res.status_code == 200
|
|
assert update_res.get_json()['status_funil'] == 'contatado'
|
|
|
|
def test_opt_out_lgpd(client, user_headers):
|
|
with patch('app.services.geocoding_service.GeocodingService.get_location_by_cep') as mock_geo:
|
|
mock_geo.return_value = {
|
|
'cep': '01310100', 'logradouro': 'Rua Teste', 'bairro': 'Centro',
|
|
'cidade': 'São Paulo', 'uf': 'SP', 'formatted_cep': '01310-100'
|
|
}
|
|
client.post('/api/v1/leads/import-selected', headers=user_headers, json={
|
|
'cep': '01310-100', 'ramo': 'Farmácia',
|
|
'selected_leads': [{'nome_empresa': 'Farmácia Teste', 'telefone': '(11) 97777-6666'}]
|
|
})
|
|
|
|
list_res = client.get('/api/v1/leads', headers=user_headers)
|
|
lead_id = list_res.get_json()['leads'][0]['id']
|
|
|
|
opt_res = client.post(f'/api/v1/leads/{lead_id}/opt-out', headers=user_headers)
|
|
assert opt_res.status_code == 200
|
|
assert opt_res.get_json()['lead']['status_funil'] == 'opt_out'
|
|
|
|
def test_export_leads(client, user_headers):
|
|
csv_res = client.get('/api/v1/leads/export?format=csv', headers=user_headers)
|
|
assert csv_res.status_code == 200
|
|
assert 'text/csv' in csv_res.content_type
|
|
|
|
json_res = client.get('/api/v1/leads/export?format=json', headers=user_headers)
|
|
assert json_res.status_code == 200
|
|
assert isinstance(json_res.get_json(), list)
|