Python get_instance_id Exemples

Exemple #1

def test_emit_sql_execution_end_event():
    event_name = constants.SQL_EXECUTION_END_EVENT
    session_id = 17
    language = constants.SESSION_KIND_SPARK
    success = False
    exception_type = 'ValueError'
    exception_message = 'You screwed up'

    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp), (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language),
                   (constants.SESSION_ID, session_id),
                   (constants.SQL_GUID, guid2),
                   (constants.STATEMENT_GUID, guid3),
                   (constants.SUCCESS, success),
                   (constants.EXCEPTION_TYPE, exception_type),
                   (constants.EXCEPTION_MESSAGE, exception_message)]

    spark_events.emit_sql_execution_end_event(guid1, language, session_id,
                                              guid2, guid3, success,
                                              exception_type,
                                              exception_message)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #2

def test_send_to_handler():
    kwargs_list = [(TIMESTAMP, time_stamp)]
    expected_kwargs_list = [(INSTANCE_ID, get_instance_id())] + kwargs_list

    events.send_to_handler(kwargs_list)

    events.handler.handle_event.assert_called_once_with(expected_kwargs_list)

Exemple #3

def test_emit_sql_execution_start_event():
    event_name = constants.SQL_EXECUTION_START_EVENT
    session_id = 22
    language = constants.SESSION_KIND_SPARK
    samplemethod = 'sample'
    maxrows = 12
    samplefraction = 0.5

    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language),
                   (constants.SESSION_ID, session_id),
                   (constants.SQL_GUID, guid2),
                   (constants.SAMPLE_METHOD, samplemethod),
                   (constants.MAX_ROWS, maxrows),
                   (constants.SAMPLE_FRACTION, samplefraction)]

    spark_events.emit_sql_execution_start_event(guid1, language, session_id, guid2, samplemethod,
                                                maxrows, samplefraction)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #4

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_sql_execution_end_event():
    event_name = constants.SQL_EXECUTION_END_EVENT
    session_id = 17
    language = constants.SESSION_KIND_SPARK
    success = False
    exception_type = 'ValueError'
    exception_message = 'You screwed up'

    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language),
                   (constants.SESSION_ID, session_id),
                   (constants.SQL_GUID, guid2),
                   (constants.STATEMENT_GUID, guid3),
                   (constants.SUCCESS, success),
                   (constants.EXCEPTION_TYPE, exception_type),
                   (constants.EXCEPTION_MESSAGE, exception_message)]

    spark_events.emit_sql_execution_end_event(guid1, language, session_id, guid2, guid3,
                                              success, exception_type, exception_message)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #5

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_sql_execution_start_event():
    event_name = constants.SQL_EXECUTION_START_EVENT
    session_id = 22
    language = constants.SESSION_KIND_SPARK
    samplemethod = 'sample'
    maxrows = 12
    samplefraction = 0.5

    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language),
                   (constants.SESSION_ID, session_id),
                   (constants.SQL_GUID, guid2),
                   (constants.SAMPLE_METHOD, samplemethod),
                   (constants.MAX_ROWS, maxrows),
                   (constants.SAMPLE_FRACTION, samplefraction)]

    spark_events.emit_sql_execution_start_event(guid1, language, session_id, guid2, samplemethod,
                                                maxrows, samplefraction)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #6

Fichier : test_events.py Projet : jupyter-incubator/sparkmagic

def test_send_to_handler():
    kwargs_list = [(TIMESTAMP, time_stamp)]
    expected_kwargs_list = [(INSTANCE_ID, get_instance_id())] + kwargs_list

    events.send_to_handler(kwargs_list)

    events.handler.handle_event.assert_called_once_with(expected_kwargs_list)

Exemple #7

def test_emit_library_loaded_event():
    event_name = constants.LIBRARY_LOADED_EVENT
    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp)]

    spark_events.emit_library_loaded_event()

    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #8

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_library_loaded_event():
    event_name = constants.LIBRARY_LOADED_EVENT
    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp)]

    spark_events.emit_library_loaded_event()

    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #9

Fichier : test_sparkevents.py Projet : datalayer-externals/jupyter-kernel-magic-spark

def test_not_emit_graph_render_event_when_not_registered():
    event_name = GRAPH_RENDER_EVENT
    graph_type = 'Bar'

    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp), (GRAPH_TYPE, graph_type)]

    events.emit_graph_render_event(graph_type)

    events.get_utc_date_time.assert_called_with()
    assert not events.handler.handle_event.called

Exemple #10

def test_emit_session_creation_start_event():
    language = constants.SESSION_KIND_SPARK
    event_name = constants.SESSION_CREATION_START_EVENT
    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp), (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language)]

    spark_events.emit_session_creation_start_event(guid1, language)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #11

Fichier : test_sparkevents.py Projet : ganeshraju/sparkmagic

def test_not_emit_graph_render_event_when_not_registered():
    event_name = GRAPH_RENDER_EVENT
    graph_type = 'Bar'

    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (GRAPH_TYPE, graph_type)]

    events.emit_graph_render_event(graph_type)

    events.get_utc_date_time.assert_called_with()
    assert not events.handler.handle_event.called

Exemple #12

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_session_creation_start_event():
    language = constants.SESSION_KIND_SPARK
    event_name = constants.SESSION_CREATION_START_EVENT
    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language)]

    spark_events.emit_session_creation_start_event(guid1, language)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #13

def test_emit_cluster_change_event():
    status_code = 200

    event_name = constants.CLUSTER_CHANGE_EVENT
    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.CLUSTER_DNS_NAME, guid1),
                   (constants.STATUS_CODE, status_code),
                   (constants.SUCCESS, True), (constants.ERROR_MESSAGE, None)]

    spark_events.emit_cluster_change_event(guid1, status_code, True, None)

    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #14

def test_emit_magic_execution_start_event():
    event_name = constants.MAGIC_EXECUTION_START_EVENT
    magic_name = 'sql'
    language = constants.SESSION_KIND_SPARKR

    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp), (constants.MAGIC_NAME, magic_name),
                   (constants.LIVY_KIND, language),
                   (constants.MAGIC_GUID, guid1)]

    spark_events.emit_magic_execution_start_event(magic_name, language, guid1)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #15

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_cluster_change_event():
    status_code = 200

    event_name = constants.CLUSTER_CHANGE_EVENT
    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.CLUSTER_DNS_NAME, guid1),
                   (constants.STATUS_CODE, status_code),
                   (constants.SUCCESS, True),
                   (constants.ERROR_MESSAGE, None)]

    spark_events.emit_cluster_change_event(guid1, status_code, True, None)

    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #16

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_magic_execution_start_event():
    event_name = constants.MAGIC_EXECUTION_START_EVENT
    magic_name = 'sql'
    language = constants.SESSION_KIND_SPARKR

    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.MAGIC_NAME, magic_name),
                   (constants.LIVY_KIND, language),
                   (constants.MAGIC_GUID, guid1)]

    spark_events.emit_magic_execution_start_event(magic_name, language, guid1)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #17

def test_emit_session_creation_end_event():
    language = constants.SESSION_KIND_SPARK
    event_name = constants.SESSION_CREATION_END_EVENT
    status = constants.BUSY_SESSION_STATUS
    session_id = 0
    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp), (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language),
                   (constants.SESSION_ID, session_id),
                   (constants.STATUS, status), (constants.SUCCESS, True),
                   (constants.EXCEPTION_TYPE, ""),
                   (constants.EXCEPTION_MESSAGE, "")]

    spark_events.emit_session_creation_end_event(guid1, language, session_id,
                                                 status, True, "", "")

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #18

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_session_creation_end_event():
    language = constants.SESSION_KIND_SPARK
    event_name = constants.SESSION_CREATION_END_EVENT
    status = constants.BUSY_SESSION_STATUS
    session_id = 0
    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.SESSION_GUID, guid1),
                   (constants.LIVY_KIND, language),
                   (constants.SESSION_ID, session_id),
                   (constants.STATUS, status),
                   (constants.SUCCESS, True),
                   (constants.EXCEPTION_TYPE, ""),
                   (constants.EXCEPTION_MESSAGE, "")]

    spark_events.emit_session_creation_end_event(guid1, language, session_id, status, True, "", "")

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #19

def test_emit_magic_execution_end_event():
    event_name = constants.MAGIC_EXECUTION_END_EVENT
    magic_name = 'sql'
    language = constants.SESSION_KIND_SPARKR
    success = True
    exception_type = ''
    exception_message = ''

    kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp), (constants.MAGIC_NAME, magic_name),
                   (constants.LIVY_KIND, language),
                   (constants.MAGIC_GUID, guid1), (constants.SUCCESS, success),
                   (constants.EXCEPTION_TYPE, exception_type),
                   (constants.EXCEPTION_MESSAGE, exception_message)]

    spark_events.emit_magic_execution_end_event(magic_name, language, guid1,
                                                success, exception_type,
                                                exception_message)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #20

Fichier : test_sparkevents.py Projet : jupyter-incubator/sparkmagic

def test_emit_magic_execution_end_event():
    event_name = constants.MAGIC_EXECUTION_END_EVENT
    magic_name = 'sql'
    language = constants.SESSION_KIND_SPARKR
    success = True
    exception_type = ''
    exception_message = ''

    kwargs_list = [(INSTANCE_ID, get_instance_id()),
                   (EVENT_NAME, event_name),
                   (TIMESTAMP, time_stamp),
                   (constants.MAGIC_NAME, magic_name),
                   (constants.LIVY_KIND, language),
                   (constants.MAGIC_GUID, guid1),
                   (constants.SUCCESS, success),
                   (constants.EXCEPTION_TYPE, exception_type),
                   (constants.EXCEPTION_MESSAGE, exception_message)]

    spark_events.emit_magic_execution_end_event(magic_name, language, guid1, success,
                                                exception_type, exception_message)

    spark_events._verify_language_ok.assert_called_once_with(language)
    spark_events.get_utc_date_time.assert_called_with()
    spark_events.handler.handle_event.assert_called_once_with(kwargs_list)

Exemple #21

Fichier : events.py Projet : datalayer-externals/jupyter-kernel-magic-spark

    def send_to_handler(self, kwargs_list):
        kwargs_list = [(INSTANCE_ID, get_instance_id())] + kwargs_list

        assert len(kwargs_list) <= 12

        self.handler.handle_event(kwargs_list)