Repo, Issue, Gist and Account equality

Alejandro Gómez · Alejandro Gómez · commit 6931da080a24 · 2013-03-26T14:31:48.000+01:00
diff --git a/github3/gists/gist.py b/github3/gists/gist.py
@@ -70,6 +70,9 @@ def __init__(self, data, session=None):
         #: History of this gist, list of :class:`GistHistory <GistHistory>`
         self.history = [GistHistory(h, self) for h in data.get('history', [])]
 
+    def __eq__(self, gist):
+        return self.id == gist.id
+
     def __str__(self):
         return self.id
 
diff --git a/github3/issues.py b/github3/issues.py
@@ -204,6 +204,9 @@ def __init__(self, issue, session=None):
         #: :class:`User <github3.users.User>` who opened the issue.
         self.user = User(issue.get('user'), self._session)
 
+    def __eq__(self, issue):
+        return self.id == issue.id
+
     def __repr__(self):
         return '<Issue [{r[0]}/{r[1]} #{n}]>'.format(r=self.repository,
                                                      n=self.number)
diff --git a/github3/models.py b/github3/models.py
@@ -346,6 +346,9 @@ def __init__(self, acct, session):
         #: Markdown formatted biography
         self.bio = acct.get('bio')
 
+    def __eq__(self, acc):
+        return self.id == acc.id
+
     def __repr__(self):
         return '<{s.type} [{s.login}:{s.name}]>'.format(s=self)
 
diff --git a/github3/repos/repo.py b/github3/repos/repo.py
@@ -125,6 +125,9 @@ def __init__(self, repo, session=None):
         #: default branch for the repository
         self.master_branch = repo.get('master_branch', '')
 
+    def __eq__(self, repo):
+        return self.id == repo.id
+
     def __repr__(self):
         return '<Repository [{0}]>'.format(self)
 
diff --git a/tests/test_gists.py b/tests/test_gists.py
@@ -165,6 +165,9 @@ def test_history(self):
 
         expect(repr(hist).startswith('<Gist History')).is_True()
 
+    def test_equality(self):
+        expect(self.gist) == gists.Gist(load('gist'))
+
 
 class TestGistComment(BaseCase):
     def __init__(self, methodName='runTest'):
diff --git a/tests/test_issues.py b/tests/test_issues.py
@@ -299,3 +299,6 @@ def test_reopen(self):
             ed.assert_called_once_with(
                 self.i.title, self.i.body, u, 'open', n, self.i.labels
             )
+
+    def test_equality(self):
+        expect(self.i) == github3.issues.Issue(load('issue'))
diff --git a/tests/test_orgs.py b/tests/test_orgs.py
@@ -373,3 +373,6 @@ def test_team(self):
 
         expect(self.org.team(1)).isinstance(github3.orgs.Team)
         self.mock_assertions()
+
+    def test_equality(self):
+        expect(self.org) == github3.orgs.Organization(load('org'))
diff --git a/tests/test_repos.py b/tests/test_repos.py
@@ -1000,6 +1000,9 @@ def test_update_label(self):
 
         self.mock_assertions()
 
+    def test_equality(self):
+        expect(self.repo) == repos.Repository(load('repo'))
+
 
 class TestContents(BaseCase):
     def __init__(self, methodName='runTest'):
diff --git a/tests/test_users.py b/tests/test_users.py
@@ -269,3 +269,6 @@ def test_update(self):
 
         self.response('', 404)
         expect(self.user.update(**self.conf['data'])).is_False()
+
+    def test_equality(self):
+        expect(self.user) == github3.users.User(load('user'))

Original file line number	Diff line number	Diff line change
`@@ -299,3 +299,6 @@ def test_reopen(self):`
`299`	`299`	`ed.assert_called_once_with(`
`300`	`300`	`self.i.title, self.i.body, u, 'open', n, self.i.labels`
`301`	`301`	`)`
	`302`	`+`
	`303`	`+ def test_equality(self):`
	`304`	`+ expect(self.i) == github3.issues.Issue(load('issue'))`