from argenta.command.params.flag import Flag

import unittest
import re


class TestFlag(unittest.TestCase):
    def test_get_string_entity(self):
        self.assertEqual(Flag(flag_name='test').get_string_entity(),
                         '--test')

    def test_get_string_entity2(self):
        self.assertEqual(Flag(flag_name='test',
                              flag_prefix='---').get_string_entity(),
                         '---test')

    def test_get_flag_name(self):
        self.assertEqual(Flag(flag_name='test').get_flag_name(),
                         'test')

    def test_get_flag_prefix(self):
        self.assertEqual(Flag(flag_name='test').get_flag_prefix(),
                         '--')

    def test_get_flag_prefix2(self):
        self.assertEqual(Flag(flag_name='test',
                              flag_prefix='--').get_flag_prefix(),
                         '--')

    def test_get_flag_value_without_set(self):
        self.assertEqual(Flag(flag_name='test').get_value(),
                         None)

    def test_get_flag_value_with_set(self):
        flag = Flag(flag_name='test')
        flag.set_value('example')
        self.assertEqual(flag.get_value(), 'example')

    def test_validate_incorrect_flag_value_with_list_of_possible_flag_values(self):
        flag = Flag(flag_name='test', possible_flag_values=['1', '2', '3'])
        self.assertEqual(flag.validate_input_flag_value('bad value'), False)

    def test_validate_correct_flag_value_with_list_of_possible_flag_values(self):
        flag = Flag(flag_name='test', possible_flag_values=['1', '2', '3'])
        self.assertEqual(flag.validate_input_flag_value('1'), True)

    def test_validate_incorrect_flag_value_with_pattern_of_possible_flag_values(self):
        flag = Flag(flag_name='test', possible_flag_values=re.compile(r'192.168.\d+.\d+'))
        self.assertEqual(flag.validate_input_flag_value('152.123.9.8'), False)

    def test_validate_correct_flag_value_with_pattern_of_possible_flag_values(self):
        flag = Flag(flag_name='test', possible_flag_values=re.compile(r'192.168.\d+.\d+'))
        self.assertEqual(flag.validate_input_flag_value('192.168.9.8'), True)