Support custom default extension for FileNameOption

[alexxy/gromacs.git] / src / gromacs / options / tests / filenameoptionmanager.cpp

diff --git a/src/gromacs/options/tests/filenameoptionmanager.cpp b/src/gromacs/options/tests/filenameoptionmanager.cpp
index f0e2ecd5d83a5a9ab83238cc2ea1a29b4c14d456..8e2e5238834b00239cf5b0583b421e9370bb7558 100644 (file)
--- a/src/gromacs/options/tests/filenameoptionmanager.cpp
+++ b/src/gromacs/options/tests/filenameoptionmanager.cpp
@@ -45,6 +45,7 @@
 
 #include <gtest/gtest.h>
 
+#include "gromacs/fileio/filenm.h"
 #include "gromacs/options/filenameoption.h"
 #include "gromacs/options/options.h"
 #include "gromacs/options/optionsassigner.h"
@@ -103,6 +104,25 @@ TEST_F(FileNameOptionManagerTest, AddsMissingExtension)
     EXPECT_EQ("testfile.xtc", value);
 }
 
+TEST_F(FileNameOptionManagerTest, AddsMissingCustomDefaultExtension)
+{
+    std::string value;
+    ASSERT_NO_THROW_GMX(options_.addOption(
+                                FileNameOption("f").store(&value)
+                                    .filetype(gmx::eftTrajectory).outputFile()
+                                    .defaultType(efPDB)));
+
+    gmx::OptionsAssigner assigner(&options_);
+    EXPECT_NO_THROW_GMX(assigner.start());
+    EXPECT_NO_THROW_GMX(assigner.startOption("f"));
+    EXPECT_NO_THROW_GMX(assigner.appendValue("testfile"));
+    EXPECT_NO_THROW_GMX(assigner.finishOption());
+    EXPECT_NO_THROW_GMX(assigner.finish());
+    EXPECT_NO_THROW_GMX(options_.finish());
+
+    EXPECT_EQ("testfile.pdb", value);
+}
+
 TEST_F(FileNameOptionManagerTest, GivesErrorOnMissingInputFile)
 {
     std::string value;