#!/usr/bin/python
# Copyright 2012 Google Inc. All Rights Reserved.
#
# Disable some checks that aren't important for tests:
#gpylint: disable-msg=E0602,C6409,C6111,C0111

__author__ = 'Avery Pennarun (apenwarr@google.com)'

from wvtest import *
import core


@wvtest
def testConfigCore():
  hosts = core.Hosts()
  WVPASSEQ(len(hosts), 0)
  h = hosts.FindOrAdd(ether='11:22:33:44:55:66')
  h.Set(name='testname', ip='1.2.3.4')
  h.platform = 'bob'
  WVPASSEQ(len(hosts), 1)
  h2 = hosts.FindOrAdd(ip='1.2.3.4')
  WVPASSEQ(len(hosts), 1)
  WVPASSEQ(h2.name, 'testname')
  WVPASSEQ(h2.ether, '11:22:33:44:55:66')
  WVPASSEQ(h, h2)

  # test Host.Query()
  WVPASSEQ(hosts.Query(ip=True), [h])
  WVPASSEQ(hosts.Query(ip=False), [])
  WVPASSEQ(hosts.Query(ip='1.2.3.4'), [h])
  WVPASSEQ(hosts.Query(ip='1.2.3.5'), [])
  WVPASSEQ(hosts.Query(ip='1.2.3.4', maxcount=0), [])
  WVEXCEPT(core.QueryError, hosts.Query, ip='1.2.3.5', mincount=1)