Hack for truncate test: wait 2 seconds for truncate operation

colinmarc · Mar 29, 2017 · 7779883 · 7779883
1 parent 48afdbc
commit 7779883
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/file_writer_test.go b/file_writer_test.go
@@ -348,7 +348,9 @@ func TestFileAppendRepeatedly(t *testing.T) {
 }
 
 func TestFileTruncate(t *testing.T) {
-	t.Skip("Truncate not support in Hadoop-2.6")
+	if os.Getenv("HADOOP_DISTRO") == "cdh" {
+		t.Skip("Truncate not support in Hadoop-2.6")
+	}
 	client := getClient(t)
 
 	baleet(t, "/_test/create/5.txt")
@@ -363,6 +365,8 @@ func TestFileTruncate(t *testing.T) {
 	writer.Close()
 	err = writer.Truncate(3)
 	require.NoError(t, err)
+	time.Sleep(3 * time.Second)
+
 	stat, err := client.Stat("/_test/create/5.txt")
 	require.NoError(t, err)
 	time.Sleep(time.Second)

diff --git a/truncate_test.go b/truncate_test.go
@@ -3,23 +3,27 @@ package hdfs
 import (
 	"os"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
 func TestTruncate(t *testing.T) {
-	t.Skip("Truncate not support in Hadoop-2.6")
+	if os.Getenv("HADOOP_DISTRO") == "cdh" {
+		t.Skip("Truncate not support in Hadoop-2.6")
+	}
 	client := getClient(t)
 
 	baleet(t, "/_test/truncate/1.txt")
 	mkdirp(t, "/_test/truncate")
 	writer, _ := client.Create("/_test/truncate/1.txt")
 	_, _ = writer.Write([]byte("foobar\nfoobar\n"))
- 	_ = writer.Close()
+	_ = writer.Close()
 
 	err := client.Truncate("/_test/truncate/1.txt", 4)
 	require.NoError(t, err)
+	time.Sleep(3 * time.Second)
 
 	stat, err := client.Stat("/_test/truncate/1.txt")
 	require.NoError(t, err)