def test_protocol_http(self): protocol = get_protocol("http://www.site.com") self.assertEqual("http", protocol)
def test_protocolsmb(self): protocol = get_protocol("smb://www.site.com") self.assertEqual("smb", protocol)