diff --git a/src/generator/dto-generator.ts b/src/generator/dto-generator.ts index cf1a412..71e69b4 100644 --- a/src/generator/dto-generator.ts +++ b/src/generator/dto-generator.ts @@ -202,12 +202,12 @@ export class DtoGenerator { } if (schema.format === 'date') { + decorators.push('@IsDate()'); + } + + if(schema.format === 'date-time') { decorators.push('@IsDateString()'); } - - if (schema.format === 'date-time') { - decorators.push('@IsDateTimeString()'); - } if (schema.enum) { const enumName = this.getEnumName(name, schema.enum); diff --git a/src/templates/dto.hbs b/src/templates/dto.hbs index c54e69e..b5d0a62 100644 --- a/src/templates/dto.hbs +++ b/src/templates/dto.hbs @@ -1,8 +1,9 @@ import { ApiProperty } from '@nestjs/swagger'; import { IsString, IsNumber, IsBoolean, IsArray, IsOptional, - IsEmail, IsEnum, Min, Max, MinLength, MaxLength, Matches, - ValidateNested, IsInt + IsEmail, IsEnum, IsUUID, IsDateString, + Min, Max, MinLength, MaxLength, Matches, + ValidateNested, IsInt, IsDate, } from 'class-validator'; import { Type } from 'class-transformer'; diff --git a/tests/generator/dto-generator.test.ts b/tests/generator/dto-generator.test.ts index a4fc40c..f22229a 100644 --- a/tests/generator/dto-generator.test.ts +++ b/tests/generator/dto-generator.test.ts @@ -41,7 +41,7 @@ describe('DtoGenerator', () => { expect(result).toContain('@IsEmail()'); expect(result).toContain('@IsUUID()'); expect(result).toContain('@IsDateString()'); - expect(result).toContain('@IsDateTimeString()'); + expect(result).toContain('@IsDate()'); // String length validation expect(result).toContain('@MinLength(1)');