Add a field for default beer color inside the style model

beers/migrations/0028_style_default_color.py

@@ -0,0 +1,18 @@
+# Generated by Django 2.2.1 on 2019-05-18 13:41
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('beers', '0027_add_alt_innerspace_names'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='style',
+            name='default_color',
+            field=models.CharField(blank=True, max_length=9, verbose_name='HTML color (in hex) to use if the beer has no known color'),
+        ),
+    ]

beers/models.py

@@ -12,6 +12,11 @@
 
 class Style(models.Model):
     name = CITextField(unique=True)
+    default_color = models.CharField(
+        'HTML color (in hex) to use if the beer has no known color',
+        max_length=9,  # #00112233 -> RGBA
+        blank=True,
+    )
 
     def merge_from(self, other_styles):
         alt_names = []

beers/test/factories.py

@@ -7,6 +7,7 @@
 class StyleFactory(factory.django.DjangoModelFactory):
 
     name = factory.Sequence(lambda n: 'style %d' % n)
+    default_color = factory.Sequence(lambda n: '#{:0>6X}'.format(n))
 
     class Meta:
         model = Style

beers/test/test_views.py

@@ -6,7 +6,7 @@
 from hsv_dot_beer.users.test.factories import UserFactory
 from taps.test.factories import TapFactory
 from beers.serializers import ManufacturerSerializer
-from .factories import ManufacturerFactory, BeerFactory
+from .factories import ManufacturerFactory, BeerFactory, StyleFactory
 
 fake = Faker()
 
@@ -62,7 +62,8 @@ def test_patch(self):
 
 class BeerDetailTestCase(APITestCase):
     def setUp(self):
-        self.beer = BeerFactory()
+        self.style = StyleFactory()
+        self.beer = BeerFactory(style=self.style)
         self.url = reverse(
             'beer-detail', kwargs={'pk': self.beer.pk},
         )
@@ -71,6 +72,11 @@ def setUp(self):
             HTTP_AUTHORIZATION=f'Token {self.user.auth_token}'
         )
 
+    def test_style_embedded(self):
+        response = self.client.get(self.url)
+        eq_(response.status_code, 200)
+        eq_(response.data['style']['default_color'], self.style.default_color)
+
     def test_patch(self):
         data = {
             'name': 'a beer',