Some rbuild cleanup:

[reactos.git] / reactos / tools / rbuild / testsupportcode.cpp

diff --git a/reactos/tools/rbuild/testsupportcode.cpp b/reactos/tools/rbuild/testsupportcode.cpp
index c27cf7e..3d0fdb2 100644 (file)
--- a/reactos/tools/rbuild/testsupportcode.cpp
+++ b/reactos/tools/rbuild/testsupportcode.cpp
@@ -297,8 +297,8 @@ TestSupportCode::GetSourceFilenames ( string_list& list,
        const vector<CompilationUnit*>& compilationUnits = module.non_if_data.compilationUnits;
        for ( i = 0; i < compilationUnits.size (); i++ )
        {
-               FileLocation* sourceFileLocation = compilationUnits[i]->GetFilename ( NULL );
-               string filename = sourceFileLocation->filename;
+               const FileLocation* sourceFileLocation = compilationUnits[i]->GetFilename ();
+               string filename = sourceFileLocation->relative_path + sSep + sourceFileLocation->name;
                if ( !compilationUnits[i]->IsGeneratedFile () && IsTestFile ( filename ) )
                        list.push_back ( filename );
        }
@@ -317,8 +317,8 @@ TestSupportCode::GetSourceFilenames ( string_list& list,
                for ( j = 0; j < compilationUnits.size (); j++ )
                {
                        CompilationUnit& compilationUnit = *compilationUnits[j];
-                       FileLocation* sourceFileLocation = compilationUnits[j]->GetFilename ( NULL );
-                       string filename = sourceFileLocation->filename;
+                       const FileLocation* sourceFileLocation = compilationUnits[j]->GetFilename ();
+                       string filename = sourceFileLocation->relative_path + sSep + sourceFileLocation->name;
                        if ( !compilationUnit.IsGeneratedFile () && IsTestFile ( filename ) )
                                list.push_back ( filename );
                }