Python ReadDatabase.build_model_ignore_deleted Exemples

Exemple #1

def test_build_model_ignore_deleted_invalid_fqid(
        read_database: ReadDatabase, connection: ConnectionHandler):
    fqid = MagicMock()
    events = [{"fqid": MagicMock()}]
    model = MagicMock()
    connection.query = q = MagicMock(return_value=events)
    read_database.build_model_from_events = bmfe = MagicMock(
        return_value=model)

    with pytest.raises(ModelDoesNotExist):
        read_database.build_model_ignore_deleted(fqid)

    assert q.call_args.args[1] == [(fqid, )]
    bmfe.assert_called_with(events)

Exemple #2

def test_build_model_ignore_deleted(read_database: ReadDatabase,
                                    connection: ConnectionHandler):
    fqid = MagicMock()
    events = [{
        "fqid": fqid,
        "data": MagicMock()
    }, {
        "fqid": fqid,
        "data": MagicMock()
    }]
    model = MagicMock()
    connection.query = q = MagicMock(return_value=events)
    read_database.build_model_from_events = bmfe = MagicMock(
        return_value=model)

    result = read_database.build_model_ignore_deleted(fqid)

    assert q.call_args.args[1] == [(fqid, )]
    bmfe.assert_called_with(events)
    assert result == model