1 files changed, 135 insertions, 0 deletions
diff --git a/xfa/fxfa/fm2js/cxfa_fmparse_unittest.cpp b/xfa/fxfa/fm2js/cxfa_fmparse_unittest.cpp
new file mode 100644
index 0000000000..a17808ae88
--- /dev/null
+++ b/xfa/fxfa/fm2js/cxfa_fmparse_unittest.cpp
@@ -0,0 +1,135 @@
+// Copyright 2017 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "xfa/fxfa/fm2js/cxfa_fmparse.h"
+
+#include <vector>
+
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/test_support.h"
+#include "third_party/base/ptr_util.h"
+
+TEST(CXFA_FMParseTest, Empty) {
+  CXFA_FMErrorInfo errorInfo;
+  auto parser = pdfium::MakeUnique<CXFA_FMParse>(L"", &errorInfo);
+  std::unique_ptr<CXFA_FMFunctionDefinition> ast = parser->Parse();
+  ASSERT(ast != nullptr);
+  EXPECT_TRUE(errorInfo.message.IsEmpty());
+
+  CFX_WideTextBuf buf;
+  EXPECT_TRUE(ast->ToJavaScript(buf));
+  // TODO(dsinclair): This is a little weird .....
+  EXPECT_EQ(L"// comments only", buf.AsStringC());
+}
+
+TEST(CXFA_FMParseTest, CommentOnlyIsError) {
+  CXFA_FMErrorInfo errorInfo;
+  auto parser = pdfium::MakeUnique<CXFA_FMParse>(L"; Just comment", &errorInfo);
+  std::unique_ptr<CXFA_FMFunctionDefinition> ast = parser->Parse();
+  ASSERT(ast != nullptr);
+  // TODO(dsinclair): This isn't allowed per the spec.
+  EXPECT_TRUE(errorInfo.message.IsEmpty());
+  // EXPECT_FALSE(errorInfo.message.IsEmpty());
+
+  CFX_WideTextBuf buf;
+  EXPECT_TRUE(ast->ToJavaScript(buf));
+  EXPECT_EQ(L"// comments only", buf.AsStringC());
+}
+
+TEST(CXFA_FMParseTest, CommentThenValue) {
+  CXFA_FMErrorInfo errorInfo;
+  auto parser =
+      pdfium::MakeUnique<CXFA_FMParse>(L"; Just comment\n12", &errorInfo);
+  std::unique_ptr<CXFA_FMFunctionDefinition> ast = parser->Parse();
+  ASSERT(ast != nullptr);
+  EXPECT_TRUE(errorInfo.message.IsEmpty());
+
+  CFX_WideTextBuf buf;
+  EXPECT_TRUE(ast->ToJavaScript(buf));
+  EXPECT_EQ(L"(\n"
+      L"function ()\n"
+      L"{\n"
+        L"var foxit_xfa_formcalc_runtime_func_return_value = null;\n"
+        L"foxit_xfa_formcalc_runtime_func_return_value = 12;\n"
+        L"return foxit_xfa_formcalc_runtime.get_fm_value("
+          L"foxit_xfa_formcalc_runtime_func_return_value);\n"
+      L"}\n"
+      L").call(this);\n", buf.AsStringC());
+}
+
+TEST(CXFA_FMParseTest, Parse) {
+  const wchar_t input[] = L"$ = Avg (-3, 5, -6, 12, -13);\n"
+      L"$ = Avg (Table2..Row[*].Cell1);\n"
+      L"\n"
+      L"if ($ ne -1)then\n"
+      L"  border.fill.color.value = \"255,64,64\";\n"
+      L"else\n"
+      L"  border.fill.color.value = \"20,170,13\";\n"
+      L"endif\n"
+      L"\n"
+      L"$";
+
+  CXFA_FMErrorInfo errorInfo;
+  auto parser = pdfium::MakeUnique<CXFA_FMParse>(input, &errorInfo);
+  std::unique_ptr<CXFA_FMFunctionDefinition> ast = parser->Parse();
+  ASSERT(ast != nullptr);
+  EXPECT_TRUE(errorInfo.message.IsEmpty());
+
+  CFX_WideTextBuf buf;
+  EXPECT_TRUE(ast->ToJavaScript(buf));
+  EXPECT_EQ(
+      L"(\nfunction ()\n{\n"
+      L"var foxit_xfa_formcalc_runtime_func_return_value = null;\n"
+      L"if (foxit_xfa_formcalc_runtime.is_fm_object(this))\n{\n"
+        L"foxit_xfa_formcalc_runtime.assign_value_operator(this, "
+          L"foxit_xfa_formcalc_runtime.Avg("
+            L"foxit_xfa_formcalc_runtime.negative_operator(3), 5, "
+              L"foxit_xfa_formcalc_runtime.negative_operator(6), 12, "
+                L"foxit_xfa_formcalc_runtime.negative_operator(13)));\n"
+      L"}\n"
+      L"if (foxit_xfa_formcalc_runtime.is_fm_object(this))\n{\n"
+        L"foxit_xfa_formcalc_runtime.assign_value_operator(this, "
+          L"foxit_xfa_formcalc_runtime.Avg("
+            L"foxit_xfa_formcalc_runtime.dot_accessor("
+              L"foxit_xfa_formcalc_runtime.dotdot_accessor(Table2, \"Table2\", "
+                L"\"Row\", 1), \"\", \"Cell1\", 0, 0)));\n"
+      L"}\n"
+      L"if (foxit_xfa_formcalc_runtime.get_fm_value("
+        L"foxit_xfa_formcalc_runtime.notequality_operator(this, "
+          L"foxit_xfa_formcalc_runtime.negative_operator(1))))\n{\n"
+            L"if (foxit_xfa_formcalc_runtime.is_fm_object("
+              L"foxit_xfa_formcalc_runtime.dot_accessor("
+                L"foxit_xfa_formcalc_runtime.dot_accessor("
+                  L"foxit_xfa_formcalc_runtime.dot_accessor(border, "
+                    L"\"border\", \"fill\", 0, 0), \"\", \"color\", 0, 0), "
+                    L"\"\", \"value\", 0, 0)))\n{\n"
+                      L"foxit_xfa_formcalc_runtime.assign_value_operator("
+                        L"foxit_xfa_formcalc_runtime.dot_accessor("
+                          L"foxit_xfa_formcalc_runtime.dot_accessor("
+                            L"foxit_xfa_formcalc_runtime.dot_accessor("
+                              L"border, \"border\", \"fill\", 0, 0), \"\", "
+                              L"\"color\", 0, 0), \"\", \"value\", 0, 0), "
+                              L"\"255,64,64\");\n"
+            L"}\n"
+      L"}\nelse\n{\n"
+        L"if (foxit_xfa_formcalc_runtime.is_fm_object("
+          L"foxit_xfa_formcalc_runtime.dot_accessor("
+            L"foxit_xfa_formcalc_runtime.dot_accessor("
+              L"foxit_xfa_formcalc_runtime.dot_accessor("
+                L"border, \"border\", \"fill\", 0, 0), \"\", \"color\", "
+                L"0, 0), \"\", \"value\", 0, 0)))\n{\n"
+                  L"foxit_xfa_formcalc_runtime.assign_value_operator("
+                    L"foxit_xfa_formcalc_runtime.dot_accessor("
+                      L"foxit_xfa_formcalc_runtime.dot_accessor("
+                        L"foxit_xfa_formcalc_runtime.dot_accessor("
+                          L"border, \"border\", \"fill\", 0, 0), \"\", "
+                          L"\"color\", 0, 0), \"\", \"value\", 0, 0), "
+                          L"\"20,170,13\");\n"
+        L"}\n"
+      L"}\n"
+      L"foxit_xfa_formcalc_runtime_func_return_value = this;\n"
+      L"return foxit_xfa_formcalc_runtime.get_fm_value("
+        L"foxit_xfa_formcalc_runtime_func_return_value);\n"
+      L"}\n).call(this);\n", buf.AsStringC());
+}