Function test_postgres_table_reading

tests/comparison/tests/test_cursor.py:172–184

Test that when a postgres table is read by the Postgresql cursor, the Table object contains the correct primary keys. This tests interacts with the local PostgreSQL database that's part of the minicluster.

(postgresql_cursor, sql_primary_key_map)

         'primary_keys': ('col1', 'col2', 'col3')}
    ])
def test_postgres_table_reading(postgresql_cursor, sql_primary_key_map):
  """
  Test that when a postgres table is read by the Postgresql cursor, the Table object
  contains the correct primary keys. This tests interacts with the local PostgreSQL
  database that's part of the minicluster.
  """
  try:
    postgresql_cursor.execute('DROP TABLE IF EXISTS mytable')
    postgresql_cursor.execute(sql_primary_key_map['sql'])
    table = postgresql_cursor.describe_table('mytable')
    assert table.primary_key_names == sql_primary_key_map['primary_keys']
  finally:
    postgresql_cursor.execute('DROP TABLE mytable')