added enter pin in atm state floe

Author: codeimmortal

atm-machine/atm/atm.go

@@ -18,6 +18,7 @@ type ATM struct {
 	account         Account
 	uiOption        []string
 	WithdrawAs      *WithdrawNote
+	insertPin       ATMState
 	insertCard      ATMState
 	readCard        ATMState
 	selectAccount   ATMState
@@ -62,6 +63,12 @@ func (a *ATM) GetCardDetail() error {
 	return a.currentState.GetCardDetail()
 }
 
+func (a *ATM) InsertPin() error {
+	a.mu.Lock()
+	defer a.mu.Unlock()
+	return a.currentState.InsertPin()
+}
+
 func (a *ATM) DispenserAmount() error {
 	a.mu.Lock()
 	if err := a.currentState.DispenserAmount(); err != nil {
@@ -109,6 +116,10 @@ func NewATM() *ATM {
 		atm: atm,
 	}
 
+	atm.insertPin = &InsertPin{
+		atm: atm,
+	}
+
 	atm.selectAccount = &SelectAccount{
 		atm: atm,
 	}

atm-machine/atm/insert_pin.go

@@ -19,6 +19,7 @@ func (s *InsertPin) InsertPin() error {
 	fmt.Scanf("%s", &pin)
 
 	if pin == "1111" {
+		s.atm.SetState(s.atm.selectAccount)
 		return nil
 	} else {
 		return errors.New("Pin is not valid")

atm-machine/atm/read_card.go

@@ -11,7 +11,7 @@ func (d *ReadCard) GetCardDetail() error {
 		d.atm.SetState(d.atm.insertCard)
 	}
 	d.atm.card = card
-	d.atm.SetState(d.atm.selectAccount)
+	d.atm.SetState(d.atm.insertPin)
 	return nil
 }
 

atm-machine/main.go

@@ -5,6 +5,8 @@ import (
 	"fmt"
 )
 
+// flow of ATM
+// insertCard state -> readCard state -> insertPin state -> selectAccount state -> withdrawAmount state
 func main() {
 	atm := atm.NewATM()
 
@@ -29,6 +31,10 @@ func main() {
 
 		atm.Execute(atm.GetCardDetail)
 
+		fmt.Printf("\nATM state:%s\n", atm.StateName())
+
+		atm.Execute(atm.InsertPin)
+
 		fmt.Printf("\nATM state:%s\n", atm.StateName())
 		// insert card
 		atm.Execute(atm.SelectAccount)