[ISSUE-35] add getRealT

ozontech · Mar 17, 2024 · 592bb00 · 592bb00
1 parent 0c7c5f6
commit 592bb00
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/framework/core/common/step_context_test.go b/pkg/framework/core/common/step_context_test.go
@@ -22,6 +22,8 @@ type providerTMockStep struct {
 	failNow bool
 	failed  bool
 	name    string
+
+	testingT provider.TestingT
 }
 
 func (m *providerTMockStep) Break(args ...interface{}) {
@@ -87,7 +89,11 @@ func (m *providerTMockStep) Helper() {
 }
 
 func (m *providerTMockStep) GetRealT() provider.TestingT {
-	return nil
+	return m.testingT
+}
+
+func (m *providerTMockStep) SetRealT(realT provider.TestingT) {
+	m.testingT = realT
 }
 
 type providerMockStep struct {
@@ -243,6 +249,7 @@ func TestStepCtx_Step(t *testing.T) {
 
 func TestStepCtx_Errorf_withParent(t *testing.T) {
 	mockT := new(providerTMockStep)
+	mockT.SetRealT(t)
 	parentStep := allure.NewSimpleStep("parentStep")
 	parentCtx := &stepCtx{t: mockT, currentStep: parentStep}
 	step := allure.NewSimpleStep("testStep")