def test_sort_place_holders(self, find_markers_for_methods_mock, insert_update_marker_task_mock): """Ensure an exsting short_id is returned, and contact merged, if there is a matching marker. """ from google.appengine.ext import ndb from sosbeacon.event.contact_marker import ContactMarker from sosbeacon.event.contact_marker import create_or_update_marker from sosbeacon.event.event import Event from sosbeacon.event.message import Message from sosbeacon.student import Student marker1_key = mock.Mock(spec=ndb.Key) marker1_key.kind.return_value = "ContactMarker" marker1_key.urlsafe.return_value = "CONTACTMARKER" marker1 = ContactMarker(key=marker1_key, short_id='A') marker1_key.get.return_value = marker1 find_markers_for_methods_mock.return_value = (marker1,) event_key = ndb.Key(Event, 'EVENTKEY') student_key = ndb.Key(Student, 'STUDENTKEY') message_key = ndb.Key(Message, 'MESSAGEKEY') ret_value = create_or_update_marker( event_key, student_key, message_key, {'a': True}, ['123']) self.assertEqual('A', ret_value) self.assertEqual(1, insert_update_marker_task_mock.call_count)
def test_sort_place_holders(self, find_markers_for_methods_mock, insert_update_marker_task_mock): """Ensure an exsting short_id is returned, and contact merged, if there is a matching marker. """ from google.appengine.ext import ndb from sosbeacon.event.contact_marker import ContactMarker from sosbeacon.event.contact_marker import create_or_update_marker from sosbeacon.event.event import Event from sosbeacon.event.message import Message from sosbeacon.student import Student marker1_key = mock.Mock(spec=ndb.Key) marker1_key.kind.return_value = "ContactMarker" marker1_key.urlsafe.return_value = "CONTACTMARKER" marker1 = ContactMarker(key=marker1_key, short_id='A') marker1_key.get.return_value = marker1 find_markers_for_methods_mock.return_value = (marker1, ) event_key = ndb.Key(Event, 'EVENTKEY') student_key = ndb.Key(Student, 'STUDENTKEY') message_key = ndb.Key(Message, 'MESSAGEKEY') ret_value = create_or_update_marker(event_key, student_key, message_key, {'a': True}, ['123']) self.assertEqual('A', ret_value) self.assertEqual(1, insert_update_marker_task_mock.call_count)
def test_no_matches(self, find_markers_for_methods_mock, insert_update_marker_task_mock): """Ensure a new short id is returned if there are non existing.""" from google.appengine.ext import ndb from sosbeacon.event.contact_marker import create_or_update_marker from sosbeacon.event.event import Event from sosbeacon.event.message import Message find_markers_for_methods_mock.return_value = () event_key = ndb.Key(Event, 'EVENTKEY') message_key = ndb.Key(Message, 'MESSAGEKEY') student_key = mock.Mock() student_key.id.return_value = "STUDENTKEY" ret_value = create_or_update_marker( event_key, student_key, message_key, {'a': True}, ['123']) self.assertIsNotNone(ret_value) self.assertFalse(insert_update_marker_task_mock.called)
def test_no_matches(self, find_markers_for_methods_mock, insert_update_marker_task_mock): """Ensure a new short id is returned if there are non existing.""" from google.appengine.ext import ndb from sosbeacon.event.contact_marker import create_or_update_marker from sosbeacon.event.event import Event from sosbeacon.event.message import Message find_markers_for_methods_mock.return_value = () event_key = ndb.Key(Event, 'EVENTKEY') message_key = ndb.Key(Message, 'MESSAGEKEY') student_key = mock.Mock() student_key.id.return_value = "STUDENTKEY" ret_value = create_or_update_marker(event_key, student_key, message_key, {'a': True}, ['123']) self.assertIsNotNone(ret_value) self.assertFalse(insert_update_marker_task_mock.called)