import sys
import os
import pytest 

sys.path.insert(1, os.path.join(sys.path[0], '../../'))
import rips

import dataroot

def test_Launch(rips_instance, initializeTest):
    assert(rips_instance is not None)

def test_EmptyProject(rips_instance, initializeTest):
    cases = rips_instance.project.cases()
    assert(len(cases) is 0)

def test_OneCase(rips_instance, initializeTest):
    case = rips_instance.project.loadCase(dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID")
    assert(case.name == "TEST10K_FLT_LGR_NNC")
    assert(case.id == 0)
    cases = rips_instance.project.cases()
    assert(len(cases) is 1)

def test_MultipleCases(rips_instance, initializeTest):
    casePaths = []
    casePaths.append(dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID")
    casePaths.append(dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID")
    casePaths.append(dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID")
    
    caseNames = []
    for casePath in casePaths:
        caseName = os.path.splitext(os.path.basename(casePath))[0]
        caseNames.append(caseName)
        rips_instance.project.loadCase(path=casePath)

    cases = rips_instance.project.cases()
    assert(len(cases) == len(caseNames))
    for i, caseName in enumerate(caseNames):
        assert(caseName == cases[i].name)

def test_10k(rips_instance, initializeTest):
    casePath = dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID"
    case = rips_instance.project.loadCase(path=casePath)
    assert(case.grid_count() == 2)
    cellCountInfo = case.cell_count()
    assert(cellCountInfo.active_cell_count == 11125)
    assert(cellCountInfo.reservoir_cell_count == 316224)
    timeSteps = case.time_steps()
    assert(len(timeSteps) == 9)
    daysSinceStart = case.days_since_start()
    assert(len(daysSinceStart) == 9)

def test_PdmObject(rips_instance, initializeTest):
    casePath = dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID"
    case = rips_instance.project.loadCase(path=casePath)
    assert(case.id == 0)
    assert(case.address() is not 0)
    assert(case.classKeyword() == "EclipseCase")
    caseId = case.getValue('CaseId')
    assert(caseId == case.id)

@pytest.mark.skipif(sys.platform.startswith('linux'), reason="Brugge is currently exceptionally slow on Linux")
def test_brugge_0010(rips_instance, initializeTest):
    casePath = dataroot.PATH + "/Case_with_10_timesteps/Real10/BRUGGE_0010.EGRID"
    case = rips_instance.project.loadCase(path=casePath)
    assert(case.grid_count() == 1)
    cellCountInfo = case.cell_count()
    assert(cellCountInfo.active_cell_count == 43374)
    assert(cellCountInfo.reservoir_cell_count == 60048)
    timeSteps = case.time_steps()
    assert(len(timeSteps) == 11)
    daysSinceStart = case.days_since_start()
    assert(len(daysSinceStart) == 11)

@pytest.mark.skipif(sys.platform.startswith('linux'), reason="Brugge is currently exceptionally slow on Linux")
def test_replaceCase(rips_instance, initializeTest):
    project = rips_instance.project.open(dataroot.PATH + "/TEST10K_FLT_LGR_NNC/10KWithWellLog.rsp")
    casePath = dataroot.PATH + "/Case_with_10_timesteps/Real0/BRUGGE_0000.EGRID"
    case = project.case(id=0)
    assert(case is not None)
    assert(case.name == "TEST10K_FLT_LGR_NNC")
    assert(case.id == 0)
    cases = rips_instance.project.cases()
    assert(len(cases) is 1)

    rips_instance.commands.replaceCase(newGridFile=casePath, caseId=case.id)
    cases = rips_instance.project.cases()
    assert(len(cases) is 1)
    case = project.case(id=0)
    assert(case.name == "Real0--BRUGGE_0000.EGRID")
    assert(case.id == 0)