[WIP] Custom element upgrades

third_party/WebKit/Source/core/dom/custom/CustomElementTestHelpers.h

+// 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