Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests(ui): coverage for validateConnectionTypes

Browse Source
This commit is contained in:
psychedelicious 2024-05-19 00:53:27 +10:00
parent 059d5a682c
commit 8074a802d6
2 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/frontend/web/src/features/nodes/store/util/validateConnectionTypes.test.ts
View File

@ -175,7 +175,7 @@ describe(validateConnectionTypes.name, () => {
it.each(typePairs)('should accept Collection $t1 to Collection $t2', ({ t1, t2 }: TypePair) => {
const r = validateConnectionTypes(
{ name: t1, isCollection: true, isCollectionOrScalar: false },
{ name: t2, isCollection: false, isCollectionOrScalar: false }
{ name: t2, isCollection: true, isCollectionOrScalar: false }
);
expect(r).toBe(true);
});

30
invokeai/frontend/web/src/features/nodes/store/util/validateConnectionTypes.ts
View File

@ -40,18 +40,25 @@ export const validateConnectionTypes = (sourceType: FieldType, targetType: Field
const isCollectionToGenericCollection = targetType.name === 'CollectionField' && sourceType.isCollection;
const areBothTypesSingle =
!sourceType.isCollection &&
!sourceType.isCollectionOrScalar &&
!targetType.isCollection &&
!targetType.isCollectionOrScalar;
const isSourceScalar = !sourceType.isCollection && !sourceType.isCollectionOrScalar;
const isTargetScalar = !targetType.isCollection && !targetType.isCollectionOrScalar;
const isScalarToScalar = isSourceScalar && isTargetScalar;
const isScalarToCollectionOrScalar = isSourceScalar && targetType.isCollectionOrScalar;
const isCollectionToCollection = sourceType.isCollection && targetType.isCollection;
const isCollectionToCollectionOrScalar = sourceType.isCollection && targetType.isCollectionOrScalar;
const isCollectionOrScalarToCollectionOrScalar = sourceType.isCollectionOrScalar && targetType.isCollectionOrScalar;
const isPluralityMatch =
isScalarToScalar ||
isCollectionToCollection ||
isCollectionToCollectionOrScalar ||
isCollectionOrScalarToCollectionOrScalar ||
isScalarToCollectionOrScalar;
const isIntToFloat = areBothTypesSingle && sourceType.name === 'IntegerField' && targetType.name === 'FloatField';
const isIntToFloat = sourceType.name === 'IntegerField' && targetType.name === 'FloatField';
const isIntToString = sourceType.name === 'IntegerField' && targetType.name === 'StringField';
const isFloatToString = sourceType.name === 'FloatField' && targetType.name === 'StringField';
const isIntOrFloatToString =
areBothTypesSingle &&
(sourceType.name === 'IntegerField' || sourceType.name === 'FloatField') &&
targetType.name === 'StringField';
const isSubTypeMatch = isPluralityMatch && (isIntToFloat || isIntToString || isFloatToString);
const isTargetAnyType = targetType.name === 'AnyField';
@ -62,8 +69,7 @@ export const validateConnectionTypes = (sourceType: FieldType, targetType: Field
isAnythingToCollectionOrScalarOfSameBaseType ||
isGenericCollectionToAnyCollectionOrCollectionOrScalar ||
isCollectionToGenericCollection ||
isIntToFloat ||
isIntOrFloatToString ||
isSubTypeMatch ||
isTargetAnyType
);
};