def test_remove_proxy_invalid_id_does_nothing(self):
store = Store()
proxy = Proxy('host', 'source', 'us', 'united states', True, 'type', 'source')
store.remove_proxy(1, proxy)
proxy = store.get_proxy()
self.assertIsNone(proxy)
def test_remove_proxy_removes_from_set(self):
store = Store()
id = store.add_store()
proxy = Proxy('host', 'source', 'us', 'united states', True, 'type', 'source')
store.update_store(id, {proxy, })
actual1 = store.get_proxy()
store.remove_proxy(id, proxy)
actual2 = store.get_proxy()
self.assertEqual(actual1, proxy)
self.assertIsNone(actual2)