class TaskPayloadHelperTestSuite(TestCase): def setUp(self): self.taskPayloadHelper = TaskPayloadHelper(TEST_PAYLOAD) self.owner = TaskOwner(email='*****@*****.**', key='example') self.owner.save() def test_get_ready_datetime(self): self.assertEqual(self.taskPayloadHelper.get_ready_datetime(), TEST_DATE_EXPECT) def test_is_duplicate(self): self.assertFalse(self.taskPayloadHelper.is_duplicate()) task = Task(owner=self.owner, source=TEST_PAYLOAD[const.KEY_SOURCE], dest=TEST_PAYLOAD[const.KEY_DEST], content=TEST_PAYLOAD[const.KEY_CONTENT], ready_time=TEST_DATE_EXPECT) task.save() self.assertTrue(self.taskPayloadHelper.is_duplicate()) def test_is_valid(self): self.assertTrue(self.taskPayloadHelper.is_valid())
class QueueTestSuite(TestCase): def setUp(self): self.taskowner = TaskOwner(email="*****@*****.**", key="example") self.taskowner.save() def test_ready_and_delete_tasks(self): task_1 = Task( owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.utcnow() + timedelta(minutes=10), ) task_1.save() task_2 = Task( owner=self.taskowner, source="*****@*****.**", dest="*****@*****.**", content="some message", ready_time=datetime.utcnow(), ) task_2.save() ready_tasks = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks), 1) self.assertEqual(ready_tasks[0].source, "*****@*****.**") pushQueue._delete_processed_tasks(ready_tasks) ready_tasks_after = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks_after), 0) def test_queue_task_creation(self): task_1 = Task( owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.now(), ) task_1.save() ready_tasks = pushQueue._get_ready_tasks() queue_Task = pushQueue._create_task_from_db(ready_tasks[0]) self.assertTrue(queue_Task)
class QueueTestSuite(TestCase): def setUp(self): self.taskowner = TaskOwner(email='*****@*****.**', key='example') self.taskowner.save() def test_ready_and_delete_tasks(self): task_1 = Task(owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.utcnow() + timedelta(minutes=10)) task_1.save() task_2 = Task(owner=self.taskowner, source="*****@*****.**", dest="*****@*****.**", content="some message", ready_time=datetime.utcnow()) task_2.save() ready_tasks = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks), 1) self.assertEqual(ready_tasks[0].source, "*****@*****.**") pushQueue._delete_processed_tasks(ready_tasks) ready_tasks_after = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks_after), 0) def test_queue_task_creation(self): task_1 = Task(owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.now()) task_1.save() ready_tasks = pushQueue._get_ready_tasks() queue_Task = pushQueue._create_task_from_db(ready_tasks[0]) self.assertTrue(queue_Task)
class FullRunTestSuite(TestCase): def setUp(self): self.taskowner = TaskOwner(email='*****@*****.**', key='example') self.taskowner.save() def test_POST(self): request = HttpRequest() request.method = 'POST' request.POST = {const.KEY_ID: "", const.KEY_SOURCE: "*****@*****.**", const.KEY_DEST: "*****@*****.**", const.KEY_CONTENT: "text content", const.KEY_READY_TIME: "2013-09-13T04:22:58.578Z"} response = taskrouter(request=request, owner_key=self.taskowner.key, task_id=None) response_body = response.content self.assertTrue(response_body) json_body = json.loads(response_body) self.assertTrue(json_body) self.assertEqual(json_body[const.KEY_SOURCE], request.POST[const.KEY_SOURCE]) self.assertEqual(json_body[const.KEY_DEST], request.POST[const.KEY_DEST]) self.assertEqual(json_body[const.KEY_CONTENT], request.POST[const.KEY_CONTENT]) self.assertEqual(json_body[const.KEY_READY_TIME], request.POST[const.KEY_READY_TIME]) def test_POST_then_GET(self): request_POST = HttpRequest() request_POST.method = 'POST' request_POST.POST = {const.KEY_ID: "", const.KEY_SOURCE: "*****@*****.**", const.KEY_DEST: "*****@*****.**", const.KEY_CONTENT: "text content", const.KEY_READY_TIME: "2000-09-13T04:22:58.100Z"} response_POST = taskrouter(request=request_POST, owner_key=self.taskowner.key, task_id=None) json_POST = json.loads(response_POST.content) task_id = json_POST[const.KEY_ID] request_GET = HttpRequest() request_GET.method = 'GET' response_GET = taskrouter(request=request_GET, owner_key=self.taskowner.key, task_id=task_id) json_GET = json.loads(response_GET.content) self.assertEqual(json_GET[const.KEY_ID], task_id) self.assertEqual(json_GET[const.KEY_SOURCE], request_POST.POST[const.KEY_SOURCE]) self.assertEqual(json_GET[const.KEY_DEST], request_POST.POST[const.KEY_DEST]) self.assertEqual(json_GET[const.KEY_CONTENT], request_POST.POST[const.KEY_CONTENT]) self.assertEqual(json_GET[const.KEY_READY_TIME], request_POST.POST[const.KEY_READY_TIME]) def test_POST_then_DELETE(self): request_POST = HttpRequest() request_POST.method = 'POST' request_POST.POST = {const.KEY_ID: "", const.KEY_SOURCE: "*****@*****.**", const.KEY_DEST: "*****@*****.**", const.KEY_CONTENT: "text content", const.KEY_READY_TIME: "2013-09-13T04:22:58.578Z"} response_POST = taskrouter(request=request_POST, owner_key=self.taskowner.key, task_id=None) json_POST = json.loads(response_POST.content) task_id = json_POST[const.KEY_ID] request_DELETE = HttpRequest() request_DELETE.method = 'DELETE' response_DELETE = taskrouter(request=request_DELETE, owner_key=self.taskowner.key, task_id=task_id) json_DELETE = json.loads(response_DELETE.content) self.assertEqual(json_DELETE[const.KEY_ID], task_id)
def setUp(self): self.taskowner = TaskOwner(email='*****@*****.**', key='example') self.taskowner.save()
def setUp(self): self.taskPayloadHelper = TaskPayloadHelper(TEST_PAYLOAD) self.owner = TaskOwner(email='*****@*****.**', key='example') self.owner.save()
def setUp(self): self.taskowner = TaskOwner(email="*****@*****.**", key="example") self.taskowner.save()