Question

我正在尝试为调用存储在Google云存储上的文件的模型编写单元测试，但我还没有找到有关如何模拟GCS服务进行单元测试的任何示例。

似乎应该有一个stub service我可以使用，我在那里描述的testbed docs中看到了一些gcs的引用，但是没有找到使用它的例子我可以工作。

这是我所拥有的模型的精简/示例版本：

import peewee
from google.appengine.api import app_identity
import cloudstorage

class Example(Model):
    uuid = peewee.CharField(default=uuid4)
    some_property = peewee.CharField()

    @property
    def raw_file_one(self):
        bucket = app_identity.get_default_gcs_bucket_name()
        filename = '/{0}/repo_one/{1}'.format(bucket, self.uuid)
        with cloudstorage.open(filename, 'r') as f:
            return f.read()

    def store_raw_file(self, raw_file_one):
        bucket = app_identity.get_default_gcs_bucket_name()

        filename = '/{0}/stat_doms/{1}'.format(bucket, self.uuid)
        with cloudstorage.open(filename, 'w') as f:
            f.write(raw_file_one)

我将使用以下内容构建测试用例：

import unittest

from google.appengine.ext import testbed    

class TestCase(unittest.TestCase):
    def run(self, *args, **kwargs):
        self.stub_google_services()
        result = super(TestCase, self).run(*args, **kwargs)
        self.unstub_google_services()
        return result

    def stub_google_services(self):
        self.testbed = testbed.Testbed()
        self.testbed.activate()
        self.testbed.init_all_stubs()

    def unstub_google_services(self):
        self.testbed.deactivate()

进入模块测试，如：

from application.tests.testcase import TestCase
from application.models import Example

class ExampleTest(TestCase):
    def test_store_raw_file(self):
    ...
    [assert something]

我认为我会做blobstore = self.testbed.get_stub('blobstore')这样的事情来创建我可以执行测试的服务（例如blobstore.CreateBlob(blob_key, image)） - 但我没有看到GCS服务testbed参考文档。

关于如何使用GCS实现单元测试的想法？

Answer 1

我认为你在寻找：

from google.appengine.ext.cloudstorage import cloudstorage_stub
from google.appengine.api.blobstore import blobstore_stub

和

blob_stub = blobstore_stub.BlobstoreServiceStub(blob_storage)
storage_stub = cloudstorage_stub.CloudStorageStub(blob_storage)

testbed._register_stub('blobstore', self.blob_stub)
testbed._register_stub("cloudstorage", self.storage_stub)

如何在GAE上使用Google云端存储的模型上运行本地单元测试（python）

1 个答案: