Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
Adam Lesinski | 467f171 | 2015-11-16 17:35:44 -0800 | [diff] [blame] | 17 | #include "xml/XmlDom.h" |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 18 | |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 19 | #include <string> |
| 20 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 21 | #include "io/StringInputStream.h" |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 22 | #include "test/Test.h" |
| 23 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 24 | using ::aapt::io::StringInputStream; |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 25 | using ::testing::Eq; |
Adam Lesinski | a45893a | 2017-05-30 15:19:02 -0700 | [diff] [blame] | 26 | using ::testing::NotNull; |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 27 | using ::testing::SizeIs; |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 28 | using ::testing::StrEq; |
Adam Lesinski | a45893a | 2017-05-30 15:19:02 -0700 | [diff] [blame] | 29 | |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 30 | namespace aapt { |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 31 | namespace xml { |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 32 | |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 33 | TEST(XmlDomTest, Inflate) { |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 34 | std::string input = R"(<?xml version="1.0" encoding="utf-8"?> |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 35 | <Layout xmlns:android="http://schemas.android.com/apk/res/android" |
| 36 | android:layout_width="match_parent" |
| 37 | android:layout_height="wrap_content"> |
| 38 | <TextView android:id="@+id/id" |
| 39 | android:layout_width="wrap_content" |
| 40 | android:layout_height="wrap_content" /> |
| 41 | </Layout>)"; |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 42 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 43 | StdErrDiagnostics diag; |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 44 | StringInputStream in(input); |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 45 | std::unique_ptr<XmlResource> doc = Inflate(&in, &diag, Source("test.xml")); |
Adam Lesinski | a45893a | 2017-05-30 15:19:02 -0700 | [diff] [blame] | 46 | ASSERT_THAT(doc, NotNull()); |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 47 | |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 48 | Element* el = doc->root.get(); |
| 49 | EXPECT_THAT(el->namespace_decls, SizeIs(1u)); |
| 50 | EXPECT_THAT(el->namespace_decls[0].uri, StrEq(xml::kSchemaAndroid)); |
| 51 | EXPECT_THAT(el->namespace_decls[0].prefix, StrEq("android")); |
Adam Lesinski | 75f3a55 | 2015-06-03 14:54:23 -0700 | [diff] [blame] | 52 | } |
| 53 | |
Adam Lesinski | 48448e8 | 2017-04-26 15:13:52 -0700 | [diff] [blame] | 54 | // Escaping is handled after parsing of the values for resource-specific values. |
| 55 | TEST(XmlDomTest, ForwardEscapes) { |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 56 | std::unique_ptr<XmlResource> doc = test::BuildXmlDom(R"( |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 57 | <element value="\?hello" pattern="\\d{5}">\\d{5}</element>)"); |
Adam Lesinski | ac6edc5 | 2017-03-02 19:31:28 -0800 | [diff] [blame] | 58 | |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 59 | Element* el = doc->root.get(); |
Adam Lesinski | ac6edc5 | 2017-03-02 19:31:28 -0800 | [diff] [blame] | 60 | |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 61 | Attribute* attr = el->FindAttribute({}, "pattern"); |
Adam Lesinski | a45893a | 2017-05-30 15:19:02 -0700 | [diff] [blame] | 62 | ASSERT_THAT(attr, NotNull()); |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 63 | EXPECT_THAT(attr->value, Eq("\\\\d{5}")); |
Adam Lesinski | ac6edc5 | 2017-03-02 19:31:28 -0800 | [diff] [blame] | 64 | |
Adam Lesinski | 48448e8 | 2017-04-26 15:13:52 -0700 | [diff] [blame] | 65 | attr = el->FindAttribute({}, "value"); |
Adam Lesinski | a45893a | 2017-05-30 15:19:02 -0700 | [diff] [blame] | 66 | ASSERT_THAT(attr, NotNull()); |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 67 | EXPECT_THAT(attr->value, Eq("\\?hello")); |
Adam Lesinski | ac6edc5 | 2017-03-02 19:31:28 -0800 | [diff] [blame] | 68 | |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 69 | ASSERT_THAT(el->children, SizeIs(1u)); |
| 70 | |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 71 | Text* text = xml::NodeCast<xml::Text>(el->children[0].get()); |
Adam Lesinski | a45893a | 2017-05-30 15:19:02 -0700 | [diff] [blame] | 72 | ASSERT_THAT(text, NotNull()); |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 73 | EXPECT_THAT(text->text, Eq("\\\\d{5}")); |
| 74 | } |
| 75 | |
| 76 | TEST(XmlDomTest, XmlEscapeSequencesAreParsed) { |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 77 | std::unique_ptr<XmlResource> doc = test::BuildXmlDom(R"(<element value=""" />)"); |
| 78 | Attribute* attr = doc->root->FindAttribute({}, "value"); |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 79 | ASSERT_THAT(attr, NotNull()); |
Adam Lesinski | fba0cf2 | 2017-06-29 17:53:36 -0700 | [diff] [blame] | 80 | EXPECT_THAT(attr->value, Eq("\"")); |
Adam Lesinski | ac6edc5 | 2017-03-02 19:31:28 -0800 | [diff] [blame] | 81 | } |
| 82 | |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 83 | class TestVisitor : public PackageAwareVisitor { |
| 84 | public: |
| 85 | using PackageAwareVisitor::Visit; |
| 86 | |
| 87 | void Visit(Element* el) override { |
| 88 | if (el->name == "View1") { |
| 89 | EXPECT_THAT(TransformPackageAlias("one", "local"), |
| 90 | Eq(make_value(ExtractedPackage{"com.one", false}))); |
| 91 | } else if (el->name == "View2") { |
| 92 | EXPECT_THAT(TransformPackageAlias("one", "local"), |
| 93 | Eq(make_value(ExtractedPackage{"com.one", false}))); |
| 94 | EXPECT_THAT(TransformPackageAlias("two", "local"), |
| 95 | Eq(make_value(ExtractedPackage{"com.two", false}))); |
| 96 | } else if (el->name == "View3") { |
| 97 | EXPECT_THAT(TransformPackageAlias("one", "local"), |
| 98 | Eq(make_value(ExtractedPackage{"com.one", false}))); |
| 99 | EXPECT_THAT(TransformPackageAlias("two", "local"), |
| 100 | Eq(make_value(ExtractedPackage{"com.two", false}))); |
| 101 | EXPECT_THAT(TransformPackageAlias("three", "local"), |
| 102 | Eq(make_value(ExtractedPackage{"com.three", false}))); |
| 103 | } |
| 104 | } |
| 105 | }; |
| 106 | |
| 107 | TEST(XmlDomTest, PackageAwareXmlVisitor) { |
| 108 | std::unique_ptr<XmlResource> doc = test::BuildXmlDom(R"( |
| 109 | <View1 xmlns:one="http://schemas.android.com/apk/res/com.one"> |
| 110 | <View2 xmlns:two="http://schemas.android.com/apk/res/com.two"> |
| 111 | <View3 xmlns:three="http://schemas.android.com/apk/res/com.three" /> |
| 112 | </View2> |
| 113 | </View1>)"); |
| 114 | |
| 115 | Debug::DumpXml(doc.get()); |
| 116 | TestVisitor visitor; |
| 117 | doc->root->Accept(&visitor); |
| 118 | } |
| 119 | |
| 120 | } // namespace xml |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 121 | } // namespace aapt |