
from twisted.trial.unittest import TestCase
from zope.interface.verify import verifyObject

from go_api.collections import ICollection

from go_contacts.backends.riak import (
    RiakContactsBackend, RiakContactsCollection)


class TestRiakContactsBackend(TestCase):
    def test_get_contacts_collection(self):
        backend = RiakContactsBackend()
        collection = backend.get_contact_collection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")
        self.assertTrue(isinstance(collection, RiakContactsCollection))


class TestRiakContactsCollection(TestCase):
    def test_collection_provides_ICollection(self):
        """
        The return value of .get_row_collection() is an object that provides
        ICollection.
        """
        collection = RiakContactsCollection("owner-1")
        verifyObject(ICollection, collection)

    def test_init(self):
        collection = RiakContactsCollection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")


from twisted.trial.unittest import TestCase
from zope.interface.verify import verifyObject

from go_api.collections import ICollection

from go_contacts.backends.riak import (
    RiakContactsBackend, RiakContactsCollection)


class TestRiakContactsBackend(TestCase):
    def test_get_contacts_collection(self):
        backend = RiakContactsBackend()
        collection = backend.get_contact_collection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")
        self.assertTrue(isinstance(collection, RiakContactsCollection))


class TestRiakContactsCollection(TestCase):
    def test_collection_provides_ICollection(self):
        """
        The return value of .get_row_collection() is an object that provides
        ICollection.
        """
        collection = RiakContactsCollection("owner-1")
        verifyObject(ICollection, collection)

    def test_init(self):
        collection = RiakContactsCollection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")

    def test_get(self):
        collection = RiakContactsCollection("owner-1")
        self.assertEqual(collection.get("contact-1"), {})

Add stubby test for getting a contact.
Add stubby test for getting a contact.

  
  from twisted.trial.unittest import TestCase
  from zope.interface.verify import verifyObject
  
  from go_api.collections import ICollection
  
  from go_contacts.backends.riak import (
      RiakContactsBackend, RiakContactsCollection)
  
  
  class TestRiakContactsBackend(TestCase):
      def test_get_contacts_collection(self):
          backend = RiakContactsBackend()
          collection = backend.get_contact_collection("owner-1")
          self.assertEqual(collection.owner_id, "owner-1")
          self.assertTrue(isinstance(collection, RiakContactsCollection))
  
  
  class TestRiakContactsCollection(TestCase):
      def test_collection_provides_ICollection(self):
          """
          The return value of .get_row_collection() is an object that provides
          ICollection.
          """
          collection = RiakContactsCollection("owner-1")
          verifyObject(ICollection, collection)
  
      def test_init(self):
          collection = RiakContactsCollection("owner-1")
          self.assertEqual(collection.owner_id, "owner-1")
  
+     def test_get(self):
+         collection = RiakContactsCollection("owner-1")
+         self.assertEqual(collection.get("contact-1"), {})
+ 
Add stubby test for getting a contact.
## Code Before:

from twisted.trial.unittest import TestCase
from zope.interface.verify import verifyObject

from go_api.collections import ICollection

from go_contacts.backends.riak import (
    RiakContactsBackend, RiakContactsCollection)


class TestRiakContactsBackend(TestCase):
    def test_get_contacts_collection(self):
        backend = RiakContactsBackend()
        collection = backend.get_contact_collection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")
        self.assertTrue(isinstance(collection, RiakContactsCollection))


class TestRiakContactsCollection(TestCase):
    def test_collection_provides_ICollection(self):
        """
        The return value of .get_row_collection() is an object that provides
        ICollection.
        """
        collection = RiakContactsCollection("owner-1")
        verifyObject(ICollection, collection)

    def test_init(self):
        collection = RiakContactsCollection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")

## Instruction:
Add stubby test for getting a contact.
## Code After:

from twisted.trial.unittest import TestCase
from zope.interface.verify import verifyObject

from go_api.collections import ICollection

from go_contacts.backends.riak import (
    RiakContactsBackend, RiakContactsCollection)


class TestRiakContactsBackend(TestCase):
    def test_get_contacts_collection(self):
        backend = RiakContactsBackend()
        collection = backend.get_contact_collection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")
        self.assertTrue(isinstance(collection, RiakContactsCollection))


class TestRiakContactsCollection(TestCase):
    def test_collection_provides_ICollection(self):
        """
        The return value of .get_row_collection() is an object that provides
        ICollection.
        """
        collection = RiakContactsCollection("owner-1")
        verifyObject(ICollection, collection)

    def test_init(self):
        collection = RiakContactsCollection("owner-1")
        self.assertEqual(collection.owner_id, "owner-1")

    def test_get(self):
        collection = RiakContactsCollection("owner-1")
        self.assertEqual(collection.get("contact-1"), {})
