Unit Tests

[Taar]

Tests need to be written to make sure everything happens in the correct order. Tests should include all drivers and classes. unittest.mock could be used to achive this.