optional cpp generation

Author: kris-jusiak

README.md

@@ -66,12 +66,15 @@ find project -iname "*.h" -or -iname "*.hpp" | xargs "project/externals/gmock.py
 # guard: header guard
 # interface: interface class
 # mock_methods: generated gmock methods
+# generated_dir: generated dir
+# mock_file_hpp: mock header file 
+# mock_file_cpp: mock source file 
 
 mock_file_hpp = "%(interface)sMock.hpp"
 
 file_template_hpp = """\
 /*
- * file generated by gmock: %(mock_file)s
+ * file generated by gmock: %(mock_file_hpp)s
  */
 #ifndef %(guard)s
 #define %(guard)s
@@ -93,7 +96,7 @@ public:
 
 """
 
-mock_file_cpp = None
+mock_file_cpp = ""
 file_template_cpp = None
 
 ```

gmock.conf

@@ -4,6 +4,9 @@
 # guard: header guard
 # interface: interface class
 # mock_methods: generated gmock methods
+# generated_dir: generated dir
+# mock_file_hpp: mock header file 
+# mock_file_cpp: mock source file 
 
 mock_file_hpp = "%(interface)sMock.hpp"
 
@@ -31,6 +34,6 @@ public:
 
 """
 
-mock_file_cpp = None
+mock_file_cpp = ""
 file_template_cpp = None
 

gmock.py

@@ -186,33 +186,42 @@ def __get_mock_methods(self, node, mock_methods, class_decl = ""):
         else:
             [self.__get_mock_methods(c, mock_methods, class_decl) for c in node.get_children()]
 
-    def __init__(self, cursor, decl, path, mock_file_hpp, file_template_hpp):
+    def __generate_file(self, decl, mock_methods, file_type, file_template_type):
+        interface = decl.split("::")[-1]
+        mock_file = {
+            'hpp' : self.mock_file_hpp % { 'interface' : interface },
+            'cpp' : self.mock_file_cpp % { 'interface' : interface },
+        }
+        with open(self.path + "/" + mock_file[file_type], 'w') as file:
+            file.write(file_template_type % {
+                'mock_file_hpp' : mock_file['hpp'],
+                'mock_file_cpp' : mock_file['cpp'],
+                'generated_dir' : self.path,
+                'guard' : mock_file[file_type].replace('.', '_').upper(),
+                'dir' : os.path.dirname(mock_methods[0]),
+                'file' : os.path.basename(mock_methods[0]),
+                'namespaces_begin' : self.__pretty_namespaces_begin(decl),
+                'interface' : interface,
+                'mock_methods' : self.__pretty_mock_methods(mock_methods[1:]),
+                'namespaces_end' : self.__pretty_namespaces_end(decl)
+            })
+
+    def __init__(self, cursor, decl, path, mock_file_hpp, file_template_hpp, mock_file_cpp, file_template_cpp):
         self.cursor = cursor
         self.decl = decl
         self.path = path
         self.mock_file_hpp = mock_file_hpp
         self.file_template_hpp = file_template_hpp
+        self.mock_file_cpp = mock_file_cpp
+        self.file_template_cpp = file_template_cpp
 
     def generate(self):
         mock_methods = {}
         self.__get_mock_methods(self.cursor, mock_methods)
         for decl, mock_methods in mock_methods.iteritems():
             if len(mock_methods) > 0:
-                interface = decl.split("::")[-1]
-
-                if self.mock_file_hpp != None:
-                    mock_file_hpp = self.mock_file_hpp % { 'interface' : interface }
-                    with open(self.path + "/" + mock_file_hpp, 'w') as file:
-                        file.write(self.file_template_hpp % {
-                            'mock_file_hpp' : mock_file_hpp,
-                            'guard' : mock_file_hpp.replace('.', '_').upper(),
-                            'dir' : os.path.dirname(mock_methods[0]),
-                            'file' : os.path.basename(mock_methods[0]),
-                            'namespaces_begin' : self.__pretty_namespaces_begin(decl),
-                            'interface' : interface,
-                            'mock_methods' : self.__pretty_mock_methods(mock_methods[1:]),
-                            'namespaces_end' : self.__pretty_namespaces_end(decl)
-                        })
+                self.file_template_hpp != None and self.__generate_file(decl, mock_methods, 'hpp', self.file_template_hpp)
+                self.file_template_cpp != None and self.__generate_file(decl, mock_methods, 'cpp', self.file_template_cpp)
         return 0
 
 def main(args):
@@ -247,7 +256,9 @@ def generate_includes(includes):
         decl = options.decl,
         path = options.path,
         mock_file_hpp = config['mock_file_hpp'],
-        file_template_hpp = config['file_template_hpp']
+        file_template_hpp = config['file_template_hpp'],
+        mock_file_cpp = config['mock_file_cpp'],
+        file_template_cpp = config['file_template_cpp']
     ).generate()
 
 if __name__ == "__main__":

test/gmock_test.py

@@ -56,14 +56,16 @@ def test_gmock_many_files(self):
 
     def test_gmock_custom_conf(self):
         self.assertEqual(0, gmock.main(['./gmock.py', '-c', 'test.conf', '-d', self.generated_dir, '-l', 'n1', 'given/I2.hpp']))
-        self.assertEqual(1, len([name for name in os.listdir(self.generated_dir)]))
+        self.assertEqual(2, len([name for name in os.listdir(self.generated_dir)]))
         self.assertTrue('I2_mock.hpp' in os.listdir(self.generated_dir))
+        self.assertTrue('I2_mock.cpp' in os.listdir(self.generated_dir))
         self.assertMocks();
 
     def test_gmock_long_args(self):
         self.assertEqual(0, gmock.main(['./gmock.py', '--config=test.conf', '--dir='+self.generated_dir, '--limit=n1', 'given/I2.hpp']))
-        self.assertEqual(1, len([name for name in os.listdir(self.generated_dir)]))
+        self.assertEqual(2, len([name for name in os.listdir(self.generated_dir)]))
         self.assertTrue('I2_mock.hpp' in os.listdir(self.generated_dir))
+        self.assertTrue('I2_mock.cpp' in os.listdir(self.generated_dir))
         self.assertMocks();
 
 if __name__ == '__main__':

test/test.conf

@@ -12,7 +12,8 @@ file_template_hpp = """\
 class %(interface)s_mock : public %(interface)s
 {
 public:
-    virtual ~%(interface)s_mock() { }
+    %(interface)s_mock();
+    virtual ~%(interface)s_mock();
 
 %(mock_methods)s
 };
@@ -23,6 +24,20 @@ public:
 
 """
 
-mock_file_cpp = None
-file_template_cpp = None
+mock_file_cpp = "%(interface)s_mock.cpp"
+
+file_template_cpp = """\
+/*
+ * file generated by gmock: %(mock_file_cpp)s
+ */
+#include "%(mock_file_hpp)s"
+
+%(namespaces_begin)s
+
+%(interface)s_mock::%(interface)s_mock() { }
+%(interface)s_mock::~%(interface)s_mock() { }
+
+%(namespaces_end)s
+
+"""
 

test/then/I2_mock.cpp

@@ -0,0 +1,12 @@
+/*
+ * file generated by gmock: I2_mock.cpp
+ */
+#include "I2_mock.hpp"
+
+namespace n1 {
+
+I2_mock::I2_mock() { }
+I2_mock::~I2_mock() { }
+
+} // namespace n1
+

test/then/I2_mock.hpp

@@ -9,7 +9,8 @@ namespace n1 {
 class I2_mock : public I2
 {
 public:
-    virtual ~I2_mock() { }
+    I2_mock();
+    virtual ~I2_mock();
 
     MOCK_CONST_METHOD0(f0, void());
     MOCK_METHOD1(f1, void(int));