- $this->assertEquals('entity_link', $data['entity_test']['view_entity_test']['field']['id']);
- $this->assertEquals('entity_link_edit', $data['entity_test']['edit_entity_test']['field']['id']);
- $this->assertEquals('entity_link_delete', $data['entity_test']['delete_entity_test']['field']['id']);
+ foreach (['entity_test', 'entity_test_revision'] as $table_name) {
+ $this->assertEquals('entity_link', $data[$table_name]['view_entity_test']['field']['id']);
+ $this->assertEquals('entity_link_edit', $data[$table_name]['edit_entity_test']['field']['id']);
+ $this->assertEquals('entity_link_delete', $data[$table_name]['delete_entity_test']['field']['id']);
+ }