virtualenvs now compare

hjwp · hjwp · commit 739176ff0b60 · 2018-01-05T14:05:31.000Z
diff --git a/pythonanywhere/virtualenvs.py b/pythonanywhere/virtualenvs.py
@@ -12,6 +12,8 @@ def __init__(self, domain, python_version):
         self.python_version = python_version
         self.path = Path(os.environ['WORKON_HOME']) / domain
 
+    def __eq__(self, other):
+        return self.domain == other.domain and self.python_version == other.python_version
 
     def create(self, nuke):
         pass
diff --git a/tests/test_project.py b/tests/test_project.py
@@ -5,7 +5,7 @@
 from pythonanywhere.project import Project
 from pythonanywhere.exceptions import SanityException
 from pythonanywhere.api import Webapp
-from pythonanywhere.virtualenvs import virtualenv_path
+from pythonanywhere.virtualenvs import Virtualenv
 
 
 
@@ -31,9 +31,9 @@ def test_webapp(self, fake_home):
         assert project.webapp == Webapp('mydomain.com')
 
 
-    def test_virtualenv_path(self, fake_home):
+    def test_virtualenv(self, fake_home):
         project = Project('mydomain.com', 'python.version')
-        assert project.virtualenv_path == virtualenv_path('mydomain.com')
+        assert project.virtualenv == Virtualenv('mydomain.com', 'python.version')
 
 
 
diff --git a/tests/test_virtualenvs.py b/tests/test_virtualenvs.py
@@ -32,11 +32,6 @@ def test_create_pip_installs_packages(self, mock_subprocess):
         assert command_list[2].endswith('pip install package1 package2==1.1.2')
 
 
-    def test_create_returns_virtualenv_path(self, mock_subprocess, virtualenvs_folder):
-        response = create_virtualenv('domain.com', '2.7', 'django', nuke=False)
-        assert response == virtualenvs_folder / 'domain.com'
-
-
     def test_create_nuke_option_deletes_virtualenv_first(self, mock_subprocess, virtualenvs_folder):
         create_virtualenv('domain.com', '2.7', 'django', nuke=True)
         args, kwargs = mock_subprocess.check_call.call_args
