def test_poll(self):
flexmock(helper).should_receive('get_deployment_id').and_return(None)
poll()
flexmock(helper).should_receive('get_deployment_id').\
and_return('deployment_id')
flexmock(json).should_receive('dumps').and_return("data")
flexmock(helper).should_receive('create_request').and_return()
flexmock(helper).should_receive('urlfetch').and_return(FakeResponse())
flexmock(json).should_receive('loads').and_return({})
flexmock(helper).should_receive('urlfetch_async').and_return()
poll()
def test_poll(self):
# Assume deployment is not registered.
flexmock(helper).should_receive('get_deployment_id').and_return(None)
poll()
flexmock(helper).should_receive('get_deployment_id').\
and_return('deployment_id')
br_nodes = [
{
helper.NodeInfoTags.HOST: 'node1'
},
{
helper.NodeInfoTags.HOST: 'node2'
},
]
flexmock(helper).should_receive('get_node_info').and_return(br_nodes)
# Assume backup and recovery service is down.
http_client = flexmock()
http_client.should_receive('fetch').and_raise(socket.error)
flexmock(tornado.httpclient).should_receive('HTTPClient').\
and_return(http_client)
poll()
# Assume backup and recovery service is up.
response = flexmock(body=json.dumps({'status': 'up'}))
http_client.should_receive('fetch').and_return(response)
flexmock(tornado.httpclient).should_receive('HTTPClient').\
and_return(http_client)
flexmock(json).should_receive('dumps').and_return('{}')
flexmock(helper).should_receive('create_request').and_return()
flexmock(helper).should_receive('urlfetch').and_return(FakeResponse())
flexmock(helper).should_receive('urlfetch_async').and_return()
poll()
def test_poll(self):
# Assume deployment is not registered.
flexmock(helper).should_receive('get_deployment_id').and_return(None)
poll()
flexmock(helper).should_receive('get_deployment_id').\
and_return('deployment_id')
br_nodes = [
{helper.NodeInfoTags.HOST: 'node1'},
{helper.NodeInfoTags.HOST: 'node2'},
]
flexmock(helper).should_receive('get_node_info').and_return(br_nodes)
# Assume backup and recovery service is down.
http_client = flexmock()
http_client.should_receive('fetch').and_raise(socket.error)
flexmock(tornado.httpclient).should_receive('HTTPClient').\
and_return(http_client)
poll()
# Assume backup and recovery service is up.
response = flexmock(body=json.dumps({'status': 'up'}))
http_client.should_receive('fetch').and_return(response)
flexmock(tornado.httpclient).should_receive('HTTPClient').\
and_return(http_client)
flexmock(json).should_receive('dumps').and_return('{}')
flexmock(helper).should_receive('create_request').and_return()
flexmock(helper).should_receive('urlfetch').and_return(FakeResponse())
flexmock(helper).should_receive('urlfetch_async').and_return()
poll()
