Fix unit tests

WordPress · Jun 20, 2018 · 44ab993 · 44ab993
1 parent 3d1fd0c
commit 44ab993
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/core-blocks/video/test/index.js b/core-blocks/video/test/index.js
@@ -1,11 +1,27 @@
+/**
+ * External dependencies
+ */
+import { get, mapValues } from 'lodash';
+
 /**
  * Internal dependencies
  */
-import { name, settings } from '../';
+import { name, definition, implementation } from '../';
 import { blockEditRender } from '../../test/helpers';
 
 describe( 'core/video', () => {
 	test( 'block edit matches snapshot', () => {
+		const settings = {
+			...definition,
+			...implementation,
+			attributes: mapValues( definition.attributes, ( attribute, key ) => {
+				const implementationAttribute = get( implementation.attributes, [ key ], {} );
+				return {
+					...attribute,
+					...implementationAttribute,
+				};
+			} ),
+		};
 		const wrapper = blockEditRender( name, settings );
 
 		expect( wrapper ).toMatchSnapshot();