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)