Move code block in core_test

cosmos · Nov 2, 2022 · 76ebbfd · 76ebbfd
1 parent 2496dfc
commit 76ebbfd
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/difference/core/driver/core_test.go b/tests/difference/core/driver/core_test.go
@@ -539,10 +539,6 @@ func (s *CoreSuite) executeTraces() {
 		s.Run(fmt.Sprintf("Trace num: %d", i), func() {
 			// Setup a new pair of chains for each trace
 			s.SetupTest()
-			// TODO: move these initialisation steps to somewhere sensible
-			s.vscidToKeyAssignment = map[uint64]map[int64]providerkeeper.ConsumerPublicKey{}
-			s.vscidToKeyAssignment[16] = s.readCurrentKeyAssignment()
-			s.vscidToKeyAssignment[s.offsetProviderVscId] = s.readCurrentKeyAssignment()
 
 			s.traces.CurrentTraceIx = i
 			defer func() {
@@ -594,4 +590,7 @@ func (s *CoreSuite) SetupTest() {
 	s.offsetTimeUnix = offsetTimeUnix
 	s.offsetProviderVscId = providerVscID
 	s.simibc = simibc.MakeRelayedPath(s.Suite.T(), path)
+	s.vscidToKeyAssignment = map[uint64]map[int64]providerkeeper.ConsumerPublicKey{}
+	s.vscidToKeyAssignment[16] = s.readCurrentKeyAssignment() // TODO: unhardcode, check if needed
+	s.vscidToKeyAssignment[s.offsetProviderVscId] = s.readCurrentKeyAssignment()
 }