コード例 #1
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
 def test_take_snapshot(self):
     node = NodeServer.create(ADDRESS)
     node.sync({'some_key': 'some_value'})
     response = node.take_snapshot()
     self.assertTrue(response.success)
     self.assertIsInstance(response, Success)
     self.assertEqual(response.message, {'some_key': 'some_value'})
コード例 #2
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
    def test_get(self):
        node = NodeServer.create(ADDRESS)
        node.set(key='some_key', value='some_value')
        self.assertEqual(node._cache_class, {'some_key': 'some_value'})

        response = node.get(key='some_key')
        self._validate_operation_response(response, 'some_value')
コード例 #3
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
    def test_keys(self):
        node = NodeServer.create(ADDRESS)
        node.set(key='some_key1', value='some_value1')
        node.set(key='some_key2', value='some_value2')
        self.assertEqual(
            node._cache_class,
            {'some_key1': 'some_value1', 'some_key2': 'some_value2'}
        )

        response = node.keys(pattern='some_key*')
        self._validate_operation_response(
            response, ['some_key1', 'some_key2']
        )
コード例 #4
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
 def test_create(self):
     allowed_methods = (
         'ping', 'set', 'get', 'mget', 'delete',
         'keys', 'sync', 'take_snapshot'
     )
     server = NodeServer.create(ADDRESS)
     self.assertEqual(server.__allowed_methods__, allowed_methods)
     self.assertEqual(server.__serializer__, MockJSONSerializer)
     self.assertIsInstance(server.factory, ZmqFactory)
     self.assertIsInstance(server.endpoints[0], ZmqEndpoint)
     self.assertEqual(server._address, ADDRESS)
     self.assertFalse(server._is_registered)
     self.assertEqual(type(server._cache_class), InMemoryCache)
     server.shutdown()
コード例 #5
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
    def test_register_with_broker_fail(self):
        class _MockBroker(MockBroker):
            def register_node(self, *args, **kwargs):
                return defer.succeed(
                    {'message': 'dummy_error', 'success': False}
                )

        err_back = Mock(
            side_affect=lambda: defer.succeed(None),
            return_value=defer.succeed(None)
        )

        node = NodeServer.create(ADDRESS)
        d = node.register_with_broker(_MockBroker(), 'some_address')
        d.addErrback(err_back)
        d.addCallback(lambda _: self.assertFalse(node._is_registered))
        d.addCallback(lambda _: self.assertTrue(err_back.called))
        return d
コード例 #6
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
 def test_sync(self):
     node = NodeServer.create(ADDRESS)
     node.sync({'some_key': 'some_value'})
     self.assertEqual(node._cache_class, {'some_key': 'some_value'})
コード例 #7
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
 def test_ping(self):
     node = NodeServer.create(ADDRESS)
     self.assertIsInstance(node.ping(), Success)
コード例 #8
0
ファイル: test_node_server.py プロジェクト: dr4ke616/pinky
    def test_register_with_broker(self):

        node = NodeServer.create(ADDRESS)
        d = node.register_with_broker(MockBroker(), 'some_address')
        d.addCallback(lambda _: self.assertTrue(node._is_registered))
        return d