31 lines
739 B
Python
31 lines
739 B
Python
from src.version_manager import VersionManager
|
|
|
|
|
|
def run_tests():
|
|
# Test it_1: Initialization
|
|
assert VersionManager().release() == "0.0.1"
|
|
assert VersionManager("1.2.3").release() == "1.2.3"
|
|
assert VersionManager("1").release() == "1.0.0"
|
|
|
|
# Test it_5: Rollbacks
|
|
vm = VersionManager().major().rollback()
|
|
assert vm.release() == "0.0.1"
|
|
|
|
# Test it_7: Exception Handling
|
|
try:
|
|
VersionManager("a")
|
|
except Exception as e:
|
|
assert str(e) == "Error occured while parsing version!"
|
|
|
|
# Test it_8: Rollback failure
|
|
vm2 = VersionManager()
|
|
try:
|
|
vm2.rollback()
|
|
except Exception as e:
|
|
assert str(e) == "Cannot rollback!"
|
|
|
|
print("All tests passed!")
|
|
|
|
|
|
run_tests()
|