1 files changed, 76 insertions, 0 deletions
diff --git a/xfa/fxfa/fm2js/cxfa_fmsimpleexpression_unittest.cpp b/xfa/fxfa/fm2js/cxfa_fmsimpleexpression_unittest.cpp
new file mode 100644
index 0000000000..f11459b107
--- /dev/null
+++ b/xfa/fxfa/fm2js/cxfa_fmsimpleexpression_unittest.cpp
@@ -0,0 +1,76 @@
+// Copyright 2016 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_fmsimpleexpression.h"
+
+#include <memory>
+#include <utility>
+
+#include "core/fxcrt/fx_string.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/test_support.h"
+#include "third_party/base/ptr_util.h"
+#include "xfa/fxfa/fm2js/cxfa_fmlexer.h"
+
+TEST(FMCallExpressionTest, more_than_32_arguments) {
+  // Use sign as it has 3 object parameters at positions 0, 5, and 6.
+  auto exp = pdfium::MakeUnique<CXFA_FMIdentifierExpression>(0, L"sign");
+
+  std::vector<std::unique_ptr<CXFA_FMSimpleExpression>> args;
+  for (size_t i = 0; i < 50; i++)
+    args.push_back(pdfium::MakeUnique<CXFA_FMSimpleExpression>(0, TOKnan));
+
+  CXFA_FMCallExpression callExp(0, std::move(exp), std::move(args), true);
+  CFX_WideTextBuf js;
+  callExp.ToJavaScript(js);
+
+  // Generate the result javascript string.
+  CFX_WideString result = L"sign(";
+  for (size_t i = 0; i < 50; i++) {
+    if (i > 0)
+      result += L", ";
+
+    result += L"foxit_xfa_formcalc_runtime.get_fm_";
+    // Object positions for sign() method.
+    if (i == 0 || i == 5 || i == 6)
+      result += L"jsobj()";
+    else
+      result += L"value()";
+  }
+  result += L")";
+
+  EXPECT_EQ(result.AsStringC(), js.AsStringC());
+}
+
+TEST(FMStringExpressionTest, Empty) {
+  CFX_WideTextBuf accumulator;
+  CXFA_FMStringExpression(1, CFX_WideStringC()).ToJavaScript(accumulator);
+  EXPECT_EQ(L"", accumulator.AsStringC());
+}
+
+TEST(FMStringExpressionTest, Short) {
+  CFX_WideTextBuf accumulator;
+  CXFA_FMStringExpression(1, L"a").ToJavaScript(accumulator);
+  EXPECT_EQ(L"a", accumulator.AsStringC());
+}
+
+TEST(FMStringExpressionTest, Medium) {
+  CFX_WideTextBuf accumulator;
+  CXFA_FMStringExpression(1, L".abcd.").ToJavaScript(accumulator);
+  EXPECT_EQ(L"\"abcd\"", accumulator.AsStringC());
+}
+
+TEST(FMStringExpressionTest, Long) {
+  CFX_WideTextBuf accumulator;
+  std::vector<CFX_WideStringC::UnsignedType> vec(140000, L'A');
+  CXFA_FMStringExpression(1, CFX_WideStringC(vec)).ToJavaScript(accumulator);
+  EXPECT_EQ(140000, accumulator.GetLength());
+}
+
+TEST(FMStringExpressionTest, Quoted) {
+  CFX_WideTextBuf accumulator;
+  CXFA_FMStringExpression(1, L".Simon says \"\"run\"\".")
+      .ToJavaScript(accumulator);
+  EXPECT_EQ(L"\"Simon says \\\"run\\\"\"", accumulator.AsStringC());
+}