Skip to content

Commit

Permalink
Add a really simple test.
Browse files Browse the repository at this point in the history
  • Loading branch information
happygiraffe committed Sep 27, 2010
1 parent c59c7e3 commit 4b14504
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 2 deletions.
6 changes: 4 additions & 2 deletions sites_uploader.py
Expand Up @@ -107,14 +107,16 @@ def __init__(self, domain, site, ssl=True, debug=False):
self.debug = debug
self.client = None

def _MakeClient(self, client_authz=ClientAuthorizer):
def _MakeClient(self, client_authz=None):
"""Return a populated SitesClient object."""
client = gdata.sites.client.SitesClient(source=SOURCE, site=self.site,
domain=self.domain)
client.ssl = self.ssl
client.http_client.debug = self.debug
# Make sure we've got a valid token in the client.
client_authz().FetchClientToken(client)
if not client_authz:
client_authz = ClientAuthorizer()
client_authz.FetchClientToken(client)
return client

@property
Expand Down
34 changes: 34 additions & 0 deletions sites_uploader_test.py
@@ -0,0 +1,34 @@
#!/usr/bin/python

import gdata.sites.client
import mox
import unittest

import sites_uploader

class SitesUploaderTest(unittest.TestCase):

DOMAIN = 'example.com'
SITE = 'test'

def setUp(self):
self.mox = mox.Mox()

def tearDown(self):
self.mox.UnsetStubs()

def testMakeClient(self):
mock_authz = self.mox.CreateMock(sites_uploader.ClientAuthorizer)
mock_authz.FetchClientToken(mox.IsA(gdata.sites.client.SitesClient))
self.mox.ReplayAll()

uploader = sites_uploader.SitesUploader(self.DOMAIN, self.SITE)
client = uploader._MakeClient(mock_authz)

self.mox.VerifyAll()
self.assertEquals(self.DOMAIN, client.domain)
self.assertEquals(self.SITE, client.site)
self.assertEquals(True, client.ssl)

if __name__ == '__main__':
unittest.main()

0 comments on commit 4b14504

Please sign in to comment.