def test_addosbreed():
# Arrange
os_signatures = Signatures()
# Act
os_signatures.addosbreed("Test")
# Assert
assert len(os_signatures.osbreeds) == 1
def test_breeds():
# Arrange
os_signatures = Signatures()
# Act
os_signatures.osbreeds = []
# Assert
assert [] == os_signatures.osbreeds
def test_modelstojson():
# Arrange
os_signatures = Signatures()
# Act
os_signatures.modelstojson()
# Assert
assert False
def test_jsontomodels():
# Arrange
os_signatures = Signatures()
# Act
os_signatures.jsontomodels()
# Assert
assert False
def test_signaturesjson(input_data, result):
# Arrange
os_signatures = Signatures()
os_signatures.importsignatures(ImportTypes.STRING, input_data)
# Act
os_signatures.signaturesjson = input_data
# Assert
assert result == os_signatures.signaturesjson
def test_exportsignatures_string():
# Arrange
os_signatures = Signatures()
os_signatures.addosbreed("suse")
expected = "{\"breeds\": {\"suse\": {}}"
# Act
result = os_signatures.exportsignatures(ExportTypes.STRING)
# Assert
assert expected == result
def test_removeosversion():
# Arrange
os_signatures = Signatures()
os_signatures.addosbreed("suse")
os_signatures.addosversion(0, "sles", Osversion())
# Act
os_signatures.removeosversion(0, "sles")
# Assert
assert os_signatures.osbreeds[0].osversions == {}
def test_exportsignatures_file(testpath):
# Arrange
os_signatures = Signatures()
os_signatures.addosbreed("suse")
expected = "{\"breeds\": {\"suse\": {}}"
# Act
os_signatures.exportsignatures(ExportTypes.FILE, testpath)
with open(testpath, "r") as f:
result = f.read()
# Assert
assert expected == result
def test_removeosbreed():
# Arrange
os_signatures = Signatures()
os_signatures.addosbreed("Test")
len_before = len(os_signatures.osbreeds)
# Act
os_signatures.removeosbreed(0)
len_after = len(os_signatures.osbreeds)
# Assert
assert len_before == 1
assert len_after == 0