From d906d16d1594f5aec0680df39f208f3136a4c273 Mon Sep 17 00:00:00 2001
From: Ryan Uber <ru@ryanuber.com>
Date: Sat, 13 Sep 2014 12:12:24 -0700
Subject: [PATCH] agent: add tests for keyring presence checks

---
 command/agent/config_test.go | 53 ++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/command/agent/config_test.go b/command/agent/config_test.go
index 13e8634ce4..19097e2c3c 100644
--- a/command/agent/config_test.go
+++ b/command/agent/config_test.go
@@ -1030,3 +1030,56 @@ func TestReadConfigPaths_dir(t *testing.T) {
 		t.Fatalf("bad: %#v", config)
 	}
 }
+
+func TestKeyringFileExists(t *testing.T) {
+	tempDir, err := ioutil.TempDir("", "consul")
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+	defer os.RemoveAll(tempDir)
+
+	fileLAN := filepath.Join(tempDir, SerfLANKeyring)
+	fileWAN := filepath.Join(tempDir, SerfWANKeyring)
+	config := &Config{DataDir: tempDir, Server: true}
+
+	// Returns false if we are a server and no keyring files present
+	if config.keyringFileExists() {
+		t.Fatalf("should return false")
+	}
+
+	// Returns false if we are a client and no keyring files present
+	config.Server = false
+	if config.keyringFileExists() {
+		t.Fatalf("should return false")
+	}
+
+	// Returns true if we are a client and the lan file exists
+	if err := ioutil.WriteFile(fileLAN, nil, 0600); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+	if !config.keyringFileExists() {
+		t.Fatalf("should return true")
+	}
+
+	// Returns true if we are a server and only the lan file exists
+	config.Server = true
+	if !config.keyringFileExists() {
+		t.Fatalf("should return true")
+	}
+
+	// Returns true if we are a server and both files exist
+	if err := ioutil.WriteFile(fileWAN, nil, 0600); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+	if !config.keyringFileExists() {
+		t.Fatalf("should return true")
+	}
+
+	// Returns true if we are a server and only the wan file exists
+	if err := os.Remove(fileLAN); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+	if !config.keyringFileExists() {
+		t.Fatalf("should return true")
+	}
+}