class Multiwoven::Integrations::Source::AwsAthena::Client

def check_connection(connection_config)
  connection_config = connection_config.with_indifferent_access
  athena_client = create_connection(connection_config)
  athena_client.list_work_groups
  ConnectionStatus.new(status: ConnectionStatusType["succeeded"]).to_multiwoven_message
rescue StandardError => e
  ConnectionStatus.new(status: ConnectionStatusType["failed"], message: e.message).to_multiwoven_message
end