Add Unit Tests for std version

pavelliavonau · Feb 21, 2018 · bc91dce · bc91dce
1 parent 1bebd7d
commit bc91dce
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/test/test_flags.py b/test/test_flags.py
@@ -108,6 +108,33 @@ def test_define_linux_flags(self):
 
         cmakelists_test.close()
 
+    def test_define_linux_flags_with_std(self):
+        """Define Linux Flags"""
+
+        self.data_test['cmake'] = get_cmake_lists('./')
+        self.data_test['std'] = 'c++17'
+        under_test = Flags(self.data_test)
+        under_test.define_linux_flags()
+        self.data_test['cmake'].close()
+
+        cmakelists_test = open('%s/CMakeLists.txt' % self.cur_dir, 'r')
+        content_test = cmakelists_test.read()
+
+        self.assertTrue('-std=c++17' in content_test)
+        cmakelists_test.close()
+
+        self.data_test['cmake'] = get_cmake_lists('./')
+        self.data_test['std'] = 'c++19'
+        under_test = Flags(self.data_test)
+        under_test.define_linux_flags()
+        self.data_test['cmake'].close()
+
+        cmakelists_test = open('%s/CMakeLists.txt' % self.cur_dir, 'r')
+        content_test = cmakelists_test.read()
+
+        self.assertTrue('-std=c++11' in content_test)
+        cmakelists_test.close()
+
     def test_define_windows_flags(self):
         """Define Windows Flags"""