add TestAccounts

geth/account/accounts_test.go

@@ -9,6 +9,7 @@ import (
 	"reflect"
 	"testing"
 
+	"github.com/ethereum/go-ethereum/accounts"
 	"github.com/ethereum/go-ethereum/accounts/keystore"
 	gethcommon "github.com/ethereum/go-ethereum/common"
 	whisper "github.com/ethereum/go-ethereum/whisper/whisperv5"
@@ -132,6 +133,7 @@ func TestManagerTestSuite(t *testing.T) {
 		password:       "test-password",
 		keyStore:       keyStore,
 		shh:            whisper.New(nil),
+		gethAccManager: accounts.NewManager(),
 	})
 }
 
@@ -147,6 +149,7 @@ type ManagerTestSuite struct {
 	password       string
 	keyStore       *keystore.KeyStore
 	shh            *whisper.Whisper
+	gethAccManager *accounts.Manager
 }
 
 // reinitMock is for reassigning a new mock node manager to account manager.
@@ -402,3 +405,34 @@ func (s *ManagerTestSuite) TestLogout() {
 	err = s.accManager.Logout()
 	s.Error(err)
 }
+
+func (s *ManagerTestSuite) TestAccounts() {
+	s.reinitMock()
+
+	// Create and select an account
+	s.nodeManager.EXPECT().AccountKeyStore().Return(s.keyStore, nil)
+	addr, _, _, err := s.accManager.CreateAccount(s.password)
+	s.NoError(err)
+	s.nodeManager.EXPECT().AccountKeyStore().Return(s.keyStore, nil).AnyTimes()
+	s.nodeManager.EXPECT().WhisperService().Return(s.shh, nil).AnyTimes()
+	err = s.accManager.SelectAccount(addr, s.password)
+	s.NoError(err)
+
+	// Success
+	s.nodeManager.EXPECT().AccountManager().Return(s.gethAccManager, nil)
+	accs, err := s.accManager.Accounts()
+	s.NoError(err)
+	s.NotNil(accs)
+
+	// Can't get an account manager
+	s.nodeManager.EXPECT().AccountManager().Return(nil, errors.New("Can't return an account manager"))
+	_, err = s.accManager.Accounts()
+	s.Error(err)
+
+	// Selected account is nil but doesn't fail
+	s.accManager.selectedAccount = nil
+	s.nodeManager.EXPECT().AccountManager().Return(s.gethAccManager, nil)
+	accs, err = s.accManager.Accounts()
+	s.NoError(err)
+	s.NotNil(accs)
+}