def testInstanceCompletion(self):
instances_list = data.GetDatabaseInstancesListOfTwo(
) + data.GetDatabaseInstancesListOfOne()
self.StartObjectPatch(instances._BaseInstances,
'GetDatabaseInstances',
return_value=instances_list)
completer = self.Completer(flags.InstanceCompleter, cli=self.cli)
self.assertCountEqual(
['backupless-instance1', 'backupless-instance2', 'testinstance'],
completer.Complete('', self.parameter_info))
def testGetDatabaseInstancesWithBatchSize(self):
self.mocked_client.instances.List.Expect(
self.messages.SqlInstancesListRequest(
maxResults=10,
project=self.Project(),
),
self.messages.InstancesListResponse(
items=data.GetDatabaseInstancesListOfOne(),
kind='sql#instancesList',
nextPageToken='10',
))
self.mocked_client.instances.List.Expect(
self.messages.SqlInstancesListRequest(
pageToken='10',
project=self.Project(),
maxResults=10,
),
self.messages.InstancesListResponse(
items=data.GetDatabaseInstancesListOfTwo(),
kind='sql#instancesList',
nextPageToken=None,
))
instance_list = data.GetDatabaseInstancesListOfOne(
) + data.GetDatabaseInstancesListOfTwo()
instance_list[
0].state = self.messages.DatabaseInstance.StateValueValuesEnum.RUNNABLE
instance_list[
0].settings.activationPolicy = self.messages.Settings.ActivationPolicyValueValuesEnum.NEVER
expected_instance_list = [
instances_util.DatabaseInstancePresentation(instance)
for instance in instance_list
]
self.assertEqual(
expected_instance_list,
list(
instances_util._BaseInstances.GetDatabaseInstances(
batch_size=10)))
def testInstancesListWithLimit(self):
instances_list = data.GetDatabaseInstancesListOfTwo(
) + data.GetDatabaseInstancesListOfOne()
mocked_database_instances = self.StartObjectPatch(
instances._BaseInstances, 'GetDatabaseInstances')
mocked_database_instances.return_value = instances_list
self.Run('sql instances list --limit=2')
mocked_database_instances.assert_called_once()
mocked_database_instances.assert_called_with(limit=2, batch_size=None)
# pylint:disable=line-too-long
self.AssertOutputContains("""\
NAME DATABASE_VERSION LOCATION TIER PRIMARY_ADDRESS PRIVATE_ADDRESS STATUS
testinstance MYSQL_5_5 us-central D0 - - RUNNABLE
backupless-instance1 MYSQL_5_5 us-central1-a D1 - - RUNNABLE
""",
normalize_space=True)
self.AssertOutputNotContains('backupless-instance2')
def testInstancesList(self):
instances_list = [
instances.DatabaseInstancePresentation(instance)
for instance in data.GetDatabaseInstancesListOfTwo() +
data.GetDatabaseInstancesListOfOne()
]
mocked_database_instances = self.StartObjectPatch(
instances._BaseInstances, 'GetDatabaseInstances')
mocked_database_instances.return_value = instances_list
self.Run('sql instances list')
mocked_database_instances.assert_called_once()
# pylint:disable=line-too-long
self.AssertOutputContains("""\
NAME DATABASE_VERSION LOCATION TIER PRIMARY_ADDRESS PRIVATE_ADDRESS STATUS
testinstance MYSQL_5_5 us-central D0 - - RUNNABLE
backupless-instance1 MYSQL_5_5 us-central1-a D1 - - RUNNABLE
backupless-instance2 MYSQL_5_5 us-central D1 - - STOPPED
""",
normalize_space=True)
