From 18f3933bae3fa7a2dfd60aba572d71f9ee109ecd Mon Sep 17 00:00:00 2001
From: "duanchao.bill" <duanchao.bill@bytedance.com>
Date: Wed, 27 Apr 2022 18:05:00 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E6=B7=BB=E5=8A=A0RunThriftRequest,StepThri?=
 =?UTF-8?q?ftRequestValidation,StepThriftRequestExtraction=20=E5=88=B0step?=
 =?UTF-8?q?=E4=B8=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 httprunner/step.py                  | 5 +++++
 httprunner/thrift/data_convertor.py | 4 ++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/httprunner/step.py b/httprunner/step.py
index 8c42aea5..75cc43dc 100644
--- a/httprunner/step.py
+++ b/httprunner/step.py
@@ -14,6 +14,8 @@ from httprunner.step_sql_request import (
     StepSqlRequestExtraction,
 )
 
+from httprunner.step_thrift_request import RunThriftRequest,StepThriftRequestValidation,StepThriftRequestExtraction
+
 
 class Step(object):
     def __init__(
@@ -26,6 +28,9 @@ class Step(object):
             RunSqlRequest,
             StepSqlRequestValidation,
             StepSqlRequestExtraction,
+            RunThriftRequest,
+            StepThriftRequestValidation,
+            StepThriftRequestExtraction
         ],
     ):
         self.__step = step
diff --git a/httprunner/thrift/data_convertor.py b/httprunner/thrift/data_convertor.py
index 45514926..b25af390 100644
--- a/httprunner/thrift/data_convertor.py
+++ b/httprunner/thrift/data_convertor.py
@@ -356,8 +356,8 @@ class ThriftJSONEncoder(json.JSONEncoder):
 
     def encode(self, o):
         """Return a JSON string representation of a Python data structure.
-        JSONEncoder().encode({"foo": ["bar", "baz"]})
-       '{"foo": ["bar", "baz"]}'
+         JSONEncoder().encode({"foo": ["bar", "baz"]})
+        '{"foo": ["bar", "baz"]}'
 
         """
         # This is for extremely simple cases and benchmarks.