"""Tests for tune_by_instrument endpoints."""
from tests.conftest import AUTH


def make_tune(client):
    r = client.post("/tunes/", json={"name": "Test Tune", "key": "D"}, headers=AUTH)
    assert r.status_code == 201
    return r.get_json()


def make_musician(client, name="Earl Scruggs"):
    r = client.post("/musicians/", json={"name": name}, headers=AUTH)
    assert r.status_code == 201
    return r.get_json()


def get_instrument_id(client, name):
    return next(i["id"] for i in client.get("/instruments").get_json() if i["name"] == name)


def make_tuning(client, name="double C"):
    r = client.post("/tunings/", json={"name": name}, headers=AUTH)
    assert r.status_code == 201
    return r.get_json()


class TestTuneByInstrument:
    def test_create_entry(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        tuning   = make_tuning(client, "double C")
        r = client.post(f'/tunes/{tune["id"]}/instruments',
                        json={"instrument_id": banjo_id, "tuning_id": tuning["id"], "callable": True},
                        headers=AUTH)
        assert r.status_code == 201
        body = r.get_json()
        assert body["instrument_name"] == "banjo"
        assert body["tuning"] == "double C"
        assert body["callable"] is True

    def test_create_with_musician(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        m        = make_musician(client)
        r = client.post(f'/tunes/{tune["id"]}/instruments',
                        json={"instrument_id": banjo_id, "learned_from_id": m["id"]},
                        headers=AUTH)
        assert r.status_code == 201
        assert r.get_json()["learned_from"] == "Earl Scruggs"

    def test_update_entry(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        tuning   = make_tuning(client, "sawmill")
        entry = client.post(f'/tunes/{tune["id"]}/instruments',
                            json={"instrument_id": banjo_id, "callable": False},
                            headers=AUTH).get_json()
        r = client.patch(f'/tunes/{tune["id"]}/instruments/{entry["id"]}',
                         json={"callable": True, "tuning_id": tuning["id"]}, headers=AUTH)
        assert r.status_code == 200
        assert r.get_json()["callable"] is True
        assert r.get_json()["tuning"] == "sawmill"

    def test_delete_entry(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        entry = client.post(f'/tunes/{tune["id"]}/instruments',
                            json={"instrument_id": banjo_id}, headers=AUTH).get_json()
        r = client.delete(f'/tunes/{tune["id"]}/instruments/{entry["id"]}', headers=AUTH)
        assert r.status_code == 200
        remaining = client.get(f'/tunes/{tune["id"]}/instruments').get_json()
        assert all(e["id"] != entry["id"] for e in remaining)

    def test_unique_constraint(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        client.post(f'/tunes/{tune["id"]}/instruments',
                    json={"instrument_id": banjo_id}, headers=AUTH)
        r = client.post(f'/tunes/{tune["id"]}/instruments',
                        json={"instrument_id": banjo_id}, headers=AUTH)
        assert r.status_code == 409


class TestInstrumentNotes:
    def test_add_and_list_notes(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        entry = client.post(f'/tunes/{tune["id"]}/instruments',
                            json={"instrument_id": banjo_id}, headers=AUTH).get_json()
        client.post(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes',
                    json={"note": "tricky B part"}, headers=AUTH)
        client.post(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes',
                    json={"note": "watch the timing"}, headers=AUTH)
        notes = client.get(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes').get_json()
        assert len(notes) == 2

    def test_update_note(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        entry = client.post(f'/tunes/{tune["id"]}/instruments',
                            json={"instrument_id": banjo_id}, headers=AUTH).get_json()
        note = client.post(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes',
                           json={"note": "old"}, headers=AUTH).get_json()
        r = client.patch(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes/{note["id"]}',
                         json={"note": "new"}, headers=AUTH)
        assert r.get_json()["note"] == "new"

    def test_delete_note(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        entry = client.post(f'/tunes/{tune["id"]}/instruments',
                            json={"instrument_id": banjo_id}, headers=AUTH).get_json()
        note = client.post(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes',
                           json={"note": "to delete"}, headers=AUTH).get_json()
        r = client.delete(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes/{note["id"]}',
                          headers=AUTH)
        assert r.status_code == 200

    def test_notes_included_in_entry_to_dict(self, client):
        tune     = make_tune(client)
        banjo_id = get_instrument_id(client, "banjo")
        entry = client.post(f'/tunes/{tune["id"]}/instruments',
                            json={"instrument_id": banjo_id}, headers=AUTH).get_json()
        client.post(f'/tunes/{tune["id"]}/instruments/{entry["id"]}/notes',
                    json={"note": "my note"}, headers=AUTH)
        # Notes appear in the instrument entry and in the parent tune
        tune_data = client.get(f'/tunes/{tune["id"]}').get_json()
        inst_entry = tune_data["instruments"][0]
        assert len(inst_entry["notes"]) == 1
        assert inst_entry["notes"][0]["note"] == "my note"