X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs2.git;a=blobdiff_plain;f=direct%2Ftimothy_tests%2Fquick_tests%2Ftest_yaffs_truncate.c;h=185e39d30496e02360cc48fe9a3592af9d22c0ca;hp=300c340b93e2b031bdfac412a3d4558a29fd3cbb;hb=121c055437f23d9d6bc7b57d36ecf128b447013f;hpb=26592c53da98ac887099559afbcb8dda05264e1c

diff --git a/direct/timothy_tests/quick_tests/test_yaffs_truncate.c b/direct/timothy_tests/quick_tests/test_yaffs_truncate.c
index 300c340..185e39d 100644
--- a/direct/timothy_tests/quick_tests/test_yaffs_truncate.c
+++ b/direct/timothy_tests/quick_tests/test_yaffs_truncate.c
@@ -1,24 +1,44 @@
+/*
+ * YAFFS: Yet another FFS. A NAND-flash specific file system.
+ *
+ * Copyright (C) 2002-2011 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_truncate.h"
+static int handle = -1;
 
-int test_yaffs_truncate(void){
-	int handle=test_open_file();
-	if (handle>0){
+int test_yaffs_truncate(void)
+{
+	handle = test_yaffs_open();
+	if (handle >= 0){
 		return yaffs_truncate(FILE_PATH,FILE_SIZE_TRUNCATED );
-	}
-	else {
-		printf("error opening file");
+	} else {
+		print_message("error opening file",2);
 		return -1;
 	}
 }
 
-int test_yaffs_truncate_clean(void){
+int test_yaffs_truncate_clean(void)
+{
 	/* change file size back to orignal size */
-	int handle=test_open_file();
-	if (handle>0){
-		return yaffs_truncate(FILE_PATH,FILE_SIZE );
-	}
-	else {
-		printf("error opening file in clean function");
+	int output=0;
+	if (handle >= 0){
+		output= yaffs_truncate(FILE_PATH,FILE_SIZE );
+		if (output>=0){
+			return yaffs_close(handle);
+		} else {
+			print_message("failed to truncate file\n",2);
+			return -1;
+		}
+	} else {
+		print_message("error opening file in clean function\n",2);
 		return -1;
 	}