test_indexes.py

##
# Copyright (c) 2016-present MagicStack Inc.
# All rights reserved.
#
# See LICENSE for details.
##


import uuid

from edgedb.server import _testbase as tb


class TestIndexes(tb.DDLTestCase):

    async def test_index_01(self):
        result = await self.con.execute("""
            # setup delta
            #
            CREATE MIGRATION test::d1 TO eschema $$
                concept Person:
                    link first_name to str
                    link last_name to str

                    index name_index on (__self__.first_name,
                                         __self__.last_name)
            $$;

            COMMIT MIGRATION test::d1;

            SELECT
                schema::Concept {
                    indexes: {
                        expr
                    }
                }
            FILTER schema::Concept.name = 'test::Person';

            INSERT test::Person {
                first_name := 'Elon',
                last_name := 'Musk'
            };

            WITH MODULE test
            SELECT
                Person {
                    first_name
                }
            FILTER
                Person.first_name = 'Elon' AND Person.last_name = 'Musk';
            """)

        self.assert_data_shape(result, [
            None,

            None,

            [{
                'indexes': [{
                    'expr': 'SELECT (test::Person.first_name, '
                            'test::Person.last_name)'
                }]
            }],

            [1],

            [{
                'id': uuid.UUID,
                'first_name': 'Elon'
            }]
        ])