Esempio n. 1
0
def test_create(monkeypatch, popen, version, parameters, region,
                disable_rollback, dry_run, tags):
    mock_named_tempfile = MagicMock()
    mock_tempfile = MagicMock()
    mock_tempfile.name = 'filename'
    mock_named_tempfile.__enter__.return_value = mock_tempfile
    mock_named_tempfile.return_value = mock_named_tempfile
    monkeypatch.setattr('tempfile.NamedTemporaryFile', mock_named_tempfile)
    lizzy_version_tag = 'LizzyVersion={}'.format(VERSION)

    region = region or 'region'
    senza = Senza(region)
    senza.logger = MagicMock()
    senza.create('yaml: yaml', version, parameters, disable_rollback, dry_run,
                 tags)

    mock_named_tempfile.assert_called_with()
    mock_tempfile.write.assert_called_with(b'yaml: yaml')

    expected_disabled_rollback = ['--disable-rollback'
                                  ] if disable_rollback else []
    expected_dry_run = ['--dry-run'] if dry_run else []

    cli_tags = ['-t', lizzy_version_tag]
    for tag in tags:
        cli_tags.extend(['-t', tag])

    popen.assert_called_with(
        ['senza', 'create', '--region', region, '--force'] +
        expected_disabled_rollback + expected_dry_run +
        ['--stacktrace-visible'] + cli_tags + ['filename', version] +
        parameters,
        stdout=-1,
        stderr=-2)
    assert not senza.logger.error.called
Esempio n. 2
0
def test_render_definition(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()

    mock_named_tempfile = MagicMock()
    mock_tempfile = MagicMock()
    mock_tempfile.name = 'lizzy.yaml'
    mock_named_tempfile.__enter__.return_value = mock_tempfile
    mock_named_tempfile.return_value = mock_named_tempfile
    monkeypatch.setattr('tempfile.NamedTemporaryFile', mock_named_tempfile)

    senza.render_definition('yaml content', 'version42', 'imgversion22',
                            ['Param1=app', 'SecondParam=3'])

    cmd = 'senza print --region region -o json --force lizzy.yaml version42 ' \
          'imgversion22 Param1=app SecondParam=3'

    popen.assert_called_with(cmd.split(" "), stdout=-1, stderr=-1)
    assert not senza.logger.error.called

    senza.render_definition('yaml content', None, 'imgversion22',
                            ['Param1=app', 'SecondParam=3'])
    assert not senza.logger.error.called

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaRenderError):
        senza.render_definition('yaml content', 'version42', 'imgversion22',
                                ['Param1=app', 'SecondParam=3'])
Esempio n. 3
0
def test_remove_error(popen, stack_id, dry_run, force):
    senza = Senza('region')
    senza.logger = MagicMock()
    popen.returncode = 1

    with pytest.raises(ExecutionError):
        senza.remove(stack_id, dry_run=dry_run, force=force)
Esempio n. 4
0
def test_traffic(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    traffic = senza.traffic('lizzy', 'version42', 25)

    cmd = 'senza traffic --region region -o json lizzy version42 25'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza', extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'traffic', '--region', 'region', '-o', 'json', 'lizzy', 'version42', '25'],
                             stdout=-1,
                             stderr=-1)

    assert traffic == {'stream': 'stdout'}

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaTrafficError):
        senza.traffic('lizzy', 'version42', 25)

    # traffic listing
    popen.side_effect = None
    traffic = senza.traffic('lizzy', 'version42')

    popen.assert_called_with(['senza', 'traffic', '--region', 'region', '-o', 'json', 'lizzy', 'version42'],
                             stdout=-1,
                             stderr=-1)
    # returns the output result
    assert traffic == {'stream': 'stdout'}
Esempio n. 5
0
def test_render_definition(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()

    mock_named_tempfile = MagicMock()
    mock_tempfile = MagicMock()
    mock_tempfile.name = 'lizzy.yaml'
    mock_named_tempfile.__enter__.return_value = mock_tempfile
    mock_named_tempfile.return_value = mock_named_tempfile
    monkeypatch.setattr('tempfile.NamedTemporaryFile', mock_named_tempfile)

    senza.render_definition('yaml content', 'version42', 'imgversion22',
                            ['Param1=app', 'SecondParam=3'])

    cmd = 'senza print --region region -o json --force lizzy.yaml version42 ' \
          'imgversion22 Param1=app SecondParam=3'

    popen.assert_called_with(cmd.split(" "), stdout=-1, stderr=-1)
    assert not senza.logger.error.called

    senza.render_definition('yaml content', None, 'imgversion22',
                            ['Param1=app', 'SecondParam=3'])
    assert not senza.logger.error.called

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaRenderError):
        senza.render_definition('yaml content', 'version42', 'imgversion22',
                                ['Param1=app', 'SecondParam=3'])
Esempio n. 6
0
def test_remove_error(popen, stack_id, dry_run, force):
    senza = Senza('region')
    senza.logger = MagicMock()
    popen.returncode = 1

    with pytest.raises(ExecutionError):
        senza.remove(stack_id, dry_run=dry_run, force=force)
Esempio n. 7
0
def test_create_error(monkeypatch, popen):
    mock_named_tempfile = MagicMock()
    mock_tempfile = MagicMock()
    mock_tempfile.name = 'filename'
    mock_named_tempfile.__enter__.return_value = mock_tempfile
    mock_named_tempfile.return_value = mock_named_tempfile
    monkeypatch.setattr('tempfile.NamedTemporaryFile', mock_named_tempfile)

    senza = Senza('region')
    senza.logger = MagicMock()
    popen.returncode = 1

    with pytest.raises(ExecutionError):
        senza.create('yaml: yaml', '10', ['param1', 'param2'], False, False, {})
Esempio n. 8
0
def test_create_error(monkeypatch, popen):
    mock_named_tempfile = MagicMock()
    mock_tempfile = MagicMock()
    mock_tempfile.name = 'filename'
    mock_named_tempfile.__enter__.return_value = mock_tempfile
    mock_named_tempfile.return_value = mock_named_tempfile
    monkeypatch.setattr('tempfile.NamedTemporaryFile', mock_named_tempfile)

    senza = Senza('region')
    senza.logger = MagicMock()
    popen.returncode = 1

    with pytest.raises(ExecutionError):
        senza.create('yaml: yaml', '10', ['param1', 'param2'], False, False,
                     {})
Esempio n. 9
0
def test_remove(popen, stack_id, region, dry_run, force):
    senza = Senza(region)
    senza.logger = MagicMock()
    senza.remove(stack_id, dry_run=dry_run, force=force)

    dry_run_flag = ['--dry-run'] if dry_run else []
    force_flag = ['--force'] if force else []

    stack_name, stack_ver = stack_id.rsplit("-", 1)

    popen.assert_called_with(['senza', 'delete'] + ['--region', region] +
                             dry_run_flag + force_flag +
                             [stack_name, stack_ver],
                             stdout=-1,
                             stderr=-2)

    assert not senza.logger.error.called
    assert not senza.logger.exception.called
Esempio n. 10
0
def test_domain(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    domains = senza.domains()

    cmd = 'senza domains --region region -o json'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(
        ['senza', 'domains', '--region', 'region', '-o', 'json'],
        stdout=-1,
        stderr=-1)

    assert domains == {'stream': 'stdout'}

    senza.logger.reset_mock()
    popen.reset_mock()

    popen.communicate.return_value = b'{"test": "domain2"}', b'stderr'
    domains = senza.domains('lizzy')

    cmd = 'senza domains --region region -o json lizzy'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(
        ['senza', 'domains', '--region', 'region', '-o', 'json', 'lizzy'],
        stdout=-1,
        stderr=-1)

    assert domains == {'test': 'domain2'}

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaDomainsError):
        senza.domains()
Esempio n. 11
0
def test_remove(popen, stack_id, region, dry_run, force):
    senza = Senza(region)
    senza.logger = MagicMock()
    senza.remove(stack_id, dry_run=dry_run, force=force)

    dry_run_flag = ['--dry-run'] if dry_run else []
    force_flag = ['--force'] if force else []

    stack_name, stack_ver = stack_id.rsplit("-", 1)

    popen.assert_called_with(['senza', 'delete']
                             + ['--region', region]
                             + dry_run_flag
                             + force_flag
                             + [stack_name, stack_ver],
                             stdout=-1, stderr=-2)

    assert not senza.logger.error.called
    assert not senza.logger.exception.called
Esempio n. 12
0
def test_list(monkeypatch, popen):
    popen.communicate.return_value = b'["item1", "item2"]', b'stderr'
    senza = Senza('region')
    senza.logger = MagicMock()
    list = senza.list()

    cmd = 'senza list --region region -o json'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza', extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'list', '--region', 'region', '-o', 'json'], stdout=-1, stderr=-1)

    assert list == ["item1", "item2"]

    # Test invalid json
    popen.communicate.return_value = b'"', b'stderr'
    with pytest.raises(ExecutionError):
        senza.list()
Esempio n. 13
0
def test_scale(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    senza.scale('lizzy', 'version42', 0)

    cmd = 'senza scale --region region lizzy version42 0 --force'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'scale', '--region', 'region',
                              'lizzy', 'version42', '0', '--force'],
                             stdout=-1, stderr=-2)

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaScaleError):
        senza.scale('lizzy', 'version42', 0)
Esempio n. 14
0
def test_patch(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    senza.patch('lizzy', 'version42', 'latest')

    cmd = 'senza patch --region region -o json lizzy version42 --image=latest'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'patch', '--region', 'region', '-o',
                             'json', 'lizzy', 'version42', '--image=latest'],
                             stdout=-1, stderr=-1)

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaPatchError):
        senza.patch('lizzy', 'version42', 'latest')
Esempio n. 15
0
def test_respawn_instances(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    senza.respawn_instances('lizzy', 'version42')

    cmd = 'senza respawn-instances --region region -o json lizzy version42'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'respawn-instances', '--region',
                             'region', '-o', 'json', 'lizzy', 'version42'],
                             stdout=-1, stderr=-1)

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaRespawnInstancesError):
        senza.respawn_instances('lizzy', 'version42')
Esempio n. 16
0
def test_traffic(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    traffic = senza.traffic('lizzy', 'version42', 25)

    cmd = 'senza traffic --region region -o json lizzy version42 25'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with([
        'senza', 'traffic', '--region', 'region', '-o', 'json', 'lizzy',
        'version42', '25'
    ],
                             stdout=-1,
                             stderr=-1)

    assert traffic == {'stream': 'stdout'}

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaTrafficError):
        senza.traffic('lizzy', 'version42', 25)

    # traffic listing
    popen.side_effect = None
    traffic = senza.traffic('lizzy', 'version42')

    popen.assert_called_with([
        'senza', 'traffic', '--region', 'region', '-o', 'json', 'lizzy',
        'version42'
    ],
                             stdout=-1,
                             stderr=-1)
    # returns the output result
    assert traffic == {'stream': 'stdout'}
Esempio n. 17
0
def test_create(monkeypatch, popen, version, parameters, region, disable_rollback, dry_run, tags):
    mock_named_tempfile = MagicMock()
    mock_tempfile = MagicMock()
    mock_tempfile.name = 'filename'
    mock_named_tempfile.__enter__.return_value = mock_tempfile
    mock_named_tempfile.return_value = mock_named_tempfile
    monkeypatch.setattr('tempfile.NamedTemporaryFile', mock_named_tempfile)
    lizzy_version_tag = 'LizzyVersion={}'.format(VERSION)

    region = region or 'region'
    senza = Senza(region)
    senza.logger = MagicMock()
    senza.create('yaml: yaml', version, parameters, disable_rollback,
                 dry_run, tags)

    mock_named_tempfile.assert_called_with()
    mock_tempfile.write.assert_called_with(b'yaml: yaml')

    expected_disabled_rollback = ['--disable-rollback'] if disable_rollback else []
    expected_dry_run = ['--dry-run'] if dry_run else []

    cli_tags = ['-t', lizzy_version_tag]
    for tag in tags:
        cli_tags.extend(['-t', tag])

    popen.assert_called_with(['senza', 'create',
                              '--region', region,
                              '--force']
                             + expected_disabled_rollback
                             + expected_dry_run
                             + ['--stacktrace-visible']
                             + cli_tags
                             + ['filename', version]
                             + parameters,
                             stdout=-1,
                             stderr=-2)
    assert not senza.logger.error.called
Esempio n. 18
0
def test_respawn_instances(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    senza.respawn_instances('lizzy', 'version42')

    cmd = 'senza respawn-instances --region region -o json lizzy version42'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with([
        'senza', 'respawn-instances', '--region', 'region', '-o', 'json',
        'lizzy', 'version42'
    ],
                             stdout=-1,
                             stderr=-1)

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaRespawnInstancesError):
        senza.respawn_instances('lizzy', 'version42')
Esempio n. 19
0
def test_patch(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    senza.patch('lizzy', 'version42', 'latest')

    cmd = 'senza patch --region region -o json lizzy version42 --image=latest'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with([
        'senza', 'patch', '--region', 'region', '-o', 'json', 'lizzy',
        'version42', '--image=latest'
    ],
                             stdout=-1,
                             stderr=-1)

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaPatchError):
        senza.patch('lizzy', 'version42', 'latest')
Esempio n. 20
0
def test_scale(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    senza.scale('lizzy', 'version42', 0)

    cmd = 'senza scale --region region lizzy version42 0 --force'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with([
        'senza', 'scale', '--region', 'region', 'lizzy', 'version42', '0',
        '--force'
    ],
                             stdout=-1,
                             stderr=-2)

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaScaleError):
        senza.scale('lizzy', 'version42', 0)
Esempio n. 21
0
def test_domain(monkeypatch, popen):
    senza = Senza('region')
    senza.logger = MagicMock()
    domains = senza.domains()

    cmd = 'senza domains --region region -o json'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza', extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'domains', '--region', 'region', '-o', 'json'],
                             stdout=-1,
                             stderr=-1)

    assert domains == {'stream': 'stdout'}

    senza.logger.reset_mock()
    popen.reset_mock()

    popen.communicate.return_value = b'{"test": "domain2"}', b'stderr'
    domains = senza.domains('lizzy')

    cmd = 'senza domains --region region -o json lizzy'
    senza.logger.debug.assert_called_with('Executing %s.', 'senza', extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(['senza', 'domains', '--region', 'region', '-o', 'json', 'lizzy'], stdout=-1, stderr=-1)

    assert domains == {'test': 'domain2'}

    # test error case
    popen.side_effect = ExecutionError('', '')

    with pytest.raises(SenzaDomainsError):
        senza.domains()
Esempio n. 22
0
def test_list(monkeypatch, popen):
    popen.communicate.return_value = b'["item1", "item2"]', b'stderr'
    senza = Senza('region')
    senza.logger = MagicMock()
    list = senza.list()

    cmd = 'senza list --region region -o json'
    senza.logger.debug.assert_called_with('Executing %s.',
                                          'senza',
                                          extra={'command': cmd})
    assert not senza.logger.error.called
    assert not senza.logger.exception.called

    popen.assert_called_with(
        ['senza', 'list', '--region', 'region', '-o', 'json'],
        stdout=-1,
        stderr=-1)

    assert list == ["item1", "item2"]

    # Test invalid json
    popen.communicate.return_value = b'"', b'stderr'
    with pytest.raises(ExecutionError):
        senza.list()