third_party/perfetto/ui/src/common/schema_unittest.ts

// Copyright (C) 2023 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import {DatabaseInfo, DatabaseSchema, SchemaError} from './schema';

test('DatabaseSchema > getKeys', () => {
  const info: DatabaseInfo = {
    tables: [
      {
        name: 'slice',
        columns: [
          {name: 'id'},
          {name: 'ts'},
          {name: 'dur'},
        ],
      },
    ],
  };
  const schema = new DatabaseSchema(info);
  expect(schema.getKeys('slice')).toEqual(new Set(['id', 'ts', 'dur']));
});

test('DatabaseSchema > getKeys > Sad path', () => {
  const info: DatabaseInfo = {
    tables: [],
  };
  const schema = new DatabaseSchema(info);
  expect(() => schema.getKeys('foo')).toThrow(SchemaError);
});