Test working

tests/test_integration/test_utils.py

@@ -1,4 +1,12 @@
 from lolafect.lolaconfig import build_lolaconfig
+from lolafect.connections import (
+    open_ssh_tunnel_with_s3_pkey,
+    get_local_bind_address_from_ssh_tunnel,
+    close_ssh_tunnel,
+    connect_to_mysql,
+    close_mysql_connection,
+)
+from lolafect.utils import begin_sql_transaction, end_sql_transaction
 
 # __          __     _____  _   _ _____ _   _  _____ _
 # \ \        / /\   |  __ \| \ | |_   _| \ | |/ ____| |
@@ -18,24 +26,114 @@ TEST_LOLACONFIG = build_lolaconfig(flow_name="testing-suite")
 
 
 def test_sql_transaction_persists_changes_properly():
-    # Connect
+    test_local_bind_host = "127.0.0.1"
-    # Create table in Sandbox
+    test_local_bind_port = 12345
-    # Check that table is empty
+
-    # Start transaction
+    tunnel = open_ssh_tunnel_with_s3_pkey.run(
-    # Insert value in table
+        s3_bucket_name=TEST_LOLACONFIG.S3_BUCKET_NAME,
-    # Commit transaction
+        ssh_tunnel_credentials=TEST_LOLACONFIG.SSH_TUNNEL_CREDENTIALS,
-    # Check that value is there
+        remote_target_host=TEST_LOLACONFIG.DW_CREDENTIALS["host"],
-    # assert that the table was initially empty and afterwards it got a record
+        remote_target_port=TEST_LOLACONFIG.DW_CREDENTIALS["port"],
-    pass
+        local_bind_host=test_local_bind_host,
+        local_bind_port=test_local_bind_port,
+    )
+
+    connection = connect_to_mysql.run(
+        mysql_credentials=TEST_LOLACONFIG.DW_CREDENTIALS,
+        overriding_host_and_port=get_local_bind_address_from_ssh_tunnel.run(
+            tunnel=tunnel
+        ),
+    )
+
+    cursor = connection.cursor()
+    cursor.execute("""
+    CREATE TABLE sandbox.lolafect_transaction_test_table
+    (
+        a_test_column INT
+    )
+    """)
+    cursor.execute("""
+    SELECT a_test_column
+    FROM sandbox.lolafect_transaction_test_table
+    """)
+    table_is_empty_at_first = not bool(cursor.fetchall())  # An empty tuple yields False
+
+    begin_sql_transaction.run(connection=connection)
+    cursor.execute("""
+    INSERT INTO sandbox.lolafect_transaction_test_table (a_test_column)
+    VALUES (1)
+    """)
+    end_sql_transaction.run(connection=connection, dry_run=False)
+
+    cursor.execute("""
+    SELECT a_test_column
+    FROM sandbox.lolafect_transaction_test_table
+    """)
+    table_has_a_record_after_transaction = bool(cursor.fetchall())  # A non-empty tuple yields True
+
+    cursor.execute("""
+    DROP TABLE sandbox.lolafect_transaction_test_table
+    """
+                   )
+
+    close_mysql_connection.run(connection=connection)
+    close_ssh_tunnel.run(tunnel=tunnel)
+
+    assert table_is_empty_at_first and table_has_a_record_after_transaction
 
 
 def test_sql_transaction_rollbacks_changes_properly():
-    # Connect
+    test_local_bind_host = "127.0.0.1"
-    # Create table in Sandbox
+    test_local_bind_port = 12345
-    # Check that table is empty
+
-    # Start transaction
+    tunnel = open_ssh_tunnel_with_s3_pkey.run(
-    # Insert value in table
+        s3_bucket_name=TEST_LOLACONFIG.S3_BUCKET_NAME,
-    # Commit transaction
+        ssh_tunnel_credentials=TEST_LOLACONFIG.SSH_TUNNEL_CREDENTIALS,
-    # Check that the table is still empty
+        remote_target_host=TEST_LOLACONFIG.DW_CREDENTIALS["host"],
-    # assert that the table was initially empty and afterwards as well
+        remote_target_port=TEST_LOLACONFIG.DW_CREDENTIALS["port"],
-    pass
+        local_bind_host=test_local_bind_host,
+        local_bind_port=test_local_bind_port,
+    )
+
+    connection = connect_to_mysql.run(
+        mysql_credentials=TEST_LOLACONFIG.DW_CREDENTIALS,
+        overriding_host_and_port=get_local_bind_address_from_ssh_tunnel.run(
+            tunnel=tunnel
+        ),
+    )
+
+    cursor = connection.cursor()
+    cursor.execute("""
+    CREATE TABLE sandbox.lolafect_transaction_test_table
+    (
+        a_test_column INT
+    )
+    """)
+    cursor.execute("""
+    SELECT a_test_column
+    FROM sandbox.lolafect_transaction_test_table
+    """)
+    table_is_empty_at_first = not bool(cursor.fetchall())  # An empty tuple yields False
+
+    begin_sql_transaction.run(connection=connection)
+    cursor.execute("""
+    INSERT INTO sandbox.lolafect_transaction_test_table (a_test_column)
+    VALUES (1)
+    """)
+    end_sql_transaction.run(connection=connection, dry_run=True)
+
+    cursor.execute("""
+    SELECT a_test_column
+    FROM sandbox.lolafect_transaction_test_table
+    """)
+    table_is_empty_after_rollback = not bool(cursor.fetchall())  # An tuple yields False
+
+    cursor.execute("""
+    DROP TABLE sandbox.lolafect_transaction_test_table
+    """
+                   )
+
+    close_mysql_connection.run(connection=connection)
+    close_ssh_tunnel.run(tunnel=tunnel)
+
+    assert table_is_empty_at_first and table_is_empty_after_rollback