#!/usr/bin/python
import unittest
from survery import AnonymousServery

class TestSurveyClass(unittest.TestCase):
    """Class to test a another class module"""
    def setUp(self):
        """
        Create a survey and a set of
        responses for use in all test methods.
        """
        question = "What language did you first learn to speak?"
        self.my_survey = AnonymousServery(question)
        self.responses = ['English', 'Spanish', 'Korean']


    def test_store_single_response(self):
        """Test that a single response is stored properly"""

        self.my_survey.store_response(self.responses[0])
        self.assertIn(self.responses[0], self.my_survey.responses)


    def test_multiple_responses(self):
        """Test running a loop through multiple responses"""

        for response in self.responses:
            self.my_survey.store_response(response)
        for response in self.responses:
            self.assertIn(response, self.my_survey.responses)


unittest.main()