def test_not_implemented(self, init):
        w = WriteOnlyStorage()

        self.assertRaises(NotImplementedError, lambda: w.cancer())
        self.assertRaises(NotImplementedError, lambda: w.leaderboard(None))
        self.assertRaises(NotImplementedError, lambda: w.leaderboards(None))
        self.assertRaises(NotImplementedError, lambda: w.channel(None))
        self.assertRaises(NotImplementedError, lambda: w.status())
        self.assertRaises(NotImplementedError, lambda: w.store(None, None))
        self.assertRaises(NotImplementedError, lambda: w.search(None))
Beispiel #2
0
    def record(self):
        # summaries are written to the database
        if not self.storage:
            from twitchcancer.storage.writeonlystorage import WriteOnlyStorage
            self.storage = WriteOnlyStorage()

        self.storage.record()
  def test_not_implemented(self, init):
    w = WriteOnlyStorage()
    w._store = MagicMock()
    w._store.update_leaderboard = MagicMock(side_effect=KeyboardInterrupt)
    w.socket = MagicMock()
    w.socket.recv_multipart = MagicMock(return_value=(
      b'topic',
      pickle.dumps({'data':'foo'})
    ))

    try:
      w.record()
    except KeyboardInterrupt:
      # we raised that exception to stop the infinite loop
      pass

    w._store.update_leaderboard.assert_called_once_with({'data': 'foo'})
  def test_not_implemented(self, init):
    w = WriteOnlyStorage()

    self.assertRaises(NotImplementedError, lambda: w.cancer())
    self.assertRaises(NotImplementedError, lambda: w.leaderboard(None))
    self.assertRaises(NotImplementedError, lambda: w.leaderboards(None))
    self.assertRaises(NotImplementedError, lambda: w.channel(None))
    self.assertRaises(NotImplementedError, lambda: w.status())
    self.assertRaises(NotImplementedError, lambda: w.store(None, None))
    self.assertRaises(NotImplementedError, lambda: w.search(None))
    def test_not_implemented(self, init):
        w = WriteOnlyStorage()
        w._store = MagicMock()
        w._store.update_leaderboard = MagicMock(side_effect=KeyboardInterrupt)
        w.socket = MagicMock()
        w.socket.recv_multipart = MagicMock(
            return_value=(b'topic', pickle.dumps({'data': 'foo'})))

        try:
            w.record()
        except KeyboardInterrupt:
            # we raised that exception to stop the infinite loop
            pass

        w._store.update_leaderboard.assert_called_once_with({'data': 'foo'})