class CustomPrinterTest(test_case.TestCase):
def testPrinter(self):
case = cp.Lines([
'this is a header',
cp.Labeled([('Foo', 'carrot'), ('Bar', 12),
('Baz',
cp.Labeled([('Fun', 'doooodles'),
('Sun', cp.Lines(['toot', 'taaat', 3]))])),
('Quux', cp.Mapped([('hundred', 'lots'), ('two', 'few')]))])
])
s = io.StringIO()
p = MockPrinter(out=s)
p.AddRecord(case)
self.assertEqual(
s.getvalue(),
textwrap.dedent("""\
this is a header
Foo: carrot
Bar: 12
Baz:
Fun: doooodles
Sun:
toot
taaat
3
Quux:
hundred lots
two few
------
"""))
def GetConfigMaps(record):
config_maps = {}
config_maps.update({
k: FormatConfigMapKeyRef(v)
for k, v in record.env_vars.config_maps.items()
})
config_maps.update({
k: FormatConfigMapVolumeSource(v)
for k, v in record.MountedVolumeJoin('config_maps').items()
})
return cp.Mapped(k8s_object_printer.OrderByKey(config_maps))
def _GetSecrets(self, record):
secrets = {}
secrets.update({
k: FormatSecretKeyRef(v)
for k, v in record.template.env_vars.secrets.items()
})
secrets.update({
k: FormatSecretVolumeSource(v)
for k, v in record.template.MountedVolumeJoin('secrets').items()
})
return cp.Mapped(self._OrderByKey(secrets))
def GetSecrets(record):
secrets = {}
secrets.update({
k: FormatSecretKeyRef(v)
for k, v in record.env_vars.secrets.items()
})
secrets.update({
k: FormatSecretVolumeSource(v)
for k, v in record.MountedVolumeJoin('secrets').items()
})
return cp.Mapped(k8s_object_printer.OrderByKey(secrets))
def _GetConfigMaps(self, record):
config_maps = {}
config_maps.update({
k: FormatConfigMapKeyRef(v)
for k, v in record.template.env_vars.config_maps.items()
})
config_maps.update({
k: FormatConfigMapVolumeSource(v)
for k, v in record.template.MountedVolumeJoin(
'config_maps').items()
})
return cp.Mapped(self._OrderByKey(config_maps))
def GetConfigMaps(record):
"""Returns a print mapping for env var and volume-mounted config maps."""
config_maps = {}
config_maps.update({
k: _FormatConfigMapKeyRef(v)
for k, v in record.env_vars.config_maps.items()
})
config_maps.update({
k: _FormatConfigMapVolumeSource(v)
for k, v in record.MountedVolumeJoin('config_maps').items()
})
return cp.Mapped(k8s_util.OrderByKey(config_maps))
def GetSecrets(record):
"""Returns a print mapping for env var and volume-mounted secrets."""
secrets = {}
secrets.update({
k: _FormatSecretKeyRef(v)
for k, v in record.env_vars.secrets.items()
})
secrets.update({
k: _FormatSecretVolumeSource(v)
for k, v in record.MountedVolumeJoin('secrets').items()
})
return cp.Mapped(k8s_util.OrderByKey(secrets))
def Transform(self, record):
"""Transform a service into the output structure of marker classes."""
pairs = traffic_pair.GetTrafficTargetPairs(
record.spec_traffic, record.status_traffic, record.is_managed,
record.status.latestReadyRevisionName)
fmt = cp.Lines([
self._GetServiceHeader(record),
self._GetLabels(record.labels), record.status.url, ' ',
cp.Labeled([('Traffic',
cp.Mapped((p.displayPercent, p.displayRevisionId)
for p in pairs))]), ' ',
cp.Labeled([(self._GetLastUpdated(record),
self._RevisionPrinters(record))]),
self._GetReadyMessage(record)
])
return fmt
def _GetUserEnvironmentVariables(self, record):
return cp.Mapped(self._OrderByKey(record.template.env_vars.literals))
def GetUserEnvironmentVariables(record):
return cp.Mapped(
k8s_object_printer.OrderByKey(record.env_vars.literals))
def _TransformTrafficPairs(traffic_pairs):
"""Transforms a List[TrafficTargetPair] into a marker class structure."""
return cp.Labeled([('Traffic',
cp.Mapped((p.displayPercent, p.displayRevisionId)
for p in traffic_pairs))])
