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)
