#!/usr/bin/env python
import unittest, os, logging
from geopy import Point
from geopy.parsers import gpx

from datetime import datetime
from geopy.parsers.iso8601 import TimeZone

log = logging.getLogger('geopy.parsers.gpx')

TEST_FILE = os.path.join(
              os.path.dirname(__file__), 'fells_loop.gpx')
            # From http://www.topografix.com/fells_loop.gpx

class GPX(object):
    def test_version(self):
        self.failUnless(self.GPXi.version == '1.1')

    def test_creator(self):
        self.failUnless(
          self.GPXi.creator == 'ExpertGPS 1.1 - http://www.topografix.com')

    def test_route_names(self):
        names = list(self.GPXi.route_names)
        self.failUnless(names == ['BELLEVUE (first three waypoints)',
                                  'BELLEVUE'])

    def test_route_list(self):
        route = list(self.GPXi.get_waypoints(
            'BELLEVUE (first three waypoints)'))
        self.failUnless(route == [Point(42.430950, -71.107628, 23.469600),
                                  Point(42.431240, -71.109236, 26.561890),
                                  Point(42.434980, -71.109942, 45.307495)])
        second_top_level_waypoint = list(self.GPXi.waypoints)[1]
        self.failUnless(second_top_level_waypoint ==
                          Point(42.439227, -71.119689, 57.607200))
        self.failUnless(second_top_level_waypoint.timestamp ==
          datetime(2001, 6, 2, 3, 26, 55, tzinfo = TimeZone('UTC')))
        self.failUnless(second_top_level_waypoint.name == '5067')
        self.failUnless(second_top_level_waypoint.description == '5067')

class TestInit(GPX, unittest.TestCase):
    def setUp(self):
        f = open(TEST_FILE, 'r')
        self.GPXi = gpx.GPX(f)

class TestParse(GPX, unittest.TestCase):
    def setUp(self):
        self.GPXi = gpx.GPX()
        f = open(TEST_FILE, 'r')
        self.GPXi.open(f)