Sort imports and some minor refactoring

• Add import checks to lint tests
• Remove dependency on mock and use standard library unittest.mock instead
• Remove try except for django's url import since we only support new versions now