在python中模拟BigQuery连接

Question

我在python文件中有以下代码。我必须对这个文件进行单元测试。但是为了做到这一点，我需要实例化类的对象

class BigQuery(metaclass=singleton.Singleton):
    """
    Big Query Class for operations on big query
    Will standardize in future versions.
    """

    def __init__(self):
        """
        Used for initializing client
        """
        try:
            self.client = bigquery.Client.from_service_account_json(
                SERVICE_ACCOUNT_JSON)
        except:
            logging.error("Cannot instantiate bigquery client", exc_info=True)
            raise Exception("Cannot instantiate bigquery client.")

Answer 1

我让它起作用了。基本上，您需要模拟对bigquery客户机初始化的函数调用。在mock.patch的帮助下，我们可以模拟客户机对象或函数from_service_account_json。下面是代码

with patch.object(bigquery.Client, "from_service_account_json") as srv_acc_mock:
            srv_acc_mock.return_value = Mock()

            # do something here....

我们也需要对GCS客户机遵循相同的模式，但是通过导入正确的模块将bigquery.client更改为storage.client。