From d1ca2c6f838fe7428b7da6f62ebf293d4d91b44b Mon Sep 17 00:00:00 2001
From: Anton Kaliaev <anton.kalyaev@gmail.com>
Date: Fri, 28 Jul 2017 11:31:50 -0400
Subject: [PATCH] [common] add a test for WriteFileAtomic

---
 common/os_test.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 common/os_test.go

diff --git a/common/os_test.go b/common/os_test.go
new file mode 100644
index 00000000..05359e36
--- /dev/null
+++ b/common/os_test.go
@@ -0,0 +1,29 @@
+package common
+
+import (
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"os"
+	"testing"
+	"time"
+)
+
+func TestWriteFileAtomic(t *testing.T) {
+	data := []byte("Becatron")
+	fname := fmt.Sprintf("/tmp/write-file-atomic-test-%v.txt", time.Now().UnixNano())
+	err := WriteFileAtomic(fname, data, 0664)
+	if err != nil {
+		t.Fatal(err)
+	}
+	rData, err := ioutil.ReadFile(fname)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !bytes.Equal(data, rData) {
+		t.Fatalf("data mismatch: %v != %v", data, rData)
+	}
+	if err := os.Remove(fname); err != nil {
+		t.Fatal(err)
+	}
+}