def __polling(cls, json_data):
while json_data["continue"] == 1:
sleep(2)
query = BookSearchQuery()
query.set("session", json_data["session"])
json_data = CalilService.__polling_request(
CalilQuery.adjust_next_query(query))
return json_data
def test_request(self, mock_requests):
mock_response = MagicMock()
mock_response.json.return_value = ["test"]
mock_requests.get.return_value = mock_response
query = BookSearchQuery()
json_data = RakutenBooksService._RakutenBooksService__request(query)
assert mock_requests.get.called
assert mock_response.json.called
assert json_data == ["test"]
def test_polling_request(self):
with patch("kbot.book.calil.CalilService._CalilService__request_sub"
) as mock:
response_mock = MagicMock()
response_mock.text = 'callback({"books": {"1111":{"system1":{"test":"hoge"}}}} )'
mock.return_value = response_mock
query = BookSearchQuery()
result = CalilService._CalilService__polling_request(query)
mock.assert_called_once()
assert json.dumps(
result) == '{"books": {"1111": {"system1": {"test": "hoge"}}}}'
def test_query(self):
query = BookSearchQuery()
query.set("test", "hoge")
assert query.dict().get("test") == "hoge"
def test_adjust_query(self):
query = BookSearchQuery()
RakutenBooksQuery.adjust_query(query)
assert query.dict().get("applicationId") == os.environ["RAKUTEN_APP_ID"]
assert query.dict().get("sort") == "sales"
def test_search_books(self):
query = BookSearchQuery()
query.set("title", "カンブリア")
rakuten_books = RakutenBooksService.search_books(query)
print(rakuten_books.slice(0, 5).get_message())
def test_get_one_book(self, mock_method, json_str, title):
mock_method.return_value = json.loads(json_str)
query = BookSearchQuery()
rakuten_book = RakutenBooksService.get_one_book(query)
assert mock_method.called
assert rakuten_book.title == title
def test_adjust_next_query(self):
query = BookSearchQuery()
query.set("foo", "bar")
result = CalilQuery.adjust_next_query(query)
assert result.get("foo") == query.get("foo")
def test_calil_service(self):
query = BookSearchQuery()
query.set("isbn", "9784532280208")
calil_book = CalilService.get_one_book(query)
print(calil_book.get_text_message())