def testGenerateNames(self, sections, separator, prefix, validate, expected_name): gen = self.StartObjectPatch(name_generator, '_ThreeLetterGenerator', return_value='abc') self.assertEqual( expected_name, name_generator.GenerateName(sections, separator, prefix, validate)) self.assertEqual(sections, gen.call_count) gen.assert_called_with(validate)
def _GenerateVolumeName(prefix): """Randomly generated name with the given prefix.""" return name_generator.GenerateName(sections=3, separator='-', prefix=prefix)
def _AddRevisionForcingChange(self, serv, config_changes): """Get a new revision forcing config change for the given service.""" curr_generation = serv.generation if serv is not None else 0 revision_suffix = '{}-{}'.format( str(curr_generation + 1).zfill(5), name_generator.GenerateName()) config_changes.insert(0, _NewRevisionForcingChange(revision_suffix))
def testGenerateNamesRequiresAtLeastOneSection(self): with self.assertRaises(AssertionError): name_generator.GenerateName(sections=0)
def Adjust(self, resource): resource.execution_template.labels[ job.NONCE_LABEL] = name_generator.GenerateName(3, separator='_') return resource