def test_cluster_tag_create_yaml(self): env_id = 45 tag_name = 'fake_tag' params = { "owner_type": "clusters", "owner_id": env_id, "tag_name": tag_name } args = 'tag create -e {} -n {} -f yaml -d /tmp'.format( env_id, tag_name) test_data = fake_tag.get_fake_tag(tag_name) expected_path = '/tmp/clusters_{}/fake_tag.yaml'.format(env_id) m_open = mock.mock_open(read_data=yaml.safe_dump(test_data)) with mock.patch('fuelclient.commands.tag.open', m_open, create=True): self.exec_command(args) m_open.assert_called_once_with(expected_path, 'r') self.m_get_client.assert_called_once_with('tag', mock.ANY) self.m_client.create.assert_called_once_with(test_data, **params)
def test_release_tag_create_json(self): release_id = 45 tag_name = 'fake_tag' params = { "owner_type": "releases", "owner_id": release_id, "tag_name": tag_name } args = 'tag create -r {} -n {} -f json -d /tmp'.format( release_id, tag_name) test_data = fake_tag.get_fake_tag(tag_name) expected_path = '/tmp/releases_{}/fake_tag.json'.format(release_id) m_open = mock.mock_open(read_data=json.dumps(test_data)) with mock.patch('fuelclient.commands.tag.open', m_open, create=True): self.exec_command(args) m_open.assert_called_once_with(expected_path, 'r') self.m_get_client.assert_called_once_with('tag', mock.ANY) self.m_client.create.assert_called_once_with(test_data, **params)
def test_cluster_tag_download_json(self, m_dump): env_id = 45 tag_name = 'fake_tag' test_data = fake_tag.get_fake_tag(fake_tag) args = 'tag download -e {} -n {} -f json -d /tmp'.format( env_id, tag_name) expected_path = '/tmp/clusters_{id}/{name}.json'.format(id=env_id, name=tag_name) self.m_client.get_tag.return_value = test_data m_open = mock.mock_open() with mock.patch('fuelclient.commands.tag.open', m_open, create=True): self.exec_command(args) m_open.assert_called_once_with(expected_path, 'w') m_dump.assert_called_once_with(test_data, mock.ANY, indent=4) self.m_get_client.assert_called_once_with('tag', mock.ANY) self.m_client.get_tag.assert_called_once_with('clusters', env_id, tag_name)
def test_release_tag_download_yaml(self, m_safe_dump): release_id = 45 tag_name = 'fake_tag' test_data = fake_tag.get_fake_tag(fake_tag) args = 'tag download -r {} -n {} -f yaml -d /tmp'.format( release_id, tag_name) expected_path = '/tmp/releases_{id}/{name}.yaml'.format(id=release_id, name=tag_name) self.m_client.get_tag.return_value = test_data m_open = mock.mock_open() with mock.patch('fuelclient.commands.tag.open', m_open, create=True): self.exec_command(args) m_open.assert_called_once_with(expected_path, 'w') m_safe_dump.assert_called_once_with(test_data, mock.ANY, default_flow_style=False) self.m_get_client.assert_called_once_with('tag', mock.ANY) self.m_client.get_tag.assert_called_once_with('releases', release_id, tag_name)