def test_validar_completereport_retorna_quantidade_de_estoque_correto(stock):
report = CompleteReport.generate(stock)
expected = ("Produtos Estocados por empresa: \n"
"- Forces of Nature: 1\n"
"- sanofi-aventis U.S. LLC: 2\n"
"- Newton Laboratories: 1\n")
assert expected in report
def test_validar_completereport_retorna_formato_correto(stock):
report = CompleteReport.generate(stock)
expected = ("Data de fabricação mais antiga: 2019-09-13\n"
"Data de validade mais próxima: 2023-01-17\n"
"Empresa com maior quantidade de produtos "
"estocados: sanofi-aventis U.S. LLC\n\n"
"Produtos Estocados por empresa: \n"
"- Forces of Nature: 1\n"
"- sanofi-aventis U.S. LLC: 2\n"
"- Newton Laboratories: 1\n")
assert expected in report
def import_data(self, file_path, report_type):
"""
Importa o arquivo informado em file_path e retorna o relatório
PARAMS:
- file_path: STRING - caminho do arquivo
- report_type: STRING - tipo do relatório (simples | completo)
"""
self.data.extend(self.importer.import_data(file_path))
if report_type == "simples":
return SimpleReport.generate(self.data)
if report_type == "completo":
return CompleteReport.generate(self.data)
def import_data(cls, file_path, report_type):
"""
Importa os dados para ser utilizado no relatório:
PARAMS:
- file_path: STRING, caminho do arquivo.
- report_type: STRING tipo de relatório ("simples", "completo").
"""
formats = {
".csv": cls.import_csv,
".xml": cls.import_xml,
".json": cls.import_json,
}
cls.file_path = file_path
file_format = path.splitext(file_path)[1]
formats.get(file_format)()
if report_type == "simples":
return SimpleReport.generate(cls.data)
elif report_type == "completo":
return CompleteReport.generate(cls.data)
def test_validar_completereport_retorna_empresa_com_maior_estoque(stock):
report = CompleteReport.generate(stock)
expected = ("Empresa com maior quantidade de produtos estocados:"
" sanofi-aventis U.S. LLC")
assert expected in report
def test_validar_completereport_retorna_validade_mais_proxima(stock):
report = CompleteReport.generate(stock)
assert "Data de validade mais próxima: 2023-01-17" in report
def test_validar_completereport_retorna_data_de_fabricacao_mais_antiga(stock):
report = CompleteReport.generate(stock)
assert "Data de fabricação mais antiga: 2019-09-13" in report
def import_data(cls, path, report_type):
data = cls.import_files(path)
if report_type == "simples":
return SimpleReport.generate(data)
else:
return CompleteReport.generate(data)
def report_size(cls, report, data):
if report == "simples":
return SimpleReport.generate(data)
if report == "completo":
return CompleteReport.generate(data)
def import_data(cls, filePath, reportType):
fileContent = cls.get_data_file(filePath)
if reportType == "simples":
return SimpleReport.generate(fileContent)
elif reportType == "completo":
return CompleteReport.generate(fileContent)
def import_data(self, filePath, reportType):
self.data += self.importer.import_data(filePath)
if reportType == "simples":
return SimpleReport.generate(self.data)
elif reportType == "completo":
return CompleteReport.generate(self.data)