X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs2.git;a=blobdiff_plain;f=direct%2Ftimothy_tests%2Fquick_tests%2Ftest_yaffs_ftruncate.c;h=cb380098be9c265f5df81ebb6822753b713f987c;hp=26a89a78f8ec46ba7708b87b5eefbf91de562788;hb=9032d048555cb5b86ff8c1b4319a856b43331377;hpb=26592c53da98ac887099559afbcb8dda05264e1c

diff --git a/direct/timothy_tests/quick_tests/test_yaffs_ftruncate.c b/direct/timothy_tests/quick_tests/test_yaffs_ftruncate.c
index 26a89a7..cb38009 100644
--- a/direct/timothy_tests/quick_tests/test_yaffs_ftruncate.c
+++ b/direct/timothy_tests/quick_tests/test_yaffs_ftruncate.c
@@ -1,25 +1,45 @@
+/*
+ * YAFFS: Yet another FFS. A NAND-flash specific file system.
+ *
+ * Copyright (C) 2002-2010 Aleph One Ltd.
+ *   for Toby Churchill Ltd and Brightstar Engineering
+ *
+ * Created by Timothy Manning <timothy@yaffs.net>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ */
+
 #include "test_yaffs_ftruncate.h"
 
-int test_yaffs_ftruncate(void){
-	int handle=test_open_file();
-	if (handle>0){
+static int handle = -1;
+
+int test_yaffs_ftruncate(void)
+{
+	handle = test_yaffs_open();
+	if (handle >= 0){
 		return yaffs_ftruncate(handle,FILE_SIZE_TRUNCATED );
-	}
-	else {
-		printf("error opening file");
+	} else {
+		print_message("error opening file\n", 2);
 		return -1;
 	}
 }
 
-int test_yaffs_ftruncate_clean(void){
+int test_yaffs_ftruncate_clean(void)
+{
 	/* change file size back to orignal size */
-	int handle=test_open_file();
-	if (handle>0){
-		return yaffs_ftruncate(handle,FILE_SIZE );
+	int output = 0;
+	if (handle >= 0){
+		output = yaffs_ftruncate(handle,FILE_SIZE );
+		if (output >= 0){
+			return yaffs_close(handle);
+		} else {
+			print_message("failed to truncate file\n", 2);
+			return -1;
+		}
+	} else {
+		/* the file was not opened so the file could not be truncated */
+		return 1;
 	}
-	else {
-		printf("error opening file in clean function");
-		return -1;
-	}
-
 }