trezor-suite/packages/schema-utils/tests/errors.test.ts

import { InvalidParameter } from '../src/errors';
import { Assert, Type } from '../src/index';

describe('Assert', () => {
    it('should not throw an error if matches schema', () => {
        const schema = Type.Object({ type: Type.String() });
        const value = { type: 'string' };
        expect(() => Assert(schema, value)).not.toThrow();
    });

    it('acts as a type guard', () => {
        const schema = Type.Object({ type: Type.String() });
        const value = { type: 'string' };
        Assert(schema, value);
        // @ts-expect-error
        value.type = 123;
    });

    it('should throw if type is mismateched', () => {
        const schema = Type.Object({ type: Type.Number() });
        const value = { type: 'string' };
        expect(() => Assert(schema, value)).toThrow(InvalidParameter);
        // Check error message
        try {
            Assert(schema, value);
        } catch (e) {
            expect(e.message).toEqual('Invalid parameter "type" (= "string"): Expected number');
        }
    });

    it('should throw if required field is missing', () => {
        const schema = Type.Object({ type: Type.Number() });
        const value = {};
        expect(() => Assert(schema, value)).toThrow(InvalidParameter);
        // Check error message
        try {
            Assert(schema, value);
        } catch (e) {
            expect(e.message).toEqual(
                'Invalid parameter "type" (= undefined): Expected required property',
            );
        }
    });

    it('should throw in case of custom types', () => {
        const schema = Type.Object({ type: Type.Uint() });
        const value = { type: 'xyz' };
        expect(() => Assert(schema, value)).toThrow(InvalidParameter);
        // Check error message
        try {
            Assert(schema, value);
        } catch (e) {
            expect(e.message).toEqual('Invalid parameter "type" (= "xyz"): Expected kind "Uint"');
        }
    });

    it('should throw when conditions are not met', () => {
        const schema = Type.Object({ type: Type.String({ minLength: 3 }) });
        const value = { type: 'A' };
        expect(() => Assert(schema, value)).toThrow(InvalidParameter);
        // Check error message
        try {
            Assert(schema, value);
        } catch (e) {
            expect(e.message).toEqual(
                'Invalid parameter "type" (= "A"): Expected string length greater or equal to 3',
            );
        }
    });

    it('should throw in case with unions of objects', () => {
        const schema = Type.Union([
            Type.Object({ type: Type.String() }),
            Type.Object({ something: Type.String() }),
        ]);
        const value = { type: 'string' };
        expect(() => Assert(schema, value)).not.toThrow();

        const badValue = { type: 123 };
        expect(() => Assert(schema, badValue)).toThrow(InvalidParameter);
    });

    it('should also accept null if optional', () => {
        const schema = Type.Object({ type: Type.Optional(Type.String()) });
        const value = { type: null };
        expect(() => Assert(schema, value)).not.toThrow();
    });

    it('should also accept null if optional on nested object', () => {
        const schema = Type.Object({ type: Type.Object({ deeper: Type.Optional(Type.String()) }) });
        const value = { type: { deeper: null } };
        expect(() => Assert(schema, value)).not.toThrow();
    });

    it('should also accept null if optional on union of objects', () => {
        const schema = Type.Union([
            Type.Object({ type: Type.Optional(Type.String()) }),
            Type.Object({ something: Type.String() }),
        ]);
        const value = { type: null };
        expect(() => Assert(schema, value)).not.toThrow();
    });
});