From 3e88fd2d5e9ce8b8deee707e4de5c4b194b21059 Mon Sep 17 00:00:00 2001
From: geek <geek@loafle.com>
Date: Tue, 27 Jun 2017 11:32:24 +0900
Subject: [PATCH] email modify

---
 proxy/email/email_service.go      |  6 ++++++
 proxy/email/email_service_test.go | 14 +++++++++++++-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/proxy/email/email_service.go b/proxy/email/email_service.go
index 5c8ed61..b03287d 100644
--- a/proxy/email/email_service.go
+++ b/proxy/email/email_service.go
@@ -50,6 +50,7 @@ func NewEmail(member *member.Member, subject string) *Email {
 }
 
 type EmailService struct {
+	email *Email
 }
 
 func NewEmailService() *EmailService {
@@ -83,6 +84,11 @@ func (es *EmailService) getSendMailMessage(e *Email) string {
 	return message
 }
 
+func (es *EmailService) SetEmail(m *member.Member, sub string) {
+	e := NewEmail(m, sub)
+	es.email = e
+}
+
 func (es *EmailService) generationAuthToken(e *Email) string {
 	var tempToken string
 
diff --git a/proxy/email/email_service_test.go b/proxy/email/email_service_test.go
index 6ff98ef..6e05ad8 100644
--- a/proxy/email/email_service_test.go
+++ b/proxy/email/email_service_test.go
@@ -9,7 +9,7 @@ import (
 
 func getEmailObj() (*Email, *EmailService) {
 	es := NewEmailService()
-	m := member.Member{
+	m := &member.Member{
 		Email:       "geek@loafle.com",
 		Name:        "geek",
 		CompanyName: "loafle",
@@ -19,6 +19,18 @@ func getEmailObj() (*Email, *EmailService) {
 
 	return e, es
 }
+
+func TestNewEmail(t *testing.T) {
+	e := NewEmail(&member.Member{
+		Email:       "geek@loafle.com",
+		Name:        "geek",
+		CompanyName: "loafle",
+		Id:          json.Number("2"),
+	},"")
+
+	t.Log(e.Member.Email)
+}
+
 func TestSendEmailForAuth(t *testing.T) {
 	e, es := getEmailObj()