fix imports

Signed-off-by: Seva Kaloshin <[email protected]>

Author: arzonus

service/sharddistributor/store/etcd/executorstore/etcdstore_test.go

@@ -6,8 +6,6 @@ import (
 	"testing"
 	"time"
 
-	types3 "github.com/uber/cadence/service/sharddistributor/store/etcd/types"
-
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"go.uber.org/fx/fxtest"
@@ -16,6 +14,7 @@ import (
 	"github.com/uber/cadence/common/types"
 	"github.com/uber/cadence/service/sharddistributor/store"
 	"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdkeys"
+	"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdtypes"
 	"github.com/uber/cadence/service/sharddistributor/store/etcd/executorstore/common"
 	"github.com/uber/cadence/service/sharddistributor/store/etcd/leaderstore"
 	"github.com/uber/cadence/service/sharddistributor/store/etcd/testhelper"
@@ -60,7 +59,7 @@ func TestRecordHeartbeat(t *testing.T) {
 	resp, err := tc.Client.Get(ctx, heartbeatKey)
 	require.NoError(t, err)
 	assert.Equal(t, int64(1), resp.Count, "Heartbeat key should exist")
-	assert.Equal(t, types3.FromTime(now), string(resp.Kvs[0].Value))
+	assert.Equal(t, etcdtypes.FromTime(now), string(resp.Kvs[0].Value))
 
 	resp, err = tc.Client.Get(ctx, stateKey)
 	require.NoError(t, err)
@@ -545,7 +544,7 @@ func TestShardStatisticsPersistence(t *testing.T) {
 	stats := store.ShardStatistics{SmoothedLoad: 12.5, LastUpdateTime: time.Unix(1234, 0).UTC(), LastMoveTime: time.Unix(5678, 0).UTC()}
 	shardStatsKey, err := etcdkeys.BuildShardKey(tc.EtcdPrefix, tc.Namespace, shardID, etcdkeys.ShardStatisticsKey)
 	require.NoError(t, err)
-	payload, err := json.Marshal(types3.FromShardStatistics(&stats))
+	payload, err := json.Marshal(etcdtypes.FromShardStatistics(&stats))
 	require.NoError(t, err)
 	_, err = tc.Client.Put(ctx, shardStatsKey, string(payload))
 	require.NoError(t, err)