class TestConnectionUnit(unittest.TestCase): def setUp(self): self.tc = TaskdConnection() self.tc.server = "localhost" self.tc.group = "Public" self.tc.uuid = str(uuid.uuid4()) self.tc.username = "******" self.tc.client_cert = "taskc/fixture/pki/client.cert.pem" self.tc.client_key = "taskc/fixture/pki/client.key.pem" self.tc.cacert_file = "taskc/fixture/pki/ca.cert.pem" @mock.patch("taskc.simple.TaskdConnection.recv") @mock.patch("taskc.simple.TaskdConnection.conn", create=True) @mock.patch("taskc.simple.TaskdConnection._mkmsg") @mock.patch("taskc.simple.TaskdConnection._connect") def test_pull(self, tdc_connect, mk_msg, conn, recv): output = "timbo?" recv.return_value = output self.assertEqual(self.tc.pull(), output) tdc_connect.assert_called_with() mk_msg.assert_called_with('sync') @mock.patch("taskc.simple.TaskdConnection.recv") @mock.patch("taskc.simple.TaskdConnection.conn", create=True) # @mock.patch("taskc.transaction.prep_message") @mock.patch("taskc.simple.TaskdConnection._connect") def test_put(self, tdc_connect, conn, recv): tasks = "{This is a task}\n{this is another task}" self.tc.put(tasks) # prep_msg.assert_ tdc_connect.assert_called_with()