Added tests for file generator