def test_splitting(self, init):
m = MemoryStorage()
m._store = MagicMock()
m._store.archive = MagicMock(
return_value={
'date1': {
'foo': {
'cancer': 0,
'messages': 0
},
'bar': {
'cancer': 0,
'messages': 0
}
},
'date2': {
'foo': {
'cancer': 0,
'messages': 0
},
'bar': {
'cancer': 0,
'messages': 0
}
},
})
m.pubsub_socket = MagicMock()
m._archive()
self.assertEqual(m.pubsub_socket.send_multipart.call_count, 4)
def test_splitting(self, init):
m = MemoryStorage()
m._store = MagicMock()
m._store.archive = MagicMock(return_value={
'date1': {'foo': {'cancer': 0, 'messages': 0}, 'bar': {'cancer': 0, 'messages': 0}},
'date2': {'foo': {'cancer': 0, 'messages': 0}, 'bar': {'cancer': 0, 'messages': 0}},
})
m.pubsub_socket = MagicMock()
m._archive()
self.assertEqual(m.pubsub_socket.send_multipart.call_count, 4)
def test_formatting(self, init):
m = MemoryStorage()
m._store = MagicMock()
m._store.archive = MagicMock(
return_value={'foo': {
'bar': {
'cancer': 10,
'messages': 20
}
}})
m.pubsub_socket = MagicMock()
m._archive()
pickled = m.pubsub_socket.send_multipart.call_args[0][0][1]
self.assertEqual(pickle.loads(pickled), {
'date': 'foo',
'channel': 'bar',
'cancer': 10,
'messages': 20
})
def test_formatting(self, init):
m = MemoryStorage()
m._store = MagicMock()
m._store.archive = MagicMock(return_value={
'foo': {
'bar': {
'cancer': 10,
'messages': 20
}
}
})
m.pubsub_socket = MagicMock()
m._archive()
pickled = m.pubsub_socket.send_multipart.call_args[0][0][1]
self.assertEqual(pickle.loads(pickled), {
'date': 'foo',
'channel': 'bar',
'cancer': 10,
'messages': 20
})
