def _create_mock_bucket(): """Returns a mock bucket Args: Returns: """ mock_bucket = Mock(spec=Bucket) mock_bucket.label = "bucket" mock_bucket.color = "#000000" mock_bucket.types = [] mock_bucket.id = ObjectId() return mock_bucket
def _create_mock_bucket(types=None): """Returns a mock bucket Args: types: Returns: """ if types is None: types = [] mock_bucket = Mock(spec=Bucket) mock_bucket.label = 'bucket' mock_bucket.label = '#000000' mock_bucket.types = types return mock_bucket