def test_export_remote_accesses_xlsx():
"""Tests for export remote accesses xlsx file method"""
client = CBWXlsx(API_URL, API_KEY, SECRET_KEY)
file_xlsx = "test.xlsx"
with vcr.use_cassette(
'spec/fixtures/vcr_cassettes/export_file_xlsx.yaml'):
response = client.export_remote_accesses_xlsx(file_xlsx)
workbook = openpyxl.load_workbook(file_xlsx)
worksheet = workbook.active
result = []
for cell in worksheet[1]:
result.append(cell.value)
assert response is True and result == [
'HOST', 'PORT', 'TYPE', 'NODE_ID', 'SERVER_GROUPS'
]
result = []
for cell in worksheet[2]:
result.append(cell.value)
# The group is not assigned yet
assert result == [
'10.0.2.15', 22, 'CbwRam::RemoteAccess::Ssh::WithPassword', 1,
None
]
def test_export_remote_accesses_xlsx():
"""Tests for export remote accesses xlsx file method"""
client = CBWXlsx(API_URL, API_KEY, SECRET_KEY)
file_xlsx = "test.xlsx"
with vcr.use_cassette(
'spec/fixtures/vcr_cassettes/export_file_xlsx.yaml'):
response = client.export_remote_accesses_xlsx(file_xlsx)
workbook = xlrd.open_workbook(file_xlsx)
sheet = workbook.sheet_by_index(0)
for rownum in range(sheet.nrows):
result = sheet.row_values(rownum)
break
assert response is True and result == [
'HOST', 'PORT', 'TYPE', 'NODE_ID', 'SERVER_GROUPS'
]
for rownum in range(1, sheet.nrows):
result = sheet.row_values(rownum)
break
# The group is not assigned yet
assert result == [
'10.0.2.15', 22, 'CbwRam::RemoteAccess::Ssh::WithPassword', 1,
''
]
def test_import_remote_accesses_xlsx():
"""Tests for method import_xls"""
client = CBWXlsx(API_URL, API_KEY, SECRET_KEY)
file_xlsx = "spec/fixtures/xlsx_files/batch_import_model.xlsx"
remote_accesses_validate = [
"cbw_object(id=10, type='CbwRam::RemoteAccess::Ssh::WithPassword', \
address='10.0.2.15', port=22, is_valid=None, last_error=None, server_id=None, node_id=1)",
"cbw_object(id=11, type='CbwRam::RemoteAccess::Ssh::WithPassword', address='server02.example.com', \
port=22, is_valid=None, last_error=None, server_id=None, node_id=1)",
"cbw_object(id=12, type='CbwRam::RemoteAccess::Ssh::WithPassword', address='server01.example.com', port=22\
, is_valid=None, last_error=None, server_id=None, node_id=1)"
]
with vcr.use_cassette('spec/fixtures/vcr_cassettes/'
'import_remote_accesses_xlsx_file.yaml'):
response = client.import_remote_accesses_xlsx(file_xlsx)
assert len(response) == 3
assert str(response[0]) == remote_accesses_validate[0]
assert str(response[1]) == remote_accesses_validate[1]
assert str(response[2]) == remote_accesses_validate[2]
file_xlsx = "spec/fixtures/xlsx_files/batch_import_model_false.xlsx"
with vcr.use_cassette('spec/fixtures/vcr_cassettes/'
'import_remote_accesses_xlsx_file_false.yaml'):
response = client.import_remote_accesses_xlsx(file_xlsx)
assert len(response) == 3
assert response[0] is False
assert response[1] is False
assert str(response[2]), remote_accesses_validate[2]
def test_export_remote_accesses_xlsx():
"""Tests for export remote accesses xlsx file method"""
client = CBWXlsx(API_URL, API_KEY, SECRET_KEY)
file_xlsx = "test.xlsx"
with vcr.use_cassette(
'spec/fixtures/vcr_cassettes/export_file_xlsx.yaml'):
response = client.export_remote_accesses_xlsx(file_xlsx)
workbook = xlrd.open_workbook(file_xlsx)
sheet = workbook.sheet_by_index(0)
for rownum in range(sheet.nrows):
result = sheet.row_values(rownum)
break
assert response is True and result == [
'HOST', 'PORT', 'TYPE', 'NODE', 'GROUPS'
]
for rownum in range(1, sheet.nrows):
result = sheet.row_values(rownum)
break
assert result == [
'server01.example.com', 22.0,
'CbwRam::RemoteAccess::Ssh::WithPassword', 'master',
'Production,test'
]
def test_import_remote_accesses_xlsx():
"""Tests for method import_xls"""
client = CBWXlsx(API_URL, API_KEY, SECRET_KEY)
file_xlsx = "spec/fixtures/xlsx_files/batch_import_model.xlsx"
with vcr.use_cassette('spec/fixtures/vcr_cassettes/'
'import_remote_accesses_xlsx_file.yaml'):
response = client.import_remote_accesses_xlsx(file_xlsx)
assert len(response) == 3
assert isinstance(response[0], CBWRemoteAccess) is True
assert isinstance(response[1], CBWRemoteAccess) is True
assert isinstance(response[2], CBWRemoteAccess) is True
file_xlsx = "spec/fixtures/xlsx_files/batch_import_model_false.xlsx"
with vcr.use_cassette('spec/fixtures/vcr_cassettes/'
'import_remote_accesses_xlsx_file_false.yaml'):
response = client.import_remote_accesses_xlsx(file_xlsx)
assert len(response) == 3
assert isinstance(response[0], CBWRemoteAccess) is False
assert isinstance(response[1], CBWRemoteAccess) is False
assert isinstance(response[2], CBWRemoteAccess) is True
"""Export remote accesses in file xlsx""" from cbw_api_toolbox.cbw_file_xlsx import CBWXlsx API_KEY = '' SECRET_KEY = '' API_URL = '' FILE_XLSX = "" #Optional parameter XLSX = CBWXlsx(API_URL, API_KEY, SECRET_KEY) print(XLSX.export_remote_accesses_xlsx())
"""import file xlsx"""
from cbw_api_toolbox.cbw_file_xlsx import CBWXlsx
API_KEY = ''
SECRET_KEY = ''
API_URL = ''
FILE_XLSX = 'examples/import_file.xlsx'
XLSX = CBWXlsx(API_URL, API_KEY, SECRET_KEY)
response = XLSX.import_remote_accesses_xlsx(FILE_XLSX)
if response:
for remote_access in response:
if remote_access:
print("remote access created, id=>>>>>{}".format(remote_access.id))
else:
print("An error occurred, import_remote_accesses_xlsx failed")
else:
print(
"Error format file xlsx::HOST, PORT, TYPE, USERNAME, PASSWORD, KEY, NODE, GROUPS, COMPLIANCE_GROUPS"
)
"""import file xlsx"""
import os
from configparser import ConfigParser
from cbw_api_toolbox.cbw_file_xlsx import CBWXlsx
CONF = ConfigParser()
CONF.read(
os.path.join(os.path.abspath(os.path.dirname(__file__)), '..', 'api.conf'))
XLSX = CBWXlsx(CONF.get('cyberwatch', 'url'), CONF.get('cyberwatch',
'api_key'),
CONF.get('cyberwatch', 'secret_key'))
FILE_XLSX = 'import_file.xlsx'
RESPONSE = XLSX.import_remote_accesses_xlsx(FILE_XLSX)
if RESPONSE:
for remote_access in RESPONSE:
if remote_access:
print("remote access created, id=>>>>>{}".format(remote_access.id))
else:
print("An error occurred, import_remote_accesses_xlsx failed")
else:
print(
"Error format file xlsx::HOST, PORT, TYPE, USERNAME, PASSWORD, KEY, NODE, SERVER_GROUPS"
)
"""Export remote accesses in file xlsx"""
import os
from configparser import ConfigParser
from cbw_api_toolbox.cbw_file_xlsx import CBWXlsx
CONF = ConfigParser()
CONF.read(
os.path.join(os.path.abspath(os.path.dirname(__file__)), '..', 'api.conf'))
XLSX = CBWXlsx(CONF.get('cyberwatch', 'url'), CONF.get('cyberwatch',
'api_key'),
CONF.get('cyberwatch', 'secret_key'))
FILE_XLSX = "" #Optional parameter
print(XLSX.export_remote_accesses_xlsx())