def test_serialize(self):
url = "url"
username = "******"
password = "******"
connection_string = get_connection_string(url, username, password)
http_client = MagicMock()
http_client.connection_string = connection_string
kind = "scala"
_t_config_hook({
"status_sleep_seconds": 0.01,
"statement_sleep_seconds": 0.01
})
session = LivySession(http_client, kind, "-1", False)
_t_config_hook({})
serialized = session.get_state().to_dict()
assert serialized["connectionstring"] == connection_string
assert serialized["id"] == "-1"
assert serialized["language"] == kind
assert serialized["sqlcontext"] == False
assert serialized["version"] == "0.0.0"
assert len(serialized.keys()) == 5
