green-spider/checks/green_power_test.py
2019-08-31 12:49:07 +02:00

22 lines
863 B
Python

import unittest
from checks import green_power
from checks.config import Config
class TestGreenPower(unittest.TestCase):
def test_is_green(self):
"""Check that we get a green result for green powered site"""
# google is green, Koch Industries. predictably is not
# TODO mock the result, so we don't need to hit a live API
config = Config(urls=['https://google.com/'])
checker = green_power.Checker(config=config)
result = checker.run()
self.assertEqual(result['https://google.com/'], True)
def test_is_not_green(self):
"""Check that we get a grey result for grey powered site"""
config = Config(urls=['http://www.kochind.com/'])
checker = green_power.Checker(config=config)
result = checker.run()
self.assertEqual(result['http://www.kochind.com/'], False)