コード例 #1
0
 def tearDown(self):
     User.drop_collection()
コード例 #2
0
ファイル: utils.py プロジェクト: brosquinha/legislei
def set_up_db(db):
    parlamentar_test = set_up_parlamentar()
    brasilia_tz = pytz.timezone('America/Sao_Paulo')
    orgaos_evento = [
        Orgao(nome="ÓrgãoTeste", sigla="OT", cargo="None", apelido="OhTe")
    ]
    eventos_presentes = [
        Evento(id="12345",
               nome="Evento teste",
               data_inicial=brasilia_tz.localize(datetime(2019, 1, 1)),
               data_final=brasilia_tz.localize(datetime(2019, 1, 1)),
               url="http://url.com",
               situacao="Encerrada",
               presenca=0,
               orgaos=orgaos_evento)
    ]
    eventos_ausentes = [
        Evento(id="123",
               nome="Evento teste",
               data_inicial=brasilia_tz.localize(datetime(2019, 1, 1)),
               data_final=brasilia_tz.localize(datetime(2019, 1, 1)),
               url="http://url.com",
               situacao="Cancelada",
               presenca=1,
               orgaos=orgaos_evento)
    ]
    Relatorio(
        pk=ObjectId("5c264b5e3a5efd576ecaf48e"),
        parlamentar=parlamentar_test,
        proposicoes=[],
        data_inicial=brasilia_tz.localize(datetime(2018, 12, 31)),
        data_final=brasilia_tz.localize(datetime(2019, 1, 7)),
        orgaos=[],
        eventos_presentes=eventos_presentes,
        eventos_ausentes=eventos_ausentes,
        eventos_previstos=[],
    ).save()
    Avaliacoes(
        pk=ObjectId("5c5116f5c3acc80004eada0a"),
        email="*****@*****.**",
        parlamentar=parlamentar_test,
        avaliacao="1",
        avaliado={
            "url":
            "url",
            "situacao":
            "Cancelada",
            "dataFinal":
            brasilia_tz.localize(datetime(2019, 1, 1)),
            "orgaos": [{
                "sigla": "OT",
                "nome": "ÓrgãoTeste",
                "apelido": "OhTe",
                "cargo": None
            }],
            "dataInicial":
            brasilia_tz.localize(datetime(2019, 1, 1)),
            "presenca":
            1,
            "nome":
            "Evento teste",
            "id":
            "123"
        },
        relatorioId=ObjectId("5c264b5e3a5efd576ecaf48e"),
    ).save()
    inscricoes = Inscricoes(parlamentares=[parlamentar_test], intervalo=7)
    User.drop_collection()
    User(
        username="******",
        email="*****@*****.**",
        password=
        "******",
        inscricoes=inscricoes,
        devices=[
            UserDevice(id="14",
                       name="dispositivo",
                       token="---token---",
                       active=True,
                       os="Android")
        ]).save()