# This file is part of NIT ( http://www.nitlanguage.org ).
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

module test_api is test

intrude import api

# GithubAPI testing
#
# To avoid test flakyness we test the GithubAPI against a mock of the real one.
# For each api request we return a cache file of the real API response body.
#
# Cache files can be automatically created and updated by setting
# `update_responses_cache` to `true` then running `nitunit`.
class MockGithubAPI
	super GithubAPI

	# Mock so it returns the response from a file
	#
	# See `update_responses_cache`.
	redef fun send(method, path, headers, body) do
		print path # for debugging

		assert has_response(path)

		if update_responses_cache then
			var file = response_file(path)
			save_actual_response(path, file)
		end

		var response = response_string(path)
		if response_is_error(path) then
			last_error = new GithubAPIError(
				response.parse_json.as(JsonObject)["message"].as(String),
				response_code(path).to_i,
				path
			)
			was_error = true
			return null
		end
		return response
	end

	var test_responses: Map[String, String] do
		var map = new HashMap[String, String]
		map["/user"] = "user_Morriar"
		map["/users/Morriar"] = "user_Morriar"
		map["/repos/nitlang/nit"] = "repo_nit"
		map["/repos/nitlang/nit/labels?page=1&per_page=3"] = "repo_labels_nit"
		map["/repos/nitlang/nit/labels/ok_will_merge"] = "repo_labels_ok_will_merge"
		map["/repos/nitlang/nit/milestones?page=1&per_page=3"] = "repo_milestones_nit"
		map["/repos/nitlang/nit/milestones/4"] = "repo_milestones_4"
		map["/repos/nitlang/nit/branches?page=1&per_page=2"] = "repo_branches_nit"
		map["/repos/nitlang/nit/branches/master"] = "repo_branches_master"
		map["/repos/nitlang/nit/issues?page=1&per_page=3"] = "repo_issues_nit"
		map["/repos/nitlang/nit/issues/1000"] = "repo_issues_1000"
		map["/repos/nitlang/nit/issues/1000/comments?page=1&per_page=3"] = "repo_issues_comments_nit"
		map["/repos/nitlang/nit/issues/comments/6020149"] = "repo_issues_comments_6020149"
		map["/repos/nitlang/nit/issues/1000/events?page=1&per_page=3"] = "repo_issues_events_nit"
		map["/repos/nitlang/nit/issues/events/199674194"] = "repo_issues_events_199674194"
		map["/repos/nitlang/nit/pulls?page=1&per_page=3"] = "repo_pulls_nit"
		map["/repos/nitlang/nit/pulls/1000"] = "repo_pulls_1000"
		map["/repos/nitlang/nit/pulls/945/comments?page=1&per_page=3"] = "repo_pulls_945_comments"
		map["/repos/nitlang/nit/pulls/comments/21010363"] = "repo_pulls_comment_21010363"
		map["/repos/nitlang/nit/commits/64ce1f"] = "repo_commits_64ce1f"
		map["/repos/nitlang/nit/commits/4e3c688d/status"] = "repo_commits_4e3c68_status"
		map["/repos/nitlang/nit/comments/8982707"] = "repo_comments_8982707"
		map["/search/issues?q=foo repo:nitlang/nit&page=1&per_page=3"] = "repo_search_issues_nit"
		map["/repos/nitlang/nit/stats/contributors"] = "repo_nit_contributors"
		# errors
		map["/users/not_found/not_found"] = "errors_404"
		return map
	end

	# Does `self` have a mock response for Github `path`?
	fun has_response(path: String): Bool do
		return test_responses.has_key(path)
	end

	# Root responses cache directory
	var responses_dir: String is lazy do
		var path = "NIT_TESTING_PATH".environ.dirname / "mock"
		path.mkdir
		return path
	end

	# Returns the response file path for a Github `path`
	fun response_file(path: String): String do
		assert has_response(path)
		return "{responses_dir / test_responses[path]}.res"
	end

	# Returns the response body string for a Github `path`
	fun response_string(path: String): String do
		var file = response_file(path)
		assert file.file_exists
		return file.to_path.read_all
	end

	# Is this response a simulated error?
	fun response_is_error(path: String): Bool do
		assert has_response(path)
		return test_responses[path].has_prefix("errors_")
	end

	# Status code of a simulated error
	#
	# See `response_is_error`.
	fun response_code(path: String): String do
		assert response_is_error(path)
		return test_responses[path].split("_").last
	end

	# Response caching

	# Activate caching
	#
	# Change this value to `true` then run nitunit to cache the responses
	# from the Github API.
	#
	# Default is `false`.
	var update_responses_cache = false

	# Save the actual Github API response body for `uri` to a `file`
	private fun save_actual_response(uri, file: String) do
		assert update_responses_cache

		var request = new CurlHTTPRequest("{api_url}{sanitize_uri(uri)}")
		request.user_agent = actual_api.user_agent
		request.headers = actual_api.new_headers
		var response = request.execute

		if response isa CurlResponseSuccess then
			response.body_str.write_to_file(file)
		else if response isa CurlResponseFailed then
			response.error_msg.write_to_file(file)
		else abort

		print "Response to `{uri}` saved at `{file}`"
	end

	# Actual GithubCurl instance used for caching
	private var actual_api = new GithubAPI(get_github_oauth, "nitunit")
end

class TestGithubAPI
	test

	fun api: MockGithubAPI do return new MockGithubAPI("test", "test")

	fun test_deserialize is test do
		var response = api.response_string("/users/Morriar")
		var obj = api.deserialize(response)
		assert obj isa User
		assert obj.login == "Morriar"
	end

	fun test_sanitize_url is test do
		# TODO better tests
		assert api.sanitize_uri("/repos/Nit with spaces/") == "/repos/Nit%20with%20spaces/"
	end

	fun test_get is test do
		var api = self.api
		var obj = api.get("/users/Morriar")
		assert not api.was_error
		assert api.last_error == null
		assert obj isa User
		assert obj.login == "Morriar"
	end

	fun test_get_404 is test do
		var api = self.api
		var res = api.get("/users/not_found/not_found")
		assert res == null
		assert api.was_error
		var err = api.last_error
		assert err isa GithubAPIError
		assert err.status_code == 404
		assert err.message == "Not Found"
	end

	# TODO test more error cases

	fun test_get_auth_user is test do
		var user = api.get_auth_user
		assert user isa User
		assert user.login == "Morriar"
		assert user.avatar_url == "https://avatars2.githubusercontent.com/u/583144?v=4"
		assert user.name == "Alexandre Terrasa"
		assert user.email == "alexandre@moz-code.org"
		assert user.blog == "moz-code.org"
	end

	fun test_get_user is test do
		var user = api.get_user("Morriar")
		assert user isa User
		assert user.login == "Morriar"
		assert user.avatar_url == "https://avatars2.githubusercontent.com/u/583144?v=4"
		assert user.name == "Alexandre Terrasa"
		assert user.email == "alexandre@moz-code.org"
		assert user.blog == "moz-code.org"
	end

	fun test_get_repo is test do
		var repo = api.get_repo("nitlang/nit")
		assert repo isa Repo
		assert repo.full_name == "nitlang/nit"
		assert repo.name == "nit"
		assert repo.owner.login == "nitlang"
		assert repo.default_branch == "master"
	end

	fun test_get_branches is test do
		var branches = api.get_repo_branches("nitlang/nit", 1, 2)
		assert branches.length == 2
		assert branches.first.name == "master"
		assert branches.last.name == "next"
	end

	fun test_get_issues is test do
		var issues = api.get_repo_issues("nitlang/nit", 1, 3)
		assert issues.length == 3
		assert issues.first.title == "nitrpg: Move `nitrpg` to its own repository"
		assert issues.last.title == "Mock Github API tests"
	end

	fun test_search_issues is test do
		var results = api.search_repo_issues("nitlang/nit", "foo", 1, 3)
		assert results isa SearchResults
		assert results.items.length == 3
		assert results.items.first.as(Issue).title == "Introduction of contracts in Nit"
		assert results.items.last.as(Issue).title == "Appel de méthodes abstraites non redéfinies"
	end

	fun test_get_labels is test do
		var labels = api.get_repo_labels("nitlang/nit", 1, 3)
		assert labels.length == 3
		assert labels.first.name == "API"
		assert labels.last.name == "NEP"
	end

	fun test_get_milestones is test do
		var milestones = api.get_repo_milestones("nitlang/nit", 1, 3)
		assert milestones.length == 3
		assert milestones.first.title == "v1.0prealpha"
		assert milestones.last.title == "nitdoc - Abstraction levels"
	end

	fun test_get_pulls is test do
		var pulls = api.get_repo_pulls("nitlang/nit", 1, 3)
		assert pulls.length == 3
		assert pulls.first.title == "nitrpg: Move `nitrpg` to its own repository"
		assert pulls.last.title == "Mock Github API tests"
	end

	fun test_get_branch is test do
		var branch = api.get_branch("nitlang/nit", "master")
		assert branch isa Branch
		assert branch.name == "master"
	end

	fun test_get_commit is test do
		var commit = api.get_commit("nitlang/nit", "64ce1f")
		assert commit isa Commit
		assert commit.sha == "64ce1f587209024f5de46d06c70526a569ff537f"
		# TODO other fields
	end

	fun test_get_commit_status is test do
		var status = api.get_commit_status("nitlang/nit", "4e3c688d")
		assert status isa CommitStatus
		assert status.state == "failure"
		assert status.sha == "4e3c688d2c4b875c00f206eb4c4b6f2c4f34c096"
		assert status.total_count == 1

		var sub = status.statuses.first
		assert sub.state == "failure"
		assert sub.description == "Merged pipeline on gitlab: failed"
		assert sub.context == "gitlab-ci"

		var repo = status.repository
		assert repo isa Repo
		assert repo.full_name == "nitlang/nit"
	end

	fun test_get_issue is test do
		var issue = api.get_issue("nitlang/nit", 1000)
		assert issue isa Issue
		assert issue.number == 1000
		assert issue.title == "Raise nitc from the dead"
		assert issue.user.as(User).login == "privat"
		assert issue.comments == 7
		assert issue.created_at == "2014-12-11T02:55:09Z"
		assert issue.closed_at == "2014-12-13T15:38:09Z"
		assert issue.closed_by.as(User).login == "privat"
		assert issue.body == "Raise dead on `nitc`.\nIt's super effective...\n"
		assert issue.is_pull_request
	end

	fun test_get_issue_comments is test do
		var comments = api.get_issue_comments("nitlang/nit", 1000, 1, 3)
		assert comments.length == 3
		assert comments.first.user.login == "R4PaSs"
		assert comments.last.user.login == "xymus"
	end

	fun test_get_issue_events is test do
		var events = api.get_issue_events("nitlang/nit", 1000, 1, 3)
		assert events.length == 3
		assert events.first.actor.login == "privat"
		assert events.last.actor.login == "xymus"
	end

	fun test_get_pull is test do
		var pull = api.get_pull("nitlang/nit", 1000)
		assert pull isa PullRequest
		assert pull.number == 1000
		assert pull.title == "Raise nitc from the dead"
		assert pull.user.as(User).login == "privat"
		assert pull.comments == 7
		assert pull.created_at == "2014-12-11T02:55:09Z"
		assert pull.closed_at == "2014-12-13T15:38:09Z"
		assert pull.merged_by.as(User).login == "privat"
		assert pull.body == "Raise dead on `nitc`.\nIt's super effective...\n"
	end

	fun test_get_pull_comments is test do
		var comments = api.get_pull_comments("nitlang/nit", 945, 1, 3)
		assert comments.length == 2
		assert comments.first.path == "src/modelize/modelize_property.nit"
	end

	fun test_get_pull_comment is test do
		var comment = api.get_pull_comment("nitlang/nit", 21010363)
		assert comment isa PullComment
		assert comment.path == "src/modelize/modelize_property.nit"
		assert comment.original_position == 26
		assert comment.pull_number == 945
		# TODO other fields
	end

	fun test_get_label is test do
		var labl = api.get_label("nitlang/nit", "ok_will_merge")
		assert labl isa Label
		assert labl.name == "ok_will_merge"
	end

	fun test_get_milestone is test do
		var milestone = api.get_milestone("nitlang/nit", 4)
		assert milestone isa Milestone
		assert milestone.title == "v1.0prealpha"
		# TODO other fields
	end

	fun test_get_issue_event is test do
		var event = api.get_issue_event("nitlang/nit", 199674194)
		assert event isa IssueEvent
		assert event.actor.login == "privat"
		assert event.event == "labeled"
		assert event.labl.as(Label).name == "need_review"
	end

	fun test_get_issue_comment is test do
		var comment = api.get_issue_comment("nitlang/nit", 6020149)
		assert comment isa IssueComment
		assert comment.user.login == "privat"
		assert comment.created_at.to_s == "2012-05-30T20:16:54Z"
		assert comment.issue_number == 10
	end

	fun test_get_comment is test do
		var comment = api.get_commit_comment("nitlang/nit", 8982707)
		assert comment isa CommitComment
		assert comment.user.login == "Morriar"
		assert comment.body == "For testing purposes...\n"
		assert comment.commit_id == "7eacb86d1e24b7e72bc9ac869bf7182c0300ceca"
	end

	fun test_contributor_stats is test do
		var stats = api.get_repo_contrib_stats("nitlang/nit")
		assert stats.last.author.login == "privat"
		assert stats.last.total == 4536
		assert stats.last.weeks.length == 575
	end
end