1 files changed, 136 insertions, 0 deletions
diff --git a/FreeRTOS-Plus/Test/CMock/test/unit/cmock_generator_plugin_return_thru_ptr_test.rb b/FreeRTOS-Plus/Test/CMock/test/unit/cmock_generator_plugin_return_thru_ptr_test.rb
new file mode 100644
index 000000000..3c8b075da
--- /dev/null
+++ b/FreeRTOS-Plus/Test/CMock/test/unit/cmock_generator_plugin_return_thru_ptr_test.rb
@@ -0,0 +1,136 @@
+# ==========================================
+#   CMock Project - Automatic Mock Generation for C
+#   Copyright (c) 2007 Mike Karlesky, Mark VanderVoord, Greg Williams
+#   [Released under MIT License. Please refer to license.txt for details]
+# ==========================================
+
+require File.expand_path(File.dirname(__FILE__)) + "/../test_helper"
+require File.expand_path(File.dirname(__FILE__)) + '/../../lib/cmock_generator_plugin_return_thru_ptr'
+
+describe CMockGeneratorPluginReturnThruPtr, "Verify CMockGeneratorPluginReturnThruPtr Module" do
+
+  before do
+    create_mocks :config, :utils
+
+    # int *Oak(void)"
+    @void_func = {:name => "Oak", :args => [], :return => test_return[:int_ptr]}
+
+    # char *Maple(int blah)
+    @simple_func = {:name => "Maple",
+                    :args => [{:name => "blah", :type => "int", :ptr? => false}],
+                    :return  => test_return[:string],
+                    :contains_ptr? => false}
+
+    # void Pine(int chicken, const int beef, int *tofu)
+    @complex_func = {:name => "Pine",
+                     :args => [{ :type => "int",
+                                 :name => "chicken",
+                                 :ptr? => false,
+                               },
+                               { :type   => "const int*",
+                                 :name   => "beef",
+                                 :ptr?   => true,
+                                 :const? => true,
+                               },
+                               { :type => "int*",
+                                 :name => "tofu",
+                                 :ptr? => true,
+                               }],
+                     :return => test_return[:void],
+                     :contains_ptr? => true }
+
+    #no strict ordering
+    @cmock_generator_plugin_return_thru_ptr = CMockGeneratorPluginReturnThruPtr.new(@config, @utils)
+  end
+
+  after do
+  end
+
+  def simple_func_expect
+    @utils.expect :ptr_or_str?, false, ['int']
+  end
+
+  def complex_func_expect
+    @utils.expect :ptr_or_str?, false, ['int']
+    @utils.expect :ptr_or_str?, true, ['const int*']
+    @utils.expect :ptr_or_str?, true, ['int*']
+  end
+
+  it "have set up internal priority correctly on init" do
+    assert_equal(9, @cmock_generator_plugin_return_thru_ptr.priority)
+  end
+
+  it "not include any additional include files" do
+    assert(!@cmock_generator_plugin_return_thru_ptr.respond_to?(:include_files))
+  end
+
+  it "not add to typedef structure for functions of style 'int* func(void)'" do
+    returned = @cmock_generator_plugin_return_thru_ptr.instance_typedefs(@void_func)
+    assert_equal("", returned)
+  end
+
+  it "add to tyepdef structure mock needs of functions of style 'void func(int chicken, int* pork)'" do
+    complex_func_expect()
+    expected = "  char ReturnThruPtr_tofu_Used;\n" +
+               "  int* ReturnThruPtr_tofu_Val;\n" +
+               "  int ReturnThruPtr_tofu_Size;\n"
+    returned = @cmock_generator_plugin_return_thru_ptr.instance_typedefs(@complex_func)
+    assert_equal(expected, returned)
+  end
+
+  it "not add an additional mock interface for functions not containing pointers" do
+    simple_func_expect()
+    returned = @cmock_generator_plugin_return_thru_ptr.mock_function_declarations(@simple_func)
+    assert_equal("", returned)
+  end
+
+  it "add a mock function declaration only for non-const pointer arguments" do
+    complex_func_expect();
+
+    expected =
+      "#define Pine_ReturnThruPtr_tofu(tofu)" +
+      " Pine_CMockReturnMemThruPtr_tofu(__LINE__, tofu, sizeof(int))\n" +
+      "#define Pine_ReturnArrayThruPtr_tofu(tofu, cmock_len)" +
+      " Pine_CMockReturnMemThruPtr_tofu(__LINE__, tofu, (int)(cmock_len * (int)sizeof(*tofu)))\n" +
+      "#define Pine_ReturnMemThruPtr_tofu(tofu, cmock_size)" +
+      " Pine_CMockReturnMemThruPtr_tofu(__LINE__, tofu, cmock_size)\n" +
+      "void Pine_CMockReturnMemThruPtr_tofu(UNITY_LINE_TYPE cmock_line, int* tofu, int cmock_size);\n"
+
+    returned = @cmock_generator_plugin_return_thru_ptr.mock_function_declarations(@complex_func)
+    assert_equal(expected, returned)
+  end
+
+  it "add mock interfaces only for non-const pointer arguments" do
+    complex_func_expect();
+
+    expected =
+      "void Pine_CMockReturnMemThruPtr_tofu(UNITY_LINE_TYPE cmock_line, int* tofu, int cmock_size)\n" +
+      "{\n" +
+      "  CMOCK_Pine_CALL_INSTANCE* cmock_call_instance = " +
+      "(CMOCK_Pine_CALL_INSTANCE*)CMock_Guts_GetAddressFor(CMock_Guts_MemEndOfChain(Mock.Pine_CallInstance));\n" +
+      "  UNITY_TEST_ASSERT_NOT_NULL(cmock_call_instance, cmock_line, CMockStringPtrPreExp);\n" +
+      "  cmock_call_instance->ReturnThruPtr_tofu_Used = 1;\n" +
+      "  cmock_call_instance->ReturnThruPtr_tofu_Val = tofu;\n" +
+      "  cmock_call_instance->ReturnThruPtr_tofu_Size = cmock_size;\n" +
+      "}\n\n"
+
+    returned = @cmock_generator_plugin_return_thru_ptr.mock_interfaces(@complex_func).join("")
+    assert_equal(expected, returned)
+  end
+
+  it "add mock implementations only for non-const pointer arguments" do
+    complex_func_expect()
+
+    expected =
+      "  if (cmock_call_instance->ReturnThruPtr_tofu_Used)\n" +
+      "  {\n" +
+      "    UNITY_TEST_ASSERT_NOT_NULL(tofu, cmock_line, CMockStringPtrIsNULL);\n" +
+      "    memcpy((void*)tofu, (void*)cmock_call_instance->ReturnThruPtr_tofu_Val,\n" +
+      "      cmock_call_instance->ReturnThruPtr_tofu_Size);\n" +
+      "  }\n"
+
+    returned = @cmock_generator_plugin_return_thru_ptr.mock_implementation(@complex_func).join("")
+    assert_equal(expected, returned)
+  end
+
+end