change type in test

algolia · Oct 23, 2020 · dc4ef68 · dc4ef68
1 parent 4b9f503
commit dc4ef68
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/packages/autocomplete-core/src/__tests__/getAutocompleteSetters.test.ts b/packages/autocomplete-core/src/__tests__/getAutocompleteSetters.test.ts
@@ -1,4 +1,5 @@
 import { createAutocomplete, AutocompleteSuggestion } from '..';
+import { InternalAutocompleteSource } from '../types';
 
 function createSuggestion<TItem extends { label: string }>(
   items: TItem[] = []
@@ -15,6 +16,26 @@ function createSuggestion<TItem extends { label: string }>(
   };
 }
 
+interface AutocompleteMultiSuggestion<TItem> {
+  source: InternalAutocompleteSource<TItem>;
+  items: TItem[][];
+}
+
+function createMultiSuggestion<TItem extends { label: string }>(
+  items: TItem[][] = []
+): AutocompleteMultiSuggestion<TItem> {
+  return {
+    source: {
+      getInputValue: ({ suggestion }) => suggestion.label,
+      getSuggestionUrl: () => undefined,
+      onHighlight: () => {},
+      onSelect: () => {},
+      getSuggestions: () => items,
+    },
+    items,
+  };
+}
+
 describe('createAutocomplete', () => {
   test('setHighlightedIndex', () => {
     const onStateChange = jest.fn();
@@ -95,10 +116,9 @@ describe('createAutocomplete', () => {
         onStateChange,
       });
 
-      // @ts-expect-error AutocompleteSuggestion has an array of items
       // while it also accepts a nested array.
       // There's only one type for both input and output however,
-      setSuggestions([createSuggestion([[{ label: 'hi' }]])]);
+      setSuggestions([createMultiSuggestion([[{ label: 'hi' }]])]);
 
       expect(onStateChange).toHaveBeenCalledWith({
         state: expect.objectContaining({