From cf9e53573ed0fc46f8bf83c299afe99098de5465 Mon Sep 17 00:00:00 2001
From: Kusal Kithul-Godage <git@kusal.io>
Date: Tue, 14 Nov 2023 01:45:43 +1100
Subject: [PATCH] WW-5363 Add test coverage

---
 .../velocity/StrutsVelocityContextTest.java   | 88 +++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 plugins/velocity/src/test/java/org/apache/struts2/views/velocity/StrutsVelocityContextTest.java

diff --git a/plugins/velocity/src/test/java/org/apache/struts2/views/velocity/StrutsVelocityContextTest.java b/plugins/velocity/src/test/java/org/apache/struts2/views/velocity/StrutsVelocityContextTest.java
new file mode 100644
index 000000000..9d78ae1d5
--- /dev/null
+++ b/plugins/velocity/src/test/java/org/apache/struts2/views/velocity/StrutsVelocityContextTest.java
@@ -0,0 +1,88 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.struts2.views.velocity;
+
+import com.opensymphony.xwork2.util.ValueStack;
+import org.apache.velocity.VelocityContext;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnit;
+import org.mockito.junit.MockitoRule;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static java.util.Collections.singletonList;
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.when;
+
+public class StrutsVelocityContextTest {
+
+    @Rule
+    public MockitoRule mockitoRule = MockitoJUnit.rule();
+
+    private StrutsVelocityContext strutsVelocityContext;
+
+    @Mock
+    private VelocityContext chainedContext;
+
+    @Mock
+    private ValueStack stack;
+
+    private Map<String, Object> stackContext;
+
+    @Before
+    public void setUp() throws Exception {
+        stackContext = new HashMap<>();
+        when(stack.getContext()).thenReturn(stackContext);
+        strutsVelocityContext = new StrutsVelocityContext(singletonList(chainedContext), stack);
+    }
+
+    @Test
+    public void getChainedValue() {
+        when(chainedContext.internalGet("foo")).thenReturn("bar");
+        assertEquals("bar", strutsVelocityContext.internalGet("foo"));
+    }
+
+    @Test
+    public void getStackValue() {
+        when(stack.findValue("foo")).thenReturn("bar");
+        assertEquals("bar", strutsVelocityContext.internalGet("foo"));
+    }
+
+    @Test
+    public void getStackContextValue() {
+        stackContext.put("foo", "bar");
+        assertEquals("bar", strutsVelocityContext.internalGet("foo"));
+    }
+
+    @Test
+    public void getValuePrecedence() {
+        when(chainedContext.internalGet("foo")).thenReturn("bar");
+        assertEquals("bar", strutsVelocityContext.internalGet("foo"));
+
+        when(stack.findValue("foo")).thenReturn("baz");
+        assertEquals("bar", strutsVelocityContext.internalGet("foo"));
+
+        stackContext.put("foo", "qux");
+        assertEquals("bar", strutsVelocityContext.internalGet("foo"));
+    }
+}