From d318f612a9101625cba575a44428ced9d2aafa84 Mon Sep 17 00:00:00 2001
From: Matthias Loibl <mail@matthiasloibl.com>
Date: Sun, 6 Nov 2016 09:47:25 +0100
Subject: [PATCH] Add tests for CurrentUsername() (#88)

---
 modules/user/user.go      |  4 +---
 modules/user/user_test.go | 22 ++++++++++++++++++++++
 2 files changed, 23 insertions(+), 3 deletions(-)
 create mode 100644 modules/user/user_test.go

diff --git a/modules/user/user.go b/modules/user/user.go
index 4415632ee..972629eae 100644
--- a/modules/user/user.go
+++ b/modules/user/user.go
@@ -4,9 +4,7 @@
 
 package user
 
-import (
-	"os"
-)
+import "os"
 
 func CurrentUsername() string {
 	curUserName := os.Getenv("USER")
diff --git a/modules/user/user_test.go b/modules/user/user_test.go
new file mode 100644
index 000000000..d7df15153
--- /dev/null
+++ b/modules/user/user_test.go
@@ -0,0 +1,22 @@
+package user
+
+import (
+	"os"
+	"testing"
+)
+
+func TestCurrentUsername(t *testing.T) {
+	os.Setenv("USER", "")
+	os.Setenv("USERNAME", "foobar")
+
+	user := CurrentUsername()
+	if user != "foobar" {
+		t.Errorf("expected foobar as user, got: %s", user)
+	}
+
+	os.Setenv("USER", "gitea")
+	user = CurrentUsername()
+	if user != "gitea" {
+		t.Errorf("expected gitea as user, got: %s", user)
+	}
+}