from django.test import TestCase
from dockerhub.marketo.utils import flatten


class TestFlatten(TestCase):

    key = 'id'
    items = [
        {'id': 1, 'name': 'johnny'},
        {'id': 2, 'name': 'test', 'email': 'fake@fake_email.com'},
        {'id': 3, 'name': 'johnnytest'}
    ]
    expected_results = [1, 2, 3]

    def setUp(self):
        self.results = flatten(self.key, self.items)

    def test(self):
        self.assertEqual(self.results, self.expected_results)


class TestEmptyItems(TestFlatten):

    items = None
    expected_results = []


class TestEmptyKey(TestFlatten):

    key = None
    expected_results = []


class TestEmptyListOfItems(TestFlatten):

    items = []
    expected_results = []