feat: multi elements support for tag

src/selector-tag.ts

@@ -1,9 +1,16 @@
-import {sanitizeSelectorItem} from './utilities-selectors'
-import {CssSelector} from './types'
+import {
+  sanitizeSelectorItem,
+  sanitizeSelectorNeedle
+} from './utilities-selectors';
+import {CssSelector, SelectorNeedle} from './types';
 
 /**
  * Get tag selector for an element.
  */
-export function getTagSelector (element: Element): Array<CssSelector> {
-  return [sanitizeSelectorItem(element.tagName.toLowerCase())]
+export function getTagSelector (needle: SelectorNeedle): Array<CssSelector> {
+  const elements = sanitizeSelectorNeedle(needle)
+  const selectors = [...new Set(elements.map((element) => {
+    return sanitizeSelectorItem(element.tagName.toLowerCase())
+  }))]
+  return (selectors.length === 0 || selectors.length > 1) ? [] : [selectors[0]]
 }

test/selector-tag.spec.js

@@ -17,7 +17,7 @@ describe('selector - tag', function () {
     root.innerHTML = '<div></div>'
     const element = root.firstChild
     const selector = getTagSelector(element)
-    assert.equal(selector, 'div')
+    assert.sameMembers(selector, ['div'])
     assert.equal(root.querySelector(selector), element)
   })
 
@@ -29,4 +29,11 @@ describe('selector - tag', function () {
     assert.equal(root.querySelector(selector), element)
   })
 
+  it('should generate tag selector for multiple elements', () => {
+    root.innerHTML = '<div></div><div></div>'
+    const elements = [...root.querySelectorAll('div')]
+    const result = getTagSelector(elements)
+    assert.sameMembers(result, ['div'])
+  })
+
 })