def test_getCompany(self): db = Database() db.insertCompany( Company(cnpj="93.612.749/0001-70", name="Companhia Teste 1", rate=25, term=72, warranty="Terreno")) db.insertCompany( Company(cnpj="27.456.797/0001-92", name="Companhia Teste 2", rate=25, term=48, warranty="Imóvel")) db.insertCompany( Company(cnpj="26.805.322/0001-00", name="Companhia Teste 3", rate=35, term=36, warranty="Imóvel")) company = db.getCompany(cnpj="93.612.749/0001-70") db.database.company.delete_one({'cnpj': "93.612.749/0001-70"}) db.database.company.delete_one({'cnpj': "27.456.797/0001-92"}) db.database.company.delete_one({'cnpj': "26.805.322/0001-00"}) self.assertEqual(company.name, "Companhia Teste 1") self.assertEqual(company.rate, 25) self.assertEqual(company.term, 72) self.assertEqual(company.warranty, "Terreno")
def test_insertCompany(self): db = Database() db.insertCompany( Company(cnpj="42.109.758/0001-02", name="Companhia Teste 1", rate=25, term=72, warranty="Terreno")) db.insertCompany( Company(cnpj="42.742.578/0001-63", name="Companhia Teste 2", rate=25, term=48, warranty="Imóvel")) db.insertCompany( Company(cnpj="45.441.634/0001-18", name="Companhia Teste 3", rate=35, term=36, warranty="Imóvel")) db.database.company.delete_one({'cnpj': "42.109.758/0001-02"}) db.database.company.delete_one({'cnpj': "42.742.578/0001-63"}) db.database.company.delete_one({'cnpj': "45.441.634/0001-18"})
# coding: utf-8 import os from modules.database import Database from modules.models.financier import Financier from modules.models.company import Company saida = os.system("pip install -r requirements.pip") db = Database() db.client.drop_database('credit') db.insertFinancier(Financier(cnpj = "42.109.758/0001-02", name = "Banco do Pernambuco", rate = 25, term = 72, warranty = "Terreno")) db.insertFinancier(Financier(cnpj = "42.742.578/0001-63", name = "Banco Camargo de Araújo", rate = 10, term = 48, warranty = "Imóvel")) db.insertFinancier(Financier(cnpj = "27.948.482/0001-62", name = "InovaCred", rate = 12, term = 36, warranty = "Terreno")) db.insertFinancier(Financier(cnpj = "23.918.334/0001-44", name = "Banco do Pará", rate = 25, term = 72, warranty = "Terreno")) db.insertFinancier(Financier(cnpj = "81.396.636/0001-04", name = "Dinheiro Certo", rate = 10, term = 48, warranty = "Imóvel")) db.insertFinancier(Financier(cnpj = "38.265.772/0001-28", name = "CredFácil", rate = 12, term = 36, warranty = "Terreno")) db.insertCompany(Company(cnpj = "18.259.473/0001-73", name = "Empreendimentos do Sol", rate = 25, term = 72, warranty = "Terreno")) db.insertCompany(Company(cnpj = "14.886.257/0001-05", name = "Floricultura do Azevedo", rate = 25, term = 48, warranty = "Imóvel")) db.insertCompany(Company(cnpj = "63.816.653/0001-63", name = "Bar's Rock", rate = 12, term = 36 , warranty = "Terreno"))