tests/testthat/test-DoordaHostCursor.R

In Doorda/doorda-r-sdk: DBI Connector to DoordaHost

context('DoordaHostCursor')

source('utilities.R')

test_that('DoordaHostCursor methods work correctly', {
  post.content <- list(stats=list(
    state=jsonlite::unbox('INITIAL')
  ))
  post.response <- structure(
    list(
      url='http://localhost:8000/v1/statement',
      status_code=200,
      headers=list(
        'content-type'='application/json'
      ),
      content=charToRaw(jsonlite::toJSON(post.content))
    ),
    class='response'
  )
  
  cursor <- DoordaHostCursor(post.response)
  expect_equal(cursor$infoUri(), '')
  expect_equal(cursor$nextUri(), '')
  expect_equal(cursor$state(), 'INITIAL')
  expect_equal(cursor$fetchedRowCount(), 0)
  expect_equal(cursor$stats(), list(state='INITIAL'))
  expect_true(cursor$hasCompleted())
  expect_equal(cursor$postResponse(), post.response)
  expect_true(cursor$postDataParsed())
  expect_true(cursor$postDataParsed(FALSE))
  expect_false(cursor$postDataParsed(TRUE))
  
  cursor$state('__TEST')
  expect_equal(cursor$state(), '__TEST')
  
  uri <- 'http://localhost:8000/query_1/1'
  r <- mock_httr_response(
    url='http://localhost:8000/v1/statement',
    status_code=200,
    state='RUNNING',
    next_uri=uri,
    info_uri='http://localhost:8000/v1/query/query_1'
  )
  content <- DoordaHostSDK:::response.to.content(r[['response']])
  cursor$updateCursor(content, 1)
  
  expect_equal(
    cursor$infoUri(),
    'http://localhost:8000/v1/query/query_1'
  )
  expect_equal(cursor$nextUri(), uri)
  expect_equal(cursor$state(), 'RUNNING')
  expect_equal(cursor$fetchedRowCount(), 1)
  expect_is(cursor$stats(), 'list')
  expect_false(cursor$hasCompleted())
  
  r <- mock_httr_response(
    url=uri,
    status_code=200,
    info_uri='http://localhost:8000/v1/query/query_1_updated',
    state='FINISHED'
  )
  content <- DoordaHostSDK:::response.to.content(r[['response']])
  cursor$updateCursor(content, 1)
  
  expect_equal(
    cursor$infoUri(),
    'http://localhost:8000/v1/query/query_1_updated'
  )
  expect_equal(cursor$nextUri(), '')
  expect_equal(cursor$state(), 'FINISHED')
  expect_equal(cursor$fetchedRowCount(), 2)
  expect_is(cursor$stats(), 'list')
  expect_true(cursor$hasCompleted())
})

test_that('DoordaHostCursor methods work correctly with POST data', {
  response <- mock_httr_response(
    url='http://localhost:8000/v1/statement',
    status_code=200,
    state='FINISHED',
    data=data.frame(x=1)
  )[['response']]
  cursor <- DoordaHostCursor(response)
  expect_equal(cursor$infoUri(), '')
  expect_equal(cursor$nextUri(), '')
  expect_equal(cursor$state(), 'FINISHED')
  expect_equal(cursor$fetchedRowCount(), 0)
  expect_equal(cursor$stats(), list(state='FINISHED'))
  expect_false(cursor$hasCompleted())
  expect_equal(cursor$postResponse(), response)
  expect_false(cursor$postDataParsed())
  expect_false(cursor$postDataParsed(TRUE))
  expect_true(cursor$postDataParsed())
  expect_true(cursor$hasCompleted())
})