| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CustomElementTestHelpers_h |
| #define CustomElementTestHelpers_h |
| |
| #include "bindings/core/v8/ExceptionState.h" |
| #include "core/dom/Document.h" |
| #include "core/dom/Element.h" |
| #include "core/dom/QualifiedName.h" |
| #include "core/html/HTMLDocument.h" |
| #include "platform/heap/Handle.h" |
| #include "wtf/text/AtomicString.h" |
| |
| #include <utility> |
| #include <vector> |
| |
| namespace blink { |
| |
| class CreateElement { |
| STACK_ALLOCATED() |
| public: |
| CreateElement(const AtomicString& localName) |
| : m_namespaceURI(HTMLNames::xhtmlNamespaceURI) |
| , m_localName(localName) |
| { |
| } |
| |
| CreateElement& inDocument(Document* document) |
| { |
| m_document = document; |
| return *this; |
| } |
| |
| CreateElement& inNamespace(const AtomicString& uri) |
| { |
| m_namespaceURI = uri; |
| return *this; |
| } |
| |
| CreateElement& withId(const AtomicString& id) |
| { |
| m_attributes.push_back(std::make_pair(HTMLNames::idAttr, id)); |
| return *this; |
| } |
| |
| CreateElement& withIsAttribute(const AtomicString& value) |
| { |
| m_attributes.push_back(std::make_pair(HTMLNames::isAttr, value)); |
| return *this; |
| } |
| |
| operator Element*() const |
| { |
| Document* document = m_document.get(); |
| if (!document) |
| document = HTMLDocument::create(); |
| NonThrowableExceptionState noExceptions; |
| Element* element = document->createElementNS( |
| m_namespaceURI, |
| m_localName, |
| noExceptions); |
| for (const auto& attribute : m_attributes) |
| element->setAttribute(attribute.first, attribute.second); |
| return element; |
| } |
| |
| private: |
| Member<Document> m_document; |
| AtomicString m_namespaceURI; |
| AtomicString m_localName; |
| std::vector<std::pair<QualifiedName, AtomicString>> m_attributes; |
| }; |
| |
| } // namespace blink |
| |
| #endif // CustomElementTestHelpers_h |