from unittest import TestCase

from gamelib.lab import Lab
from gamelib import research, products


class TestLab(TestCase):
    def setUp(self):
        self.lab = Lab()

    def test_find_new_products(self):
        new_products = self.lab.find_new_products(None)
        self.assertTrue(products.MachineGun in new_products)
        self.assertTrue(products.DoomsdayVirus not in new_products)

    def test_find_new_research(self):
        new_research = self.lab.find_new_research(None)
        self.assertTrue(research.Tesla in new_research)
        self.assertTrue(research.Space not in new_research)