Fix default values for enum service args #298 (#299)

This commit is contained in:
Kim Gustyr
2021-12-04 00:26:48 +03:00
committed by GitHub
parent b0a36d12e4
commit bc3cfc5562
3 changed files with 26 additions and 4 deletions

View File

@@ -2,9 +2,16 @@ syntax = "proto3";
package service;
enum ThingType {
UNKNOWN = 0;
LIVING = 1;
DEAD = 2;
}
message DoThingRequest {
string name = 1;
repeated string comments = 2;
ThingType type = 3;
}
message DoThingResponse {

View File

@@ -1,7 +1,8 @@
import betterproto
from dataclasses import dataclass
from typing import Optional, List, Dict
from datetime import datetime, timedelta
from datetime import datetime
from inspect import signature
def test_has_field():
@@ -476,3 +477,10 @@ def test_iso_datetime_list():
msg.from_dict({"timestamps": iso_candidates})
assert all([isinstance(item, datetime) for item in msg.timestamps])
def test_enum_service_argument__expected_default_value():
from tests.output_betterproto.service.service import ThingType, TestStub
sig = signature(TestStub.do_thing)
assert sig.parameters["type"].default == ThingType.UNKNOWN