class ConnectionTests(unittest.TestCase):
def setUp(self):
# NOTE: this won't actually work, idea for this suite of unit tests
# is to mock the actual server responses and just test logic in the
# UEPConnection:
self.cp = UEPConnection(username="******", password="******",
insecure=True)
def test_get_environment_by_name_requires_owner(self):
self.assertRaises(Exception, self.cp.getEnvironment, None, {"name": "env name"})
def test_get_environment_urlencoding(self):
self.cp.conn = Mock()
self.cp.conn.request_get = Mock(return_value=[])
self.cp.getEnvironment(owner_key="myorg", name="env name__++=*&")
self.cp.conn.request_get.assert_called_with(
"/owners/myorg/environments?name=env+name__%2B%2B%3D%2A%26")
def test_entitle_date(self):
self.cp.conn = Mock()
self.cp.conn.request_post = Mock(return_value=[])
self.cp.bind("abcd", date(2011, 9, 2))
self.cp.conn.request_post.assert_called_with(
"/consumers/abcd/entitlements?entitle_date=2011-09-02")
def test_no_entitle_date(self):
self.cp.conn = Mock()
self.cp.conn.request_post = Mock(return_value=[])
self.cp.bind("abcd")
self.cp.conn.request_post.assert_called_with("/consumers/abcd/entitlements")
def setUp(self):
# NOTE: this won't actually work, idea for this suite of unit tests
# is to mock the actual server responses and just test logic in the
# UEPConnection:
self.cp = UEPConnection(username="******", password="******",
insecure=True)
