def run(): with grpc.insecure_channel('localhost:50051') as channel: stub = hbi_pb2_grpc.HostInventoryStub(channel) for name in util.names(): display_name = "-".join(name) host_fact = Fact(namespace="demo", key="hostname", value=f"{display_name}.com") stub.CreateOrUpdate( HostList(hosts=[ Host(display_name=display_name, facts=[ host_fact, ]) ]))
def test_servicer(): host_list = HostList(hosts=[ p.Host( display_name="-".join(display_name), canonical_facts=[ CanonicalFact( key="hostname", value=f"{'-'.join(display_name)}.com", ) ]) for display_name in names() ]) service = Servicer() ret = service.CreateOrUpdate(host_list, None) assert len(ret.hosts) == len(host_list.hosts) assert host_list.hosts[0].display_name == ret.hosts[0].display_name assert len(service.Get(HostList(hosts=ret.hosts[:1]), None).hosts) == 1
def host_list(): return [ Host({"hostname": n}, display_name=f"{n}.com") for n in ("-".join(dn) for dn in names()) ]
def run(): stub = Client() for name in util.names(): display_name = "-".join(name) facts = {"demo": {"hostname": f"{display_name}"}} stub.create_or_update(Host(display_name=display_name, facts=facts))