Add kyb to kyc verified check, add tests.

eth/ethadmin/camino_ethadmin.go

@@ -32,8 +32,12 @@ var (
 )
 
 const (
-	KYC_VERIFIED = 1
-	KYC_EXPIRED  = 2
+	NOT_VERIFIED = 0b0000 // 0
+	KYC_VERIFIED = 0b0001 // 1, bit 0
+	KYC_EXPIRED  = 0b0010 // 2, bit 1
+	KYB_VERIFIED = 0b1000 // 8, bit 3
+
+	VERIFIED = KYC_VERIFIED | KYB_VERIFIED
 )
 
 type AdminControllerBackend interface {
@@ -139,16 +143,18 @@ func (a *AdminController) GetFixedBaseFee(head *types.Header, state admin.StateD
 
 func (a *AdminController) KycVerified(head *types.Header, state admin.StateDB, addr common.Address) bool {
 	if a.cfg.IsSunrisePhase0(head.Time) {
-		// Calculate storage position
-		storagePos := crypto.Keccak256Hash(append(addr.Hash().Bytes(), common.HexToHash("0x2").Bytes()...))
 		// Get the KYC states
-		kycStates := new(big.Int).SetBytes(state.GetState(contractAddr, storagePos).Bytes()).Uint64()
+		kycStates := new(big.Int).SetBytes(state.GetState(contractAddr, kycStoragePosition(addr)).Bytes()).Uint64()
 		// Return true if KYC flag is set
-		return (kycStates & KYC_VERIFIED) != 0
+		return (kycStates & VERIFIED) != 0
 	}
 	return true
 }
 
 func (a *AdminController) inScanRange(head *types.Header) bool {
 	return head.Number.Cmp(&a.lastChangeHeight) >= 0 && head.Number.Cmp(&a.scanHeight) <= 0
 }
+
+func kycStoragePosition(addr common.Address) common.Hash {
+	return crypto.Keccak256Hash(append(addr.Hash().Bytes(), common.HexToHash("0x2").Bytes()...))
+}

eth/ethadmin/camino_ethadmin_test.go

@@ -0,0 +1,74 @@
+package ethadmin
+
+import (
+	"math/big"
+	"testing"
+
+	"github.com/ava-labs/coreth/core/admin"
+	"github.com/ava-labs/coreth/core/types"
+	"github.com/ava-labs/coreth/params"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/stretchr/testify/require"
+	"go.uber.org/mock/gomock"
+)
+
+func TestKycVerified(t *testing.T) {
+	address := common.Address{1}
+	sunriseTimestamp := uint64(1000)
+	sunriseActiveHeader := &types.Header{
+		Time: sunriseTimestamp,
+	}
+
+	adminCtrl := NewController(nil, &params.ChainConfig{
+		SunrisePhase0BlockTimestamp: &sunriseTimestamp,
+	})
+
+	tests := map[string]struct {
+		stateDB        func(c *gomock.Controller) *admin.MockStateDB
+		header         *types.Header
+		address        common.Address
+		expectedResult bool
+	}{
+		"Not verified": {
+			stateDB: func(c *gomock.Controller) *admin.MockStateDB {
+				stateDB := admin.NewMockStateDB(c)
+				stateDB.EXPECT().GetState(contractAddr, kycStoragePosition(address)).
+					Return(common.BigToHash(big.NewInt(NOT_VERIFIED)))
+				return stateDB
+			},
+			header:         sunriseActiveHeader,
+			address:        address,
+			expectedResult: false,
+		},
+		"KYC verified": {
+			stateDB: func(c *gomock.Controller) *admin.MockStateDB {
+				stateDB := admin.NewMockStateDB(c)
+				stateDB.EXPECT().GetState(contractAddr, kycStoragePosition(address)).
+					Return(common.BigToHash(big.NewInt(KYC_VERIFIED)))
+				return stateDB
+			},
+			header:         sunriseActiveHeader,
+			address:        address,
+			expectedResult: true,
+		},
+		"KYB verified": {
+			stateDB: func(c *gomock.Controller) *admin.MockStateDB {
+				stateDB := admin.NewMockStateDB(c)
+				stateDB.EXPECT().GetState(contractAddr, kycStoragePosition(address)).
+					Return(common.BigToHash(big.NewInt(KYB_VERIFIED)))
+				return stateDB
+			},
+			header:         sunriseActiveHeader,
+			address:        address,
+			expectedResult: true,
+		},
+	}
+
+	for name, tt := range tests {
+		t.Run(name, func(t *testing.T) {
+			c := gomock.NewController(t)
+			result := adminCtrl.KycVerified(tt.header, tt.stateDB(c), tt.address)
+			require.Equal(t, tt.expectedResult, result)
+		})
+	}
+}